diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/task_server/globeop.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 302407ec..b3dbe39c 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -67,12 +67,12 @@ def get_gpg(): return gpg -def convert_to_csv(f): +def convert_to_csv(f, fund): mapping = (("Credit Default Swap", "CDS"), ("Swaption", "Swaption"), ("ALL", "All")) if f.exists(): for sheet, name in mapping: df = pd.read_excel(f, sheet_name=sheet, skiprows=[0, 1, 2, 3]) - df.to_csv(f.parent / f"{f.name.removesuffix('.xls')}.csv", index=False) + df.to_csv(f.parent / f"{name}_Report_{fund}.csv", index=False) f.unlink() @@ -142,7 +142,7 @@ def download_data(engine, workdate: datetime.date, fund="SERCGMAST"): logger.info(f"{filename}: {dec.status}") (reports_dir / filename).unlink() # convert xls to csv - convert_to_csv(reports_dir / f"CDS_Report_{fund}.xls") + convert_to_csv(reports_dir / f"CDS_Report_{fund}.xls", fund) insert_todb(engine, workdate, fund) |
