aboutsummaryrefslogtreecommitdiffstats
path: root/python/ctm_allocation.py
blob: 8a4d6581f65f33082b383d1b573406a67a9e2773 (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
38
from serenitas.utils.db import dawn_engine, dbconn
from serenitas.utils.exchange import ExchangeMessage, FileAttachment
from exchangelib import HTMLBody
import pandas as pd
from io import StringIO
import datetime
import argparse
from pandas.tseries.offsets import BDay
from serenitas.ops.trade_dataclasses import BondDeal
from serenitas.ops.funds import CTM

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "trade_date",
        type=datetime.date.fromisoformat,
        default=datetime.date.today(),
    )
    args = parser.parse_args()
    dawndb = dbconn("dawndb")

    alert = {
        "SERCGMAST": "Serenitas Credit Gamma Master Fund",
        "BOWDST": "Bowdoin St. (ALERT acronym: LMUNDER, Account: SER003)",
        "BRINKER": "Brinker Destinations Multi-Strategy Alternatives - LMCG",
        "ISOSEL": "ISO Selene (ALERT acronym: LMUNDER, Account: SER004)",
    }

    df = pd.read_sql_query(
        "SELECT ba.id, allocated, emailed FROM bonds LEFT JOIN bond_allocation ba ON bonds.id=ba.tradeid LEFT JOIN counterparties c ON cp_code=c.code WHERE trade_date=%s AND ctm_code IS NOT null;",
        con=dawn_engine,
        params=(args.trade_date,),
    )
    with dawndb.cursor() as d:
        for row in df.itertuples():
            CTM.stage(BondDeal.from_allocationid(row.id))
    buf, dest = CTM.build_buffer()
    CTM.upload(buf)