Sqlalchemy - 根据另一列的更改更新列

Posted 2021-04-25

我正在使用sqlalchemy，但发现难以搜索的文档。

我有这两列：

    verified = Column(Boolean, default=False)
    verified_at = Column(DateTime, nullable=True)

我想创建一个像这样的函数：

    if self.verified and not oldobj.verified:
        self.verified_at = datetime.datetime.utcnow
    if not self.verified and oldobj.verified:
        self.verified_at = None

我不知道在哪里放这样的代码。我可以把它放在应用程序中，但更喜欢model对象处理这个逻辑。

答案

我认为你所寻找的是一个Hybrid Property。

from sqlalchemy.ext.hybrid import hybrid_property

class VerifiedAsset(Base):
    id = Column(Integer, primary_key=True)
    verified_at = Column('verified_at', String(24))

    @hybrid_property
    def verification(self):
        return self.verified_at;

    @verification.setter
    def verification(self, value):
        if value and not self.verification:
            self.verified_at = datetime.datetime.utcnow
        if not value and self.verification:
            self.verified_at = None
        # Presumably you want to handle your other cases here

您希望根据某些传入的新值以特定方式更新verified_at值。使用属性来包装基础值，并且仅在适当时更新，并且仅对您在db中实际持久存储的内容进行更新。

另一答案

您可以使用sqlalchemy的事件注册来放置这样的代码：http://docs.sqlalchemy.org/en/latest/core/event.html。基本上，您可以订阅Core和ORM中发生的某些事件。我认为这是一种管理你想要实现的目标的简洁方法。

您将使用listen_for()装饰器，以便在这些列更改时挂钩。

另一答案

阅读"Changing Attribute Behavior"和"ORM Events"是尝试解决此类问题的良好开端。

一种方法是设置一个更新时间戳的事件监听器：

@event.listens_for(MyModel.verified, 'set')
def mymodel_verified_set(target, value, oldvalue, initiator):
    """Set verified_at"""
    if value != oldvalue:
        target.verified_at = datetime.datetime.utcnow() if value else None