diff options
Diffstat (limited to 'python/task_server')
| -rw-r--r-- | python/task_server/globeop.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 5a73de54..c5bb1750 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -85,13 +85,13 @@ def download_data(engine, workdate: datetime.date): pnlfiles = [ filename for filename in files - if "csv" in filename and "Profit" in filename + if filename.endswith("csv") and "Profit" in filename if get_ped(filename) < workdate ] valuationfiles = [ filename for filename in files - if "csv" in filename and "Valuation_TradeID" in filename + if filename.endswith("csv") and "Valuation_TradeID" in filename if get_ped(filename) < workdate ] cdsfiles = [ @@ -153,7 +153,7 @@ def insert_todb(engine, workdate: datetime.date): / str(workdate) / "Reports" ) - for report in ["Valuation", "Pnl", "CDS"]: + for report in ("Valuation", "Pnl", "CDS"): fun = getattr(load_globeop_report, f"read_{report.lower()}_report") table = f"{report.lower()}_reports" report_file = reports_dir / f"{report}_Report.csv" @@ -165,7 +165,7 @@ def insert_todb(engine, workdate: datetime.date): sql_str = "DELETE FROM valuation_reports WHERE periodenddate=%s" else: df["date"] = period_end_date - sql_str = "DELETE FROM {} WHERE date=%s".format(table) + sql_str = f"DELETE FROM {table} WHERE date=%s" df["row"] = df.index engine.execute(sql_str, (period_end_date,)) df.to_sql(table, engine, if_exists="append", index=False) |