如何使用 SQLAlchemy 将默认/预先存在的行作为准备使用的对象？

Question

我正在开发一个使用 SQLALchemy 来保存数据的系统。 我想拥有我的默认行的对象，以便在创建新实例时可以使用。

这对我来说非常好，因为有些表应该有一些默认值。 想象一下 trow 表（order 和 order_status）之间的简单关系。 按顺序插入值时，最好有一种简单的方法来使用 order_status 中已有的一些默认值，而不必每次都查询它们。

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, sessionmaker


engine = create_engine('sqlite:///:memory:')

Base = declarative_base()

class Order(Base):
    __tablename__ = 'order'

    id = Column(Integer, primary_key=True)
    order_status_id = Column(Integer, ForeignKey('order_status.id'))
    value = Column(Integer)

    order_status = relationship('OrderStatus', lazy=True, uselist=False)

class OrderStatus(Base):
    __tablename__ = 'order_status'

    id = Column(Integer, primary_key=True)
    name = Column(String(32))

Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

finished = OrderStatus(name='Finished')
session.add(finished)
session.commit()

例如，我想使用以下内容创建 Order 的新实例：

order = Order(value=10, order_status=OrderStatus.FINISHED)

代替：

finished_status = session.query(OrderStatus).filter_by(name='Finished').first()
order = Order(value=10, order_status=finished_status)

它不需要完全像我的例子一样实现，但我想要类似的东西

Answer 1

你应该看看association_proxy ： https : //docs.sqlalchemy.org/en/13/orm/extensions/associationproxy.html

该示例非常适合您的问题。 您的Order最终可能看起来像这样（显然不能复制和粘贴）：

class OrderStatus(Base):
    ...

class Order(Base):
    __tablename__ = 'order'

    id = Column(Integer, primary_key=True)
    order_status_id = Column(Integer, ForeignKey('order_status.id'))
    value = Column(Integer)

    _order_status = relationship('OrderStatus', lazy=True, uselist=False)
    order_status = association_proxy('_order_status', 'name', creator=get_or_create_order_status)

def get_or_create_order_status(name):
    # get or create a order status by name

在get_or_create_order_status您可能必须返回查询（至少最初是为了例如填充缓存）。

如果OrderStatus实际上和您的问题一样简单，也许enum也可以解决问题。 这显然会更容易管理。