diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/upload_umb_bonds.py | 33 |
1 files changed, 23 insertions, 10 deletions
diff --git a/python/upload_umb_bonds.py b/python/upload_umb_bonds.py index d1be36b6..d6355559 100644 --- a/python/upload_umb_bonds.py +++ b/python/upload_umb_bonds.py @@ -3,15 +3,8 @@ import datetime from serenitas.ops.trade_dataclasses import BondDeal from serenitas.ops.funds import UMB -if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument( - "trade_date", - type=datetime.date.fromisoformat, - default=datetime.date.today(), - nargs="?", - ) - args = parser.parse_args() + +def upload_to_umb(trade_date, upload): sql_query = ( "SELECT bonds.*, bond_allocation.notional, fund, custodian, " "cash_account, bond_submission.id as sid, allocation_id, action, " @@ -35,4 +28,24 @@ if __name__ == "__main__": for row in c: UMB.stage(row._asdict(), trade_type="bond", redis_pipeline=None) buf, dest = UMB.build_buffer("bond") - UMB.upload(buf, dest) + if upload: + UMB.upload(buf, dest) + + +if __name__ == "__main__": + parser = argparse.ArgumentParser() + parser.add_argument( + "trade_date", + type=datetime.date.fromisoformat, + default=datetime.date.today(), + nargs="?", + ) + parser.add_argument( + "-n", + "--no-upload", + action="store_true", + help="do not upload just drop in the daily directory", + default=False, + ) + args = parser.parse_args() + upload_to_umb(args.trade_date, not args.no_upload) |