import datetime
from itertools import groupby
from serenitas.ops.trade_dataclasses import SpotDeal

from report_ops.services import get_service

fund = "BOWDST"
trade_date = datetime.date(2023, 3, 14)
conn = SpotDeal._conn

with conn.cursor() as c:
    c.execute(
        "SELECT ft.*, accounts2.cp_code as account_counterparty FROM forward_trades ft LEFT JOIN accounts2 USING (cash_account) WHERE ft.fund=%s AND ft.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 not in ("BONY",):
            continue
        service = get_service(account_counterparty)
        for t in trades:
            t["dealid"] = (
                f"{t['dealid']}_{t['fx_type'][0]}"
                if t["fx_type"] != "SPOT"
                else t["dealid"]
            )
            trade = SpotDeal.from_dict(**t)
            service.push_trade(trade, "NEW")
        service.build_buffer("spot")