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

Merge pull request #6 from cotester/main_refactor

Browse Source 
Fix import secrets. Refactor /stock-balance-sheet to use context manager
This commit is contained in:
Muslem Rahimi 2024-07-05 12:10:16 +02:00 committed by GitHub
commit fd6e850f89
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
app/main.py
View File

@ -4,6 +4,7 @@ import io
import gzip
import re
import os
import secrets
from typing import List, Dict, Set
# Third-party library imports
@ -784,16 +785,20 @@ async def stock_balance_sheet(data: TickerData):
symbol = ?
"""
try:
df = pd.read_sql_query(query_template,con, params=(ticker,))
balance_statement = ujson.loads(df['balance'].iloc[0])
balance_statement_growth = ujson.loads(df['balance_growth'].iloc[0])
res = clean_financial_data(balance_statement,balance_statement_growth)
with db_connection(STOCK_DB) as cursor:
cursor.execute(query_template, (ticker,))
result = cursor.fetchone()
if result:
balance_statement = ujson.loads(result[0])
balance_statement_growth = ujson.loads(result[1])
res = clean_financial_data(balance_statement, balance_statement_growth)
else:
res = []
except:
res = []
redis_client.set(cache_key, ujson.dumps(res))
redis_client.expire(cache_key, 3600*3600) # Set cache expiration time to 1 hour
redis_client.expire(cache_key, 3600*3600)
return res
@app.post("/stock-ratios")