aboutsummaryrefslogtreecommitdiffstats
path: root/python/report_ops/wires.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/report_ops/wires.py')
-rw-r--r--python/report_ops/wires.py42
1 files changed, 2 insertions, 40 deletions
diff --git a/python/report_ops/wires.py b/python/report_ops/wires.py
index baaae19d..6da3ac29 100644
--- a/python/report_ops/wires.py
+++ b/python/report_ops/wires.py
@@ -13,46 +13,7 @@ from serenitas.utils.env import DAILY_DIR
from serenitas.utils.db2 import dbconn
from .misc import get_dir, dt_from_fname, Custodian
-
-
-class Report:
- _sql_insert: ClassVar[str]
- _insert_queue: ClassVar[list]
- _insert_fields: ClassVar[list]
- _conn: ClassVar = dbconn("dawndb")
- _registry = {}
-
- def __init_subclass__(cls, table_name: str):
- cls._insert_fields = list(filter(cls.is_insert_field, cls.__annotations__))
- place_holders = ",".join(["%s"] * len(cls._insert_fields))
- cls._sql_insert = f"INSERT INTO {table_name}({','.join(cls._insert_fields)}) VALUES({place_holders}) ON CONFLICT DO NOTHING RETURNING *"
- cls._insert_queue = []
-
- @classmethod
- def is_insert_field(cls, attr):
- """fields are insertable by default unless specified otherwise"""
- match getattr(cls, attr, None):
- case Field(metadata={"insert": False}) | Field(init=False):
- return False
- case _:
- return True
-
- @classmethod
- def clear(cls):
- cls._insert_queue.clear()
-
- def stage(self):
- self._insert_queue.append(
- tuple([getattr(self, col) for col in self._insert_fields])
- )
-
- @classmethod
- def commit(cls):
- conn = cls._conn
- with conn.cursor() as c:
- c.executemany(cls._sql_insert, cls._insert_queue)
- conn.commit()
- cls._insert_queue.clear()
+from .base import Report
@dataclass
@@ -68,6 +29,7 @@ class WireReport(Report, table_name="custodian_wires"):
wire_details: str
unique_ref: str
dtkey: ClassVar = field(metadata={"insert": False})
+ _registry: ClassVar = field(default={}, metadata={"insert": False})
def __init_subclass__(cls, custodian, dtkey):
cls.custodian = custodian