diff options
Diffstat (limited to 'python/reallocate_cash.py')
| -rw-r--r-- | python/reallocate_cash.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/python/reallocate_cash.py b/python/reallocate_cash.py index 48069ebe..1a9d277f 100644 --- a/python/reallocate_cash.py +++ b/python/reallocate_cash.py @@ -57,7 +57,7 @@ _brokers = { } -def iam_process(obj, action): +def iam_process(obj, action, trade_date): rename_keys( obj, { @@ -136,14 +136,12 @@ def insert_iam_sql(conn, trade_date): c.executemany(insert_query, data) except Exception as e: # We already uploaded the IAM tickets today in that case, we need to update and cancel the old uploads - breakpoint() conn.rollback() c.execute( "DELETE FROM iam_tickets where trade_date=%s returning *", (trade_date,) ) for row in c: - print(row) - cancel_trades.append(iam_process(row._asdict(), "CANCEL")) + cancel_trades.append(iam_process(row._asdict(), "CANCEL", trade_date)) c.executemany(insert_query, data) return cancel_trades @@ -163,7 +161,7 @@ def process_upload(conn, trade_date, cancel_trades, upload=True): with conn.cursor() as c: c.execute(query, (trade_date, trade_date)) for row in c: - csv_lines.append(iam_process(row._asdict(), action)) + csv_lines.append(iam_process(row._asdict(), action, trade_date)) buf = StringIO() csvwriter = csv.writer(buf) |