aboutsummaryrefslogtreecommitdiffstats
path: root/python/admin_cls_upload.py
blob: c290bb6abf6b5c8cd9653e3c6442c084ec71f8e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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, 6, 8)
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")
        buf, dest = service.build_buffer("spot")
        service.upload(
            buf,
            dest.name,
            fund,
            "CLS",
        )