diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/report_ops/cash.py | 6 | ||||
| -rw-r--r-- | python/report_ops/wires.py | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/python/report_ops/cash.py b/python/report_ops/cash.py index 84933c82..3d4157a1 100644 --- a/python/report_ops/cash.py +++ b/python/report_ops/cash.py @@ -5,7 +5,7 @@ from dataclasses import dataclass import pandas as pd from serenitas.utils.env import DAILY_DIR -from serenitas.utils.db import dbconn +from serenitas.utils.db2 import dbconn from serenitas.analytics.dates import prev_business_day from .misc import get_dir, dt_from_fname, Custodian @@ -13,10 +13,10 @@ from .misc import get_dir, dt_from_fname, Custodian @dataclass class CashReport: - custodian: Custodian + custodian: ClassVar[Custodian] date: datetime.date dtkey: ClassVar - _conn: ClassVar[dbconn] = dbconn("dawndb") + _conn: ClassVar = dbconn("dawndb") _staging_queue: ClassVar[set] = set() _insert_sql = "INSERT INTO cash_balances VALUES (%s, %s, %s, %s, %s, %s) ON CONFLICT DO NOTHING" _registry = {} diff --git a/python/report_ops/wires.py b/python/report_ops/wires.py index 724230a2..103c985c 100644 --- a/python/report_ops/wires.py +++ b/python/report_ops/wires.py @@ -17,7 +17,7 @@ from .misc import get_dir, dt_from_fname, Custodian class WireReport(Deal, table_name="custodian_wires", deal_type="custodian_wires"): date: datetime.date fund: Fund - custodian: Custodian + custodian: ClassVar[Custodian] entry_date: datetime.date value_date: datetime.date pay_date: datetime.date |