diff options
Diffstat (limited to 'python/bowdst.py')
| -rw-r--r-- | python/bowdst.py | 24 |
1 files changed, 18 insertions, 6 deletions
diff --git a/python/bowdst.py b/python/bowdst.py index 16a90589..968585ea 100644 --- a/python/bowdst.py +++ b/python/bowdst.py @@ -1,5 +1,6 @@ import datetime import pandas as pd +import pathlib from dates import bus_day from env import DAILY_DIR from exchangelib import FileAttachment @@ -31,9 +32,21 @@ def download_messages(em): p.write_bytes(attach.content) -def load_val_report(workdate: datetime.date): +def get_dir(workdate: datetime.date) -> pathlib.Path: + p = DAILY_DIR / str(workdate) / "Reports" + if not p.exists(): + p = ( + DAILY_DIR + / str(workdate.year) + / f"{workdate:%Y_%m}" + / str(workdate) + / "Reports" + ) + return p - p = DAILY_DIR / str(workdate) / "Reports" / f"Asset Detail_{workdate:%d %b %Y}.csv" + +def load_val_report(workdate: datetime.date): + p = get_dir(workdate) / f"Asset Detail_{workdate:%d %b %Y}.csv" df = pd.read_csv( p, thousands=",", parse_dates=["As Of Date", "Maturity Date", "Report Run Date"] ) @@ -104,11 +117,10 @@ def load_val_report(workdate: datetime.date): def load_pnl_report(workdate: datetime.date): - + if workdate.weekday() == 0: + workdate -= datetime.timedelta(days=2) p = ( - DAILY_DIR - / str(workdate) - / "Reports" + get_dir(workdate) / f"Net Investment Earned Income by Security_{workdate:%d %b %Y}.csv" ) df = pd.read_csv(p, thousands=",", parse_dates=["Begin Date", "End Date"]) |