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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
from dataclasses import dataclass
from typing import Literal
import datetime
from serenitas.ops.trade_dataclasses import Deal, Ccy
from typing import ClassVar
from .custodians import NT, BNY, UMB
from .misc import get_dir, dt_from_fname
from dataclasses import field
from csv import DictReader
from functools import partial
_nt_to_currency = {"EURO - EUR": "EUR", "U.S. DOLLARS - USD": "USD"}
CUSTODIAN = Literal["UMB", "NT", "BNY"]
@dataclass
class Wire(Deal, table_name="custodian_wires", deal_type="custodian_wires"):
date: datetime.date
fund: ClassVar[CUSTODIAN]
custodian: ClassVar[str]
entry_date: datetime.date
value_date: datetime.date
pay_date: datetime.date
currency: Ccy
amount: float
wire_details: str
unique_ref: str
dtkey: ClassVar = field(metadata={"insert": False, "select": False})
def __init_subclass__(cls, fund, custodian, dtkey, **kwargs):
cls._sql_insert = (
cls._sql_insert.removesuffix("RETURNING *")
+ "ON CONFLICT (unique_ref) DO NOTHING RETURNING *"
)
cls.fund = fund
cls.custodian = custodian
cls._registry[
(
fund,
custodian,
)
] = cls
cls.dtkey = dtkey
def __post_init__(self):
self.amount = self.amount.replace(",", "")
if "(" in self.amount:
self.amount = -float(self.amount[1:-1])
else:
self.amount = float(self.amount)
@classmethod
def get_newest_report(cls, fname, date):
cls.download_reports(date)
p = max(
[f for f in get_dir(date).iterdir() if fname in f.name],
key=partial(dt_from_fname, dt_format=cls.dtkey),
default=None,
)
if not p:
raise ValueError(f"No reports for {cls.fund} on {date}")
return p
class BowdstBNYWire(Wire, BNY, fund="BOWDST", custodian="BNY", dtkey="%Y%m%d%H%M%S"):
@classmethod
def from_report_line(cls, line: dict):
return cls(
date=line["Report Run Date"],
entry_date=line["Cash Entry Date"],
value_date=line["Cash Value Date"],
pay_date=line["Settle / Pay Date"],
currency=line["Local Currency Code"],
amount=line["Local Amount"],
wire_details=line["Transaction Description 1"]
if line["Transaction Type Code"] == "CW"
else line["Transaction Description 2"],
unique_ref=line["Reference Number"],
)
@classmethod
def to_db(cls, date):
p = cls.get_newest_report("BowdstWires", date)
with open(p) as fh:
reader = DictReader(fh)
for line in reader:
cls.from_report_line(line).stage()
cls.commit()
class SeleneNTWire(Wire, NT, fund="ISOSEL", custodian="NT", dtkey="%Y%m%d%H%M"):
@classmethod
def from_report_line(cls, line: dict):
return cls(
date=line["Through date"],
entry_date=line["D-GL-POST"],
value_date=line["D-TRAN-EFF"],
pay_date=line["D-TRAN-EFF"],
currency=_nt_to_currency[line["N-GL-AC30"]],
amount=line["Net amount - local"],
wire_details=line["narrative"],
unique_ref=line["C-EXTL-SYS-TRN-DSC-3"],
)
@classmethod
def to_db(cls, date):
p = cls.get_newest_report("custodian_wires", date)
with open(p) as fh:
reader = DictReader(fh)
for line in reader:
if "sponsor" in line["narrative"].lower():
cls.from_report_line(line).stage()
cls.commit()
class SerenitasUMBWire(
Wire, UMB, fund="SERCGMAST", custodian="UMB", dtkey="%Y%m%d%H%M"
):
@classmethod
def from_report_line(cls, line: dict):
return
@classmethod
def to_db(cls, date):
# conn = cls._conn
# with conn.cursor() as c:
# c.execute("DELETE FROM custodian_wires WHERE date=%s AND fund=%s AND custodian=%s", (date, cls.fund, cls.custodian,))
# conn.commit()
# the unique ref will be generated from the row + the date and reports will only be delivered at 7pm daily
pass
|