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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
from serenitas.utils.db import dbconn
from serenitas.ops.funds import Bowdst
from serenitas.ops.trade_dataclasses import WireDeal
import datetime
from serenitas.analytics.dates import prev_business_day
from collateral.common import CASH_STRATEGY_MAPPING, STRATEGY_CASH_MAPPING
from serenitas.utils.misc import rename_keys
def generate_csv(date, conn, fund="BOWDST"):
wire_deals = []
with conn.cursor() as c:
c.execute(
"SELECT count(*) FROM wires WHERE trade_date=%s AND fund=%s AND author='auto';",
(
date,
fund,
),
)
(count,) = c.fetchone()
if count > 0:
return
c.execute(
"SELECT * FROM list_orphaned_cash(%s, %s) ",
(
date,
fund,
),
)
for row in c:
if row.strat not in CASH_STRATEGY_MAPPING:
obj = WireDeal(**row._asdict(), action="NEW")
offset = obj.offset(STRATEGY_CASH_MAPPING[obj.folder])
obj.stage()
offset.stage()
wire_deals.extend([obj, offset])
if wire_deals:
WireDeal.commit()
for wire in wire_deals:
wire.set_id()
Bowdst.staging_queue.append(wire.to_globeop("NEW"))
buf, dest = Bowdst.build_buffer("wire")
Bowdst.upload(buf, dest.name)
Bowdst().clear()
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser()
parser.add_argument(
"workdate",
nargs="?",
type=datetime.date.fromisoformat,
default=prev_business_day(datetime.date.today()),
help="working date",
)
args = parser.parse_args()
conn = dbconn("dawndb")
generate_csv(args.workdate, conn)
|
