我有以下 ORM 对象(简化):
import datetime as dt
from sqlalchemy import create_engine, Integer, Column, DateTime
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Session, declarative_base
Base = declarative_base()
class Metrics(Base):
__tablename__ = 'metrics'
id = Column(Integer, primary_key=True)
ts = Column(DateTime, default=dt.datetime.now())
computed_values = Column(JSONB)
dates = Column(JSONB)
calculated_values =
entry = Metrics(computed_values={'foo': 12.3, 'bar':45.6},
dates=[datetime.date.today()])
engine = create_engine('postgresql://postgres:postgres@localhost:5432/my_schema')
with Session(engine, future=True) as session:
session.add(entry)
session.commit()
每行有:
id
首要的关键ts
插入行时的时间戳computed_values
要存储的实际 JSONB 数据dates
JSONB 用于存储计算数据的日期列表。虽然我对该computed_values
列没有任何问题,但默认 SQLAlchemy JSON 序列化程序无法datetime.date
对列内列表中的对象dates
进行序列化。
我的想法是date
为那个确切的列重新定义对象的序列化器行为。为此,我必须定义自己的自定义 JSON 序列化程序,或者使用一些现成的序列化程序,例如 orjson。由于我可能会在项目中遇到许多其他 JSON 序列化问题,因此我更喜欢后者。
深入研究JSONB
类及其超类,我认为以下应该可以解决问题:
class Metrics(Base):
__tablename__ = 'metrics'
# ---%<--- snip ---%<---
dates = Column(JSONB(json_serializer=lambda obj: orjson.dumps(obj, option=orjson.OPT_NAIVE_UTC)))
# ---%<--- snip ---%<---
但它没有:
File "metrics.py", line 30, in Metrics
dates = Column(JSONB(json_serializer=lambda obj: orjson.dumps(obj, option=orjson.OPT_NAIVE_UTC)))
TypeError: __init__() got an unexpected keyword argument 'json_serializer'
我在做什么错以及如何为 JSON(和 JSONB)列正确定义自定义 SQLAlchemy 序列化程序?
看起来你应该能够通过修改你的create_engine
语句来得到你想要的。
Custom serializers and deserializers are specified at the dialect level,
that is using :func:`_sa.create_engine`. The reason for this is that when
using psycopg2, the DBAPI only allows serializers at the per-cursor
or per-connection level. E.g.::
engine = create_engine("postgresql://scott:tiger@localhost/test",
json_serializer=my_serialize_fn,
json_deserializer=my_deserialize_fn
)
因此生成的代码应如下所示:
import datetime as dt
import orjson
from sqlalchemy import create_engine, Integer, Column, DateTime
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Session, declarative_base
Base = declarative_base()
class Metrics(Base):
__tablename__ = 'metrics'
id = Column(Integer, primary_key=True)
ts = Column(DateTime, default=dt.datetime.now())
computed_values = Column(JSONB)
dates = Column(JSONB)
calculated_values =
entry = Metrics(computed_values={'foo': 12.3, 'bar':45.6},
dates=[datetime.date.today()])
def orjson_serializer(obj):
"""
Note that `orjson.dumps()` return byte array, while sqlalchemy expects string, thus `decode()` call.
"""
return orjson.dumps(obj, option=orjson.OPT_SERIALIZE_NUMPY | orjson.OPT_NAIVE_UTC).decode()
engine = create_engine('postgresql://postgres:postgres@localhost:5432/my_schema', json_serializer=orjson_serializer, json_deserializer=orjson.loads)
with Session(engine, future=True) as session:
session.add(entry)
session.commit()
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句