1 files changed, 26 insertions, 0 deletions
diff --git a/python/upload_umb_bonds.py b/python/upload_umb_bonds.py
new file mode 100644
index 00000000..e53d0071
--- /dev/null
+++ b/python/upload_umb_bonds.py
@@ -0,0 +1,26 @@
+import argparse
+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(),
+    )
+    args = parser.parse_args()
+    with BondDeal._conn.cursor() as c:
+        c.execute(
+            "SELECT id FROM bond_trades WHERE fund= %s " "AND trade_date = %s ",
+            (
+                "SERCGMAST",
+                args.trade_date,
+            ),
+        )
+        for (tid,) in c:
+            trade = BondDeal.from_allocationid(tid)
+            UMB.staging_queue.append(next(trade.to_umb("NEW")))
+    buf, dest = UMB.build_buffer("bond")
+    UMB.upload(buf, dest)