aboutsummaryrefslogtreecommitdiffstats
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/report_ops/cash.py4
-rw-r--r--python/report_ops/misc.py3
-rw-r--r--python/report_ops/wires.py6
3 files changed, 7 insertions, 6 deletions
diff --git a/python/report_ops/cash.py b/python/report_ops/cash.py
index 94837697..e1fd8319 100644
--- a/python/report_ops/cash.py
+++ b/python/report_ops/cash.py
@@ -8,13 +8,13 @@ from serenitas.utils.env import DAILY_DIR
from serenitas.utils.db import dbconn
from serenitas.analytics.dates import prev_business_day
-from .misc import get_dir, dt_from_fname
+from .misc import get_dir, dt_from_fname, CUSTODIAN
from .custodians import get_custodian_download_fun
@dataclass
class CashReport:
- custodian: ClassVar[str]
+ custodian: CUSTODIAN
date: datetime.date
dtkey: ClassVar
_conn: ClassVar[dbconn] = dbconn("dawndb")
diff --git a/python/report_ops/misc.py b/python/report_ops/misc.py
index 0ad79388..b49d62f9 100644
--- a/python/report_ops/misc.py
+++ b/python/report_ops/misc.py
@@ -110,6 +110,9 @@ _fund_custodians = {
}
+CUSTODIAN = Literal["UMB", "NT", "BNY", "SCOTIA"]
+
+
def get_dir(
workdate: datetime.date = datetime.date.today(), archived=True
) -> pathlib.Path:
diff --git a/python/report_ops/wires.py b/python/report_ops/wires.py
index 3709b0d5..cc290420 100644
--- a/python/report_ops/wires.py
+++ b/python/report_ops/wires.py
@@ -10,18 +10,16 @@ from serenitas.ops.dataclass_mapping import Fund
from serenitas.analytics.dates import prev_business_day
from serenitas.utils.env import DAILY_DIR
-from .misc import get_dir, dt_from_fname
+from .misc import get_dir, dt_from_fname, CUSTODIAN
_nt_to_currency = {"EURO - EUR": "EUR", "U.S. DOLLARS - USD": "USD"}
-CUSTODIAN = Literal["UMB", "NT", "BNY"]
-
@dataclass
class WireReport(Deal, table_name="custodian_wires", deal_type="custodian_wires"):
date: datetime.date
fund: Fund
- custodian: ClassVar[str]
+ custodian: CUSTODIAN
entry_date: datetime.date
value_date: datetime.date
pay_date: datetime.date