diff options
Diffstat (limited to 'python/task_server')
| -rw-r--r-- | python/task_server/globeop.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 7a26f170..e95823c8 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -65,10 +65,11 @@ def get_gpg(): return gpg def convert_to_csv(f): + mapping = (('Credit Default Swap', 'CDS'), ('Swaption', 'Swaption'), ('ALL', 'All')) if f.exists(): - for sheet in ["Credit Default Swap", "Swaption"]: + for sheet, name in mapping: df = pd.read_excel(f, sheet_name=sheet, skiprows=[0, 1, 2, 3]) - df.to_csv(f.parent / f"{sheet}.csv", index=False) + df.to_csv(f.parent / f"{name}_Report.csv", index=False) f.unlink() def download_data(workdate): |