AmineTMIRI/backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update screener

Browse Source
This commit is contained in:
MuslemRahimi 2024-10-17 19:59:29 +02:00
parent eed3370b61
commit 80f9323117
1 changed files with 19 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
app/restart_json.py
View File

@ -53,23 +53,23 @@ def generate_id(name):
def count_consecutive_growth_years(financial_data): def count_consecutive_growth_years(financial_data, key_element):
# Sort the financial data by date # Sort the financial data by date
financial_data = sorted(financial_data, key=lambda x: datetime.strptime(x['date'], '%Y-%m-%d')) financial_data = sorted(financial_data, key=lambda x: datetime.strptime(x['date'], '%Y-%m-%d'))
consecutive_years = 0 consecutive_years = 0
prev_revenue = None prev_val = None
for data in financial_data: for data in financial_data:
current_revenue = data['revenue'] current_val = data[key_element] #e.g. revenue
if current_revenue is not None: if current_val is not None:
if prev_revenue is not None: if prev_val is not None:
if current_revenue > prev_revenue: if current_val > prev_val:
consecutive_years += 1 consecutive_years += 1
else: else:
consecutive_years = 0 consecutive_years = 0
prev_revenue = current_revenue prev_val = current_val
# Check one last time in case the streak continues to the end # Check one last time in case the streak continues to the end
@ -569,25 +569,6 @@ async def get_stock_screener(con):
item['returnOnTangibleAssets'] = None item['returnOnTangibleAssets'] = None
item['grahamNumber'] = None item['grahamNumber'] = None
try:
with open(f"json/trend-analysis/{symbol}.json", 'r') as file:
res = orjson.loads(file.read())[-1]
if abs(res['accuracy'] - res['precision']) <=15 and res['sentiment'] == 'Bullish':
item['trendAnalysis'] = {"accuracy": res['accuracy']}
else:
item['trendAnalysis'] = {"accuracy": None}
except:
item['trendAnalysis'] = {"accuracy": None}
try:
with open(f"json/fundamental-predictor-analysis/{symbol}.json", 'r') as file:
res = orjson.loads(file.read())
if abs(res['accuracy'] - res['precision']) <=15 and res['sentiment'] == 'Bullish':
item['fundamentalAnalysis'] = {"accuracy": res['accuracy']}
else:
item['fundamentalAnalysis'] = {"accuracy": None}
except:
item['fundamentalAnalysis'] = {"accuracy": None}
try: try:
with open(f"json/ai-score/companies/{symbol}.json", 'r') as file: with open(f"json/ai-score/companies/{symbol}.json", 'r') as file:
@ -679,9 +660,15 @@ async def get_stock_screener(con):
try: try:
with open(f"json/financial-statements/income-statement/annual/{symbol}.json", "r") as file: with open(f"json/financial-statements/income-statement/annual/{symbol}.json", "r") as file:
financial_data = orjson.loads(file.read()) financial_data = orjson.loads(file.read())
item['revenueGrowthYears'] = count_consecutive_growth_years(financial_data) item['revenueGrowthYears'] = count_consecutive_growth_years(financial_data, "revenue")
item['epsGrowthYears'] = count_consecutive_growth_years(financial_data, 'eps')
item['netIncomeGrowthYears'] = count_consecutive_growth_years(financial_data, 'netIncome')
item['grossProfitGrowthYears'] = count_consecutive_growth_years(financial_data, 'grossProfit')
except: except:
item['revenueGrowthYears'] = None item['revenueGrowthYears'] = None
item['epsGrowthYears'] = None
item['netIncomeGrowthYears'] = None
item['grossProfitGrowthYears'] = None
return stock_screener_data return stock_screener_data
@ -1862,20 +1849,13 @@ async def get_most_shorted_stocks(con):
sorted_list = sorted(extracted_data, key=lambda x: x['shortOutStandingPercent'], reverse=True) sorted_list = sorted(extracted_data, key=lambda x: x['shortOutStandingPercent'], reverse=True)
query_template = """ for index, item in enumerate(sorted_list, start=1):
SELECT
name, sector
FROM
stocks
WHERE
symbol = ?
"""
for item in sorted_list:
try: try:
symbol = item['symbol'] symbol = item['symbol']
data = pd.read_sql_query(query_template, con, params=(symbol,)) with open(f"json/quote/{symbol}.json") as file:
item['name'] = data['name'].iloc[0] data = orjson.loads(file.read())
item['name'] = data['name']
item['rank'] = index
#item['sector'] = data['sector'].iloc[0] #item['sector'] = data['sector'].iloc[0]
except Exception as e: except Exception as e:
print(e) print(e)