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 2025-04-07 14:19:31 +02:00
parent fdde91673f
commit a2d8b23005
1 changed files with 30 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
app/main.py
View File

@ -922,7 +922,7 @@ async def economic_calendar(data:TickerData, api_key: str = Security(get_api_key
async def get_data(data: FinancialStatement, api_key: str = Security(get_api_key)): async def get_data(data: FinancialStatement, api_key: str = Security(get_api_key)):
ticker = data.ticker.upper() ticker = data.ticker.upper()
statement = data.statement.lower() statement = data.statement.lower()
cache_key = f"financial-statement-{ticker}" cache_key = f"financial-statement-{ticker}-{statement}"
cached_result = redis_client.get(cache_key) cached_result = redis_client.get(cache_key)
if cached_result: if cached_result:
@ -932,7 +932,7 @@ async def get_data(data: FinancialStatement, api_key: str = Security(get_api_key
headers={"Content-Encoding": "gzip"} headers={"Content-Encoding": "gzip"}
) )
keys_to_remove = {"symbol","reportedCurrency", "cik", "filingDate", "acceptedDate"} keys_to_remove = {"symbol", "cik", "filingDate", "acceptedDate"}
base_path = f"json/financial-statements/{statement}" base_path = f"json/financial-statements/{statement}"
def load_and_clean(path: str): def load_and_clean(path: str):
@ -942,24 +942,39 @@ async def get_data(data: FinancialStatement, api_key: str = Security(get_api_key
raw_data = orjson.loads(f.read()) raw_data = orjson.loads(f.read())
cleaned = [] cleaned = []
for sublist in raw_data: for sublist in raw_data:
# Ensure sublist is a list; if not, skip or wrap in a list as appropriate.
if isinstance(sublist, list): if isinstance(sublist, list):
cleaned_sublist = [ cleaned_sublist = []
{k: v for k, v in item.items() if k not in keys_to_remove} for item in sublist:
if isinstance(item, dict) else item if isinstance(item, dict):
for item in sublist cleaned_item = {}
] for k, v in item.items():
if k in keys_to_remove:
continue
# Remove "TTM" from the key
new_key = k.replace("TTM", "")
# Optionally, if you want to remove "TTM" from string values of a specific key,
# you could also do:
if new_key == "key" and isinstance(v, str):
v = v.replace("TTM", "")
cleaned_item[new_key] = v
cleaned_sublist.append(cleaned_item)
else:
cleaned_sublist.append(item)
cleaned.append(cleaned_sublist) cleaned.append(cleaned_sublist)
elif isinstance(sublist, dict):
cleaned_item = {}
for k, v in sublist.items():
if k in keys_to_remove:
continue
new_key = k.replace("TTM", "")
if new_key == "key" and isinstance(v, str):
v = v.replace("TTM", "")
cleaned_item[new_key] = v
cleaned.append(cleaned_item)
else: else:
# If sublist isn't a list, we can choose to ignore it or process it as a single dict. cleaned.append(sublist)
if isinstance(sublist, dict):
cleaned.append({k: v for k, v in sublist.items() if k not in keys_to_remove})
else:
# Append as is if it's neither a list nor a dict.
cleaned.append(sublist)
return cleaned return cleaned
quarter_res = load_and_clean(f"{base_path}/quarter/{ticker}.json") quarter_res = load_and_clean(f"{base_path}/quarter/{ticker}.json")
annual_res = load_and_clean(f"{base_path}/annual/{ticker}.json") annual_res = load_and_clean(f"{base_path}/annual/{ticker}.json")
ttm_res = load_and_clean(f"{base_path}/ttm/{ticker}.json") ttm_res = load_and_clean(f"{base_path}/ttm/{ticker}.json")