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

bugfixing

Browse Source
This commit is contained in:
MuslemRahimi 2024-11-25 12:42:20 +01:00
parent 9c39ce21bc
commit 2ba1dfbbcb
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app/cron_analyst_insight.py
View File

@ -8,6 +8,8 @@ from tqdm import tqdm
from datetime import datetime
from openai import OpenAI
import aiofiles
import time
# Load environment variables
load_dotenv()
@ -30,31 +32,19 @@ async def save_json(symbol, data):
await file.write(ujson.dumps(data))
async def get_analyst_insight(session, ticker):
#there is a bug in benzinga api where the latest data can be at the second element instead the first
#solution check which date is the latest by comparing these two
res_dict = {}
try:
querystring = {"token": benzinga_api_key, "symbols": ticker}
async with session.get(url, params=querystring) as response:
output = await response.json()
if 'analyst-insights' in output and len(output['analyst-insights']) >= 2:
insight_1, insight_2 = output['analyst-insights'][:2]
# Parse dates for comparison
date_1 = datetime.strptime(insight_1['date'], "%Y-%m-%d")
date_2 = datetime.strptime(insight_2['date'], "%Y-%m-%d")
# Choose the insight with the latest date
latest_insight = insight_1 if date_1 >= date_2 else insight_2
elif 'analyst-insights' in output and output['analyst-insights']:
latest_insight = output['analyst-insights'][0] # Only one insight available
else:
return res_dict # No insights available
output = (await response.json())['analyst-insights']
output = sorted(output, key=lambda x: datetime.strptime(x['date'], '%Y-%m-%d'), reverse=True)
latest_insight = output[0]
# Populate res_dict with the latest insight data
res_dict = {
'insight': latest_insight['analyst_insights'],
'id': latest_insight['id'],
'pt': round(float(latest_insight.get('pt'))) if latest_insight.get('pt', None) is not None else None,
'date': datetime.strptime(latest_insight['date'], "%Y-%m-%d").strftime("%b %d, %Y")
}
except Exception as e:
@ -113,6 +103,10 @@ async def main():
cursor = con.cursor()
cursor.execute("SELECT DISTINCT symbol FROM stocks WHERE symbol NOT LIKE '%.%'")
stock_symbols = [row[0] for row in cursor.fetchall()]
#TestMode
#stock_symbols = ['WMT']
con.close()
async with aiohttp.ClientSession(headers=headers) as session: