aboutsummaryrefslogtreecommitdiffstats
path: root/python/collateral/__main__.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/collateral/__main__.py')
-rw-r--r--python/collateral/__main__.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/python/collateral/__main__.py b/python/collateral/__main__.py
index 84736354..c9ccf3aa 100644
--- a/python/collateral/__main__.py
+++ b/python/collateral/__main__.py
@@ -1,10 +1,10 @@
import pandas as pd
+from exchange import ExchangeMessage
from importlib import import_module
from utils import SerenitasFileHandler
from utils.db import dawn_engine, dbconn
-from . import bus_day
from .common import get_dawn_trades, send_email
from pandas.tseries.offsets import BDay
@@ -33,9 +33,11 @@ parser.add_argument(
args = parser.parse_args()
counterparties = ["citi", "baml_isda", "ms", "gs", "bnp", "baml_fcm", "wells", "cs"]
if args.download:
+ em = ExchangeMessage()
for cp in counterparties:
cp_mod = import_module(f".{cp}", "collateral")
- cp_mod.download_files()
+ print(cp)
+ cp_mod.download_files(em)
dawn_trades = get_dawn_trades(args.workdate, dawn_engine)