diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/monthend_interest_recon.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/python/monthend_interest_recon.py b/python/monthend_interest_recon.py index b2f29309..f5382da7 100644 --- a/python/monthend_interest_recon.py +++ b/python/monthend_interest_recon.py @@ -39,9 +39,7 @@ def download_messages(em, counterparty, start, end, save=False): path=["Interest", counterparty], ): base_dir = get_fpath(counterparty, save) - if (msg.datetime_sent.date() >= datetime.date.fromisoformat(start)) and ( - msg.datetime_sent.date() <= datetime.date.fromisoformat(end) - ): + if (msg.datetime_sent.date() >= start) and (msg.datetime_sent.date() <= end): if counterparty == "BAML": soup = BeautifulSoup(msg.body, features="lxml") a = soup.find("a") @@ -153,7 +151,9 @@ def main(): parser = argparse.ArgumentParser(description="determine sender destination") parser.add_argument( - "date", default=datetime.date.today() + "date", + type=datetime.date.fromisoformat, + default=datetime.date.today(), ) parser.add_argument( "--save", @@ -174,9 +174,7 @@ def main(): download_messages(em, cp, start, args.date, args.save) df = get_interest(counterparties, save=args.save) - start, end = start_end( - args.date - ) + start, end = start_end(args.date) global interest_recon interest_recon = pd.merge( export_data(start, end).groupby("broker").sum(), |