diff options
| -rw-r--r-- | python/reallocate_cash.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/python/reallocate_cash.py b/python/reallocate_cash.py index b5f19f00..98a78664 100644 --- a/python/reallocate_cash.py +++ b/python/reallocate_cash.py @@ -10,8 +10,6 @@ from psycopg2.errors import UniqueViolation from serenitas.analytics.dates import bus_day from csv_headers.globeop_upload import HEADERS - -columns = HEADERS["iam"] _brokers = { "BAML_ISDA": "BOANNY", "CS": "CSITLN", @@ -146,8 +144,10 @@ def insert_iam_sql(conn, trade_date): def process_upload(conn, trade_date, iam_deals): buf = StringIO() csvwriter = csv.writer(buf) - csvwriter.writerow(columns) - csvwriter.writerows([[obj.get(h, None) for h in columns] for obj in iam_deals]) + csvwriter.writerow(HEADERS["iam"]) + csvwriter.writerows( + [[obj.get(h, None) for h in HEADERS["iam"]] for obj in iam_deals] + ) buf = buf.getvalue().encode() dest = ( DAILY_DIR @@ -175,5 +175,4 @@ if __name__ == "__main__": args = parser.parse_args() iam_deals = insert_iam_sql(conn, args.date) process_upload(conn, args.date, iam_deals, True) - conn.commit() |