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

rename key element in etf holding

Browse Source
This commit is contained in:
MuslemRahimi 2024-11-17 16:51:28 +01:00
parent a36aa83337
commit 64c1695a32
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
app/cron_list.py
View File

@ -81,7 +81,6 @@ async def process_category(cursor, category, condition, category_type='market-ca
async def get_etf_holding(etf_symbols, etf_con): async def get_etf_holding(etf_symbols, etf_con):
for ticker in tqdm(etf_symbols): for ticker in tqdm(etf_symbols):
res = [] res = []
df = pd.read_sql_query(query_etf_holding, etf_con, params=(ticker,)) df = pd.read_sql_query(query_etf_holding, etf_con, params=(ticker,))
@ -89,24 +88,30 @@ async def get_etf_holding(etf_symbols, etf_con):
try: try:
# Load holdings data from the SQL query result # Load holdings data from the SQL query result
data = orjson.loads(df['holding'].iloc[0]) data = orjson.loads(df['holding'].iloc[0])
res = [{key: item[key] for key in ('asset', 'weightPercentage', 'sharesNumber')} for item in data]
# Rename 'asset' to 'symbol' and keep other keys the same
res = [{'symbol': item['asset'],
'weightPercentage': item['weightPercentage'],
'sharesNumber': item['sharesNumber']}
for item in data]
for item in res: for item in res:
asset = item['asset'] symbol = item['symbol']
# Check if the asset data is already in the cache # Check if the symbol data is already in the cache
if asset in quote_cache: if symbol in quote_cache:
quote_data = quote_cache[asset] quote_data = quote_cache[symbol]
else: else:
# Load the quote data from file if not in cache # Load the quote data from file if not in cache
try: try:
with open(f"json/quote/{asset}.json") as file: with open(f"json/quote/{symbol}.json") as file:
quote_data = orjson.loads(file.read()) quote_data = orjson.loads(file.read())
quote_cache[asset] = quote_data # Cache the loaded data quote_cache[symbol] = quote_data # Cache the loaded data
except: except:
quote_data = None quote_data = None
# Assign price and changesPercentage if available, otherwise set to None # Assign price and changesPercentage if available, otherwise set to None
item['weightPercentage'] = round(item.get('weightPercentage'), 2) if item['weightPercentage'] else None
item['price'] = round(quote_data.get('price'), 2) if quote_data else None item['price'] = round(quote_data.get('price'), 2) if quote_data else None
item['changesPercentage'] = round(quote_data.get('changesPercentage'), 2) if quote_data else None item['changesPercentage'] = round(quote_data.get('changesPercentage'), 2) if quote_data else None
item['name'] = quote_data.get('name') if quote_data else None item['name'] = quote_data.get('name') if quote_data else None
@ -117,9 +122,12 @@ async def get_etf_holding(etf_symbols, etf_con):
# Save results to a file if there's data to write # Save results to a file if there's data to write
if res: if res:
for rank, item in enumerate(res, 1):
item['rank'] = rank
with open(f"json/etf/holding/{ticker}.json", 'wb') as file: with open(f"json/etf/holding/{ticker}.json", 'wb') as file:
file.write(orjson.dumps(res)) file.write(orjson.dumps(res))
async def get_etf_provider(etf_con): async def get_etf_provider(etf_con):
cursor = etf_con.cursor() cursor = etf_con.cursor()