diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/external_deriv_marks.py | 5 | ||||
| -rw-r--r-- | python/trade_dataclasses.py | 12 |
2 files changed, 12 insertions, 5 deletions
diff --git a/python/external_deriv_marks.py b/python/external_deriv_marks.py index e6d2584d..aa012958 100644 --- a/python/external_deriv_marks.py +++ b/python/external_deriv_marks.py @@ -338,7 +338,7 @@ def get_box_dimension(l): if __name__ == "__main__": import argparse import logging - from serenitas.utils.db import dbconn + from serenitas.utils.db2 import dawn_pool parser = argparse.ArgumentParser() parser.add_argument( @@ -370,7 +370,7 @@ if __name__ == "__main__": if df.empty and cp == "CS": df = globals()[f"{cp.lower()}_navs_old"](date, fund=fund) logger.debug(df) - with dbconn("dawndb") as conn: + with dawn_pool.connection() as conn: with conn.cursor() as c: for k, v in df[["local_nav", "base_nav", "ia"]].iterrows(): c.execute( @@ -387,3 +387,4 @@ if __name__ == "__main__": float(v.ia), ), ) + conn.commit() diff --git a/python/trade_dataclasses.py b/python/trade_dataclasses.py index b9201386..3f66bd98 100644 --- a/python/trade_dataclasses.py +++ b/python/trade_dataclasses.py @@ -5,12 +5,13 @@ from typing import Literal import csv import datetime from enum import Enum -from psycopg2.extensions import register_adapter, AsIs +from psycopg.adapt import Dumper +from psycopg import adapters from serenitas.analytics.dates import next_business_day, previous_twentieth from serenitas.analytics.index import CreditIndex from serenitas.utils.db2 import dbconn from lru import LRU -from psycopg2.errors import UniqueViolation +from psycopg.errors import UniqueViolation import logging logger = logging.getLogger(__name__) @@ -161,7 +162,12 @@ class Counterparty: name: str -register_adapter(Frequency, lambda f: AsIs(f.value)) +class FrequencyDumper(Dumper): + def dump(self, f): + return f.value + + +adapters.register_dumper(Frequency, FrequencyDumper) class DealType(Enum): |