diff options
Diffstat (limited to 'python')
| -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 dec57ce5..0f0f60c9 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -168,8 +168,9 @@ def insert_todb(engine, workdate: datetime.date): df["date"] = period_end_date 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) + with engine.begin() as conn: + conn.execute(sql_str, (period_end_date,)) + df.to_sql(table, conn, if_exists="append", index=False) def upload_bond_marks(engine, workdate: datetime.datetime): |