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

add ytd

Browse Source
This commit is contained in:
MuslemRahimi 2025-04-03 23:00:13 +02:00
parent 89b9470480
commit ecca70ffea
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/cron_historical_price.py
View File

@ -12,6 +12,8 @@ from dotenv import load_dotenv
load_dotenv() load_dotenv()
api_key = os.getenv('FMP_API_KEY') api_key = os.getenv('FMP_API_KEY')
current_year = datetime.today().year
# Helper to ensure directories exist and write JSON files asynchronously # Helper to ensure directories exist and write JSON files asynchronously
async def write_json(path, data): async def write_json(path, data):
os.makedirs(os.path.dirname(path), exist_ok=True) os.makedirs(os.path.dirname(path), exist_ok=True)
@ -65,10 +67,14 @@ async def get_historical_data(ticker, query_con, session):
df_max = pd.read_sql_query(query, query_con, params=(start_date_max, end_date)) df_max = pd.read_sql_query(query, query_con, params=(start_date_max, end_date))
df_max = df_max.round(2).rename(columns={"date": "time"}) df_max = df_max.round(2).rename(columns={"date": "time"})
max_list = ujson.loads(df_max.to_json(orient="records"))
ytd_data = [entry for entry in max_list if datetime.strptime(entry["time"], "%Y-%m-%d").year == current_year]
# Prepare file-writing tasks # Prepare file-writing tasks
tasks = [ tasks = [
write_json(f"json/historical-price/one-week/{ticker}.json", ujson.loads(data[0])), write_json(f"json/historical-price/one-week/{ticker}.json", ujson.loads(data[0])),
write_json(f"json/historical-price/one-month/{ticker}.json", ujson.loads(data[1])), write_json(f"json/historical-price/one-month/{ticker}.json", ujson.loads(data[1])),
write_json(f"json/historical-price/ytd/{ticker}.json", ytd_data),
write_json(f"json/historical-price/six-months/{ticker}.json", ujson.loads(df_6m.to_json(orient="records"))), write_json(f"json/historical-price/six-months/{ticker}.json", ujson.loads(df_6m.to_json(orient="records"))),
write_json(f"json/historical-price/one-year/{ticker}.json", ujson.loads(df_1y.to_json(orient="records"))), write_json(f"json/historical-price/one-year/{ticker}.json", ujson.loads(df_1y.to_json(orient="records"))),
write_json(f"json/historical-price/five-years/{ticker}.json", ujson.loads(df_5y.to_json(orient="records"))), write_json(f"json/historical-price/five-years/{ticker}.json", ujson.loads(df_5y.to_json(orient="records"))),

4
app/cron_revenue.py
View File

@ -69,7 +69,6 @@ async def get_data(symbol):
] ]
stats = await get_statistics(symbol) stats = await get_statistics(symbol)
print(stats)
res_dict = {**stats, 'annual': annual_data, 'quarter': quarter_data} res_dict = {**stats, 'annual': annual_data, 'quarter': quarter_data}
if annual_data and quarter_data: if annual_data and quarter_data:
@ -84,7 +83,8 @@ async def run():
con.close() con.close()
#Testing mode #Testing mode
#total_symbols = ['AAPL'] #total_symbols = ['DGICA']
for symbol in tqdm(total_symbols): for symbol in tqdm(total_symbols):
try: try:
await get_data(symbol) await get_data(symbol)