diff options
| -rw-r--r-- | python/monthend_interest_recon.py | 35 |
1 files changed, 17 insertions, 18 deletions
diff --git a/python/monthend_interest_recon.py b/python/monthend_interest_recon.py index 8bfd51e8..43fc8d82 100644 --- a/python/monthend_interest_recon.py +++ b/python/monthend_interest_recon.py @@ -1,16 +1,17 @@ -from serenitas.utils.env import DAILY_DIR -from serenitas.utils.exchange import ExchangeMessage import datetime import logging import argparse - -from collateral.common import load_pdf from pathlib import Path -import pandas as pd from collections import defaultdict +import pandas as pd +from dateutil.relativedelta import relativedelta +from serenitas.utils.env import DAILY_DIR +from serenitas.utils.exchange import ExchangeMessage + +from collateral.common import load_pdf from interest_statement import export_data -from dateutil.relativedelta import relativedelta +from report_ops.misc import em_date_filter logger = logging.getLogger(__name__) @@ -27,20 +28,18 @@ def get_fpath(counterparty, save=False): def download_messages(em, counterparty, start, end, save=False): for msg in em.get_msgs( - 20, - path=["Interest", counterparty], + 20, path=["Interest", counterparty], **em_date_filter(em, start, end) ): base_dir = get_fpath(counterparty, save) - if (msg.datetime_sent.date() >= start) and (msg.datetime_sent.date() <= end): - for attach in msg.attachments: - fname = attach.name - if (counterparty == "CS") and not ("Interest" in fname): - continue - p = base_dir / fname - if not p.parent.exists(): - p.parent.mkdir(parents=True) - if not p.exists(): - p.write_bytes(attach.content) + for attach in msg.attachments: + fname = attach.name + if (counterparty == "CS") and not ("Interest" in fname): + continue + p = base_dir / fname + if not p.parent.exists(): + p.parent.mkdir(parents=True) + if not p.exists(): + p.write_bytes(attach.content) def get_CS(g): |