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

Merge pull request #5 from cotester/main_refactor

Browse Source 
Refactor main.py #3 stock-income endpoint
This commit is contained in:
Muslem Rahimi 2024-07-03 16:41:55 +02:00 committed by GitHub
commit da209c8d92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/main.py
View File

@ -749,10 +749,15 @@ async def stock_income(data: TickerData):
symbol = ? symbol = ?
""" """
try: try:
df = pd.read_sql_query(query_template,con, params=(ticker,)) with db_connection(STOCK_DB) as cursor:
income_statement = ujson.loads(df['income'].iloc[0]) cursor.execute(query_template, (ticker,))
income_statement_growth = ujson.loads(df['income_growth'].iloc[0]) result = cursor.fetchone()
res = clean_financial_data(income_statement,income_statement_growth) if result:
income_statement = ujson.loads(result[0])
income_statement_growth = ujson.loads(result[1])
res = clean_financial_data(income_statement, income_statement_growth)
else:
res = []
except: except:
res = [] res = []