diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/report_ops/__main__.py | 4 | ||||
| -rw-r--r-- | python/report_ops/custodians.py | 3 |
2 files changed, 3 insertions, 4 deletions
diff --git a/python/report_ops/__main__.py b/python/report_ops/__main__.py index b9f03917..52deb55b 100644 --- a/python/report_ops/__main__.py +++ b/python/report_ops/__main__.py @@ -1,6 +1,5 @@ from serenitas.analytics.dates import prev_business_day from serenitas.utils.exchange import ExchangeMessage -from serenitas.utils.db import dbconn import logging import argparse import datetime @@ -104,13 +103,12 @@ if args.wire_reports: logger.info(e) if args.send_to_custodians: - conn = dbconn("dawndb") em = ExchangeMessage() for account in ( "BBH", "UMB", ): try: - upload_to_custodian(account, args.date, conn, not args.no_upload, em) + upload_to_custodian(account, args.date, not args.no_upload, em) except ValueError as e: logger.info(e) diff --git a/python/report_ops/custodians.py b/python/report_ops/custodians.py index d7c81888..8eae92ce 100644 --- a/python/report_ops/custodians.py +++ b/python/report_ops/custodians.py @@ -10,9 +10,10 @@ from typing import ClassVar from dataclasses import dataclass -def upload_to_custodian(account, trade_date, conn, upload, em): +def upload_to_custodian(account, trade_date, upload, em): _service = {"BBH": "BRINKER", "UMB": "UMB"} custodian = Service[_service[account]] + conn = BondDeal._conn with conn.cursor() as c, conn.cursor() as d: c.execute( "SELECT * FROM bond_trades WHERE trade_date=%s AND account=%s", |