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)