diff options
Diffstat (limited to 'python/task_server/globeop.py')
| -rw-r--r-- | python/task_server/globeop.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 362d0cce..96ab050f 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -162,22 +162,28 @@ def insert_todb(engine, workdate: datetime.date, fund="SERCGMAST"): fun = getattr(load_globeop_report, f"read_{report.lower()}_report") table = f"{report.lower()}_reports" report_file = reports_dir / f"{report}_Report_{fund}.csv" + alias_names = { + "SERCGMAST": ("SERCGMAST", "SERCGLTD", "SERCGLLC"), + "BOWDST": ("BOWDST",), + } if not report_file.exists(): continue df = fun(report_file) if report == "Valuation": period_end_date = pd.Timestamp(df.periodenddate[0]) - sql_str = "DELETE FROM valuation_reports WHERE periodenddate=%s and fund=%s" + sql_str = ( + "DELETE FROM valuation_reports WHERE periodenddate=%s and fund in %s" + ) else: df["date"] = period_end_date - sql_str = f"DELETE FROM {table} WHERE date=%s and fund=%s" + sql_str = f"DELETE FROM {table} WHERE date=%s and fund in %s" df["row"] = df.index with engine.begin() as conn: conn.execute( sql_str, ( period_end_date, - fund, + alias_names[fund], ), ) df.to_sql(table, conn, if_exists="append", index=False) |