diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/load_bbh_reports.py | 26 |
1 files changed, 19 insertions, 7 deletions
diff --git a/python/load_bbh_reports.py b/python/load_bbh_reports.py index 6162a1f4..dc2b8854 100644 --- a/python/load_bbh_reports.py +++ b/python/load_bbh_reports.py @@ -70,19 +70,31 @@ def val_report(f): def load_reports(date): path = DAILY_DIR / str(date) / "Reports" - filename = "BBH_PNL." + date.strftime("%Y%m%d") + ".csv" + filename = f"BBH_PNL.{date:%Y%m%d}.csv" pnl_report(path / filename) - filename = "BBH_VAL." + date.strftime("%Y%m%d") + ".csv" + filename = f"BBH_VAL.{date:%Y%m%d}.csv" val_report(path / filename) def download_val_report(date): - path = DAILY_DIR / str(date) / "Reports" + reports_dir = DAILY_DIR / str(date) / "Reports" sftp = SftpClient.from_creds("bbh") - sftp.download_files("frombbh", path) + for f in sftp.client.listdir("frombbh"): + if f.split(".")[1] == f"{date:%Y%m%d}": + sftp.client.get(f"frombbh/{f}", localpath=reports_dir / f) if __name__ == "__main__": - date = datetime.date.today() - download_val_report(date) - load_reports(date) + import argparse + + parser = argparse.ArgumentParser() + parser.add_argument( + "workdate", + nargs="?", + type=datetime.date.fromisoformat, + default=datetime.date.today(), + help="working date", + ) + args = parser.parse_args() + download_val_report(args.workdate) + load_reports(args.workdate) |