diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/monthly_interest.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/python/monthly_interest.py b/python/monthly_interest.py index 1d693e48..73485241 100644 --- a/python/monthly_interest.py +++ b/python/monthly_interest.py @@ -5,25 +5,31 @@ from collateral.baml_isda import download_from_secure_id from bs4 import BeautifulSoup from urllib.parse import urlsplit, parse_qs, urlunsplit import logging +import argparse logger = logging.getLogger(__name__) -def download_messages(em, counterparty): +def download_messages(em, counterparty, start, end): for msg in em.get_msgs( 20, path=["Interest", counterparty], ): if counterparty == "CITI": base_dir = DAILY_DIR / f"{counterparty}_reports" / "Interest Statements" + elif counterparty == "BAML": + base_dir = DAILY_DIR / f"BoA_reports" / "Interest Statements" else: base_dir = ( DAILY_DIR / "Serenitas" + / "Test" / f"{counterparty}_reports" / "Interest Statements" ) - if msg.datetime_sent.date() > datetime.date(2021, 5, 1): + if (msg.datetime_sent.date() > datetime.date.fromisoformat(start)) and ( + msg.datetime_sent.date() < datetime.date.fromisoformat(end) + ): if counterparty == "BAML": soup = BeautifulSoup(msg.body, features="lxml") a = soup.find("a") @@ -52,5 +58,10 @@ def download_messages(em, counterparty): em = ExchangeMessage() counterparties = ["BAML", "BNP", "CITI", "CS", "GS", "MS"] +parser = argparse.ArgumentParser(description="determine sender destination") +parser.add_argument("start") +parser.add_argument("end") +args = parser.parse_args() + for cp in counterparties: - download_messages(em, cp) + download_messages(em, cp, args.start, args.end) |