add newsletter cron job

app/cron_newsletter.py

@@ -0,0 +1,96 @@
+import aiohttp
+import aiofiles
+import ujson
+import sqlite3
+import pandas as pd
+import asyncio
+import pytz
+import time
+import os
+from dotenv import load_dotenv
+from datetime import datetime, timedelta
+import sqlite3
+
+headers = {"accept": "application/json"}
+
+
+
+
+load_dotenv()
+benzinga_api_key = os.getenv('BENZINGA_API_KEY_EXTRA')
+
+
+
+query_template = """
+    SELECT 
+        marketCap
+    FROM 
+        stocks 
+    WHERE
+        symbol = ?
+"""
+
+async def save_json(data):
+    with open(f"json/newsletter/data.json", 'w') as file:
+        ujson.dump(data, file)
+
+
+async def get_upcoming_earnings(session):
+	url = "https://api.benzinga.com/api/v2.1/calendar/earnings"
+	querystring = {"token": benzinga_api_key,"parameters[date_from]":"2024-07-30","parameters[date_to]":"2024-07-30"}
+	try:
+		async with session.get(url, params=querystring, headers=headers) as response:
+			res_list = []
+			res = ujson.loads(await response.text())['earnings']
+			for item in res:
+				try:
+					symbol = item['ticker']
+					name = item['name']
+					time = item['time']
+					eps_prior = float(item['eps_prior']) if item['eps_prior'] != '' else 0
+					eps_est = float(item['eps_est']) if item['eps_est'] != '' else 0
+					revenue_est = float(item['revenue_est']) if item['revenue_est'] != '' else 0
+					revenue_prior = float(item['revenue_prior']) if item['revenue_prior'] != '' else 0
+					if symbol in stock_symbols and revenue_est != 0 and revenue_prior != 0 and eps_prior != 0 and eps_est != 0:
+						df = pd.read_sql_query(query_template, con, params=(symbol,))
+						market_cap = float(df['marketCap'].iloc[0]) if df['marketCap'].iloc[0] != '' else 0
+						res_list.append({
+							'symbol': symbol,
+							'name': name,
+							'time': time,
+							'marketCap': market_cap,
+							'epsPrior':eps_prior,
+							'epsEst': eps_est,
+							'revenuePrior': revenue_prior,
+							'revenueEst': revenue_est
+							})
+				except Exception as e:
+					print(e)
+					pass
+		res_list.sort(key=lambda x: x['marketCap'], reverse=True)
+		return res_list
+	except Exception as e:
+		print(e)
+
+
+async def run():
+	async with aiohttp.ClientSession() as session:
+		upcoming_earnings = await get_upcoming_earnings(session)		
+
+		data = {'upcomingEarnings': upcoming_earnings}
+		
+		if len(data) > 0:
+			await save_json(data)
+		
+
+try:
+	con = sqlite3.connect('stocks.db')
+	cursor = con.cursor()
+	cursor.execute("PRAGMA journal_mode = wal")
+	cursor.execute("SELECT DISTINCT symbol FROM stocks")
+	stock_symbols = [row[0] for row in cursor.fetchall()]
+
+	asyncio.run(run())
+	con.close()
+except Exception as e:
+    print(e)

app/main.py

@@ -3313,4 +3313,13 @@ async def get_reddit_tracker(api_key: str = Security(get_api_key)):
         io.BytesIO(compressed_data),
         media_type="application/json",
         headers={"Content-Encoding": "gzip"}
-    )
+    )
+
+@app.get("/newsletter")
+async def get_newsletter():
+    try:
+        with open(f"json/newsletter/data.json", 'rb') as file:
+            res = orjson.loads(file.read())
+    except:
+        res = []
+    return res