1 files changed, 18 insertions, 5 deletions
diff --git a/python/report_ops/__main__.py b/python/report_ops/__main__.py
index 68708109..194dd536 100644
--- a/python/report_ops/__main__.py
+++ b/python/report_ops/__main__.py
@@ -1,9 +1,19 @@
 from serenitas.analytics.dates import prev_business_day
-from serenitas.utils.db import dbconn
+from serenitas.utils.exchange import ExchangeMessage
 import logging
 import argparse
 import datetime
-from .sma import IsoselSMA, BowdstSMA
+from .sma import (
+    IsoselSMA,
+    BowdstSMA,
+    PositionReport,
+    BondPosition,
+    FuturePosition,
+    TranchePosition,
+    CDXPosition,
+    IRSwaptionPosition,
+    CDXSwaptionPosition,
+)
 from .cash import NTCashReport, UMBCashReport, BNYCashReport
 from .admin import AccruedReport, AllReport
 from .wires import BowdstWire, NTWire
@@ -92,9 +102,12 @@ if args.wire_reports:
             logger.info(e)
 
 if args.send_to_custodians:
-    conn = dbconn("dawndb")
-    for account in ("UMB", "BBH"):
+    em = ExchangeMessage()
+    for account in (
+        "BBH",
+        "UMB",
+    ):
         try:
-            upload_to_custodian(account, args.date, conn, not args.no_upload)
+            upload_to_custodian(account, args.date, not args.no_upload, em)
         except ValueError as e:
             logger.info(e)