diff options
Diffstat (limited to 'python/trade_dataclasses.py')
| -rw-r--r-- | python/trade_dataclasses.py | 12 |
1 files changed, 9 insertions, 3 deletions
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): |