diff options
| -rw-r--r-- | python/monthend_interest_recon.py | 26 |
1 files changed, 16 insertions, 10 deletions
diff --git a/python/monthend_interest_recon.py b/python/monthend_interest_recon.py index 4a45fd8b..efec80c1 100644 --- a/python/monthend_interest_recon.py +++ b/python/monthend_interest_recon.py @@ -21,16 +21,22 @@ logger = logging.getLogger(__name__) def get_fpath(counterparty, save=False): - if not save: - return DAILY_DIR / "Serenitas" / "MonthlyInterest" / f"{counterparty}_reports" - elif counterparty == "CITI": - return DAILY_DIR / f"{counterparty}_reports" / "Interest Statements" - elif counterparty == "BAML": - return DAILY_DIR / "Serenitas" / f"BoA_reports" / "Interest Statements" - else: - return ( - DAILY_DIR / "Serenitas" / f"{counterparty}_reports" / "Interest Statements" - ) + match (counterparty, save): + case (counterparty, False): + return ( + DAILY_DIR / "Serenitas" / "MonthlyInterest" / f"{counterparty}_reports" + ) + case ("CITI", True): + return DAILY_DIR / f"{counterparty}_reports" / "Interest Statements" + case ("BAML", True): + return DAILY_DIR / "Serenitas" / f"BoA_reports" / "Interest Statements" + case _: + return ( + DAILY_DIR + / "Serenitas" + / f"{counterparty}_reports" + / "Interest Statements" + ) def download_messages(em, counterparty, start, end, save=False): |