1 files changed, 48 insertions, 0 deletions
diff --git a/python/admin_bond_upload.py b/python/admin_bond_upload.py
new file mode 100644
index 00000000..240fd451
--- /dev/null
+++ b/python/admin_bond_upload.py
@@ -0,0 +1,48 @@
+from itertools import groupby
+import datetime
+import argparse
+
+from serenitas.utils.db2 import dbconn
+from serenitas.ops.trade_dataclasses import BondDeal
+
+from report_ops.services import get_service
+
+conn = dbconn("dawndb")
+
+
+def upload_to_custodian(fund, trade_date):
+    with conn.cursor() as c:
+        c.execute(
+            "SELECT bt.*, accounts.counterparty AS account_counterparty FROM bond_trades bt LEFT JOIN accounts ON bt.account=accounts.code WHERE bt.fund=%s AND bt.trade_date=%s;",
+            (
+                fund,
+                trade_date,
+            ),
+        )
+        trades = [t._asdict() for t in c]
+        for account_counterparty, trades in groupby(
+            trades, lambda x: x["account_counterparty"]
+        ):
+            if account_counterparty == "NT":
+                continue
+            service = get_service(account_counterparty)
+            for t in trades:
+                trade = BondDeal.from_dict(**t, scaled=True)
+                service.push_trade(trade, "NEW")
+            service.build_buffer("bond")
+
+
+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 to CTM"
+    )
+    args = parser.parse_args()
+    for fund in ("SERCGMAST", "BOWDST", "BRINKER", "ISOSEL"):
+        upload_to_custodian(fund, args.trade_date)