diff options
Diffstat (limited to 'python/report_ops/__main__.py')
| -rw-r--r-- | python/report_ops/__main__.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/python/report_ops/__main__.py b/python/report_ops/__main__.py index 9dec4db5..68708109 100644 --- a/python/report_ops/__main__.py +++ b/python/report_ops/__main__.py @@ -1,4 +1,5 @@ from serenitas.analytics.dates import prev_business_day +from serenitas.utils.db import dbconn import logging import argparse import datetime @@ -6,7 +7,7 @@ from .sma import IsoselSMA, BowdstSMA from .cash import NTCashReport, UMBCashReport, BNYCashReport from .admin import AccruedReport, AllReport from .wires import BowdstWire, NTWire -from .custodians import BBH, UMB +from .custodians import upload_to_custodian logger = logging.getLogger(__name__) @@ -91,9 +92,9 @@ if args.wire_reports: logger.info(e) if args.send_to_custodians: - for custodian_cls in (BBH, UMB): - custodian = custodian_cls(args.date) + conn = dbconn("dawndb") + for account in ("UMB", "BBH"): try: - custodian.upload_to_custodian(not args.no_upload) + upload_to_custodian(account, args.date, conn, not args.no_upload) except ValueError as e: logger.info(e) |