diff options
| -rw-r--r-- | python/cds_rebook.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/python/cds_rebook.py b/python/cds_rebook.py index 6d0f23b1..767f8d3a 100644 --- a/python/cds_rebook.py +++ b/python/cds_rebook.py @@ -96,6 +96,19 @@ def rebook(trade_date, company_id, seniority, fcm): dawn_engine.execute(sql_str, [trade_prev.values(), trade_new.values()]) +def insert_newids(dawndb, d: datetime.date, df: pd.DataFrame): + with dawndb.cursor() as c: + c.execute( + "INSERT INTO id_mapping ( " + " SELECT %s, 'CDS', serenitas_id, new_id FROM ( " + " SELECT * FROM unnest(%s, %s) AS t(globeop_id, new_id) " + " LEFT JOIN id_mapping USING (globeop_id)) a" + ")", + (d, df.old_ids.tolist(), df.new_ids.to_list()), + ) + dawndb.commit() + + if __name__ == "__main__": # PKD # rebook(datetime.date(2019, 1, 24), 101148) |