diff options
Diffstat (limited to 'python/bowdst.py')
| -rw-r--r-- | python/bowdst.py | 26 |
1 files changed, 9 insertions, 17 deletions
diff --git a/python/bowdst.py b/python/bowdst.py index 389e1146..16f42faa 100644 --- a/python/bowdst.py +++ b/python/bowdst.py @@ -4,7 +4,6 @@ from dates import bus_day from env import DAILY_DIR from exchangelib import FileAttachment from io import StringIO -from remote import SftpClient from typing import Tuple from utils.db import dbconn, dawn_engine @@ -32,7 +31,7 @@ def download_messages(em): p.write_bytes(attach.content) -def load_report(workdate: datetime.date): +def load_val_report(workdate: datetime.date): p = DAILY_DIR / str(workdate) / "Reports" / f"Asset Detail_{workdate:%d %b %Y}.csv" df = pd.read_csv( @@ -266,24 +265,17 @@ if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument( - "cob", + "workdate", nargs="?", type=datetime.date.fromisoformat, - default=(datetime.date.today() - bus_day).date(), - help="close of business date", + default=datetime.date.today(), + help="working date", ) args = parser.parse_args() em = ExchangeMessage() download_messages(em) - df_bonds, df_cds, df_tranches = get_positions(args.cob) - send_email(em, args.cob, df_bonds, df_cds, df_tranches) - # sftp = SftpClient.from_creds("gs") - # for f in sftp.client.listdir("outgoing"): - # rename_file = f.startswith("Account") - # local_file = DAILY_DIR / "BowdSt" / "GS_fcm_reports" / f - # if not local_file.exists(): - # sftp.client.get(f"outgoing/{f}", local_file) - # if local_file.startswith("Account"): - # df = pd.read_csv(local_file) - # cob = df["COB Date"][0] - # local_file.rename(f.split(".")[0] + cob[4:] + ".csv") + cob = args.workdate - bus_day + df_bonds, df_cds, df_tranches = get_positions(cob) + send_email(em, cob, df_bonds, df_cds, df_tranches) + load_val_report(args.workdate) + load_pnl_report(args.workdate) |