5

问题是我不知道如何使用 SQLAlchemy 在对象进入新状态时通知我。

我正在使用 SQLAlchemy ORM (Declarative) 来更新一个对象：

class Customer(declarative_base()):

    __table_name__ = "customer"

    id = Column(Integer, primary_key=True)
    status = Column(String)

我想知道对象何时进入状态。特别是在发布 UPDATE 和状态更改后。例如Customer.status == 'registered'，它以前有不同的状态。

我目前正在使用'set'属性事件执行此操作：

from sqlalchemy import event
from model import Customer

def on_set_attribute(target, value, oldvalue, initiator):
    print target.status
    print value
    print oldvalue

event.listen(
        Customer.status,
        'set',
        on_set_attribute,
        propagate=True,
        active_history=True)

每次在该属性上调用“set”时，我的代码都会触发，并且我会检查 thevalue和 theoldvalue是否不同。问题是target参数没有完全形成，所以它还没有填充所有的属性值。

有一个更好的方法吗？谢谢！

4

2 回答 2

10

我的解决方案是使用“after_flush”SessionEvent 而不是“set”AttributeEvent。

非常感谢agronholm，他提供了专门检查对象值和旧值的示例 SessionEvent 代码。

下面的解决方案是修改他的代码：

def get_old_value(attribute_state):
    history = attribute_state.history
    return history.deleted[0] if history.deleted else None


def trigger_attribute_change_events(object_):
    for mapper_property in object_mapper(object_).iterate_properties:
        if isinstance(mapper_property, ColumnProperty):
            key = mapper_property.key
            attribute_state = inspect(object_).attrs.get(key)
            history = attribute_state.history

            if history.has_changes():
                value = attribute_state.value
                # old_value is None for new objects and old value for dirty objects
                old_value = get_old_value(attribute_state)
                handler = registry.get(mapper_property)
                if handler:
                    handler(object_, value, old_value)


def on_after_flush(session, flush_context):
    changed_objects = session.new.union(session.dirty)
    for o in changed_objects:
        trigger_attribute_change_events(o)

event.listen(session, "after_flush", on_after_flush)

这registry是一个字典，其键是 MapperProperty，其值是事件处理程序。 session、event、inspect和object_mapper都是 sqlalchemy 类和函数。

于 2013-03-27T23:37:26.893 回答
2

使用before_update事件或before_flush事件在稍后的时间点拦截发生的这种情况。

于 2013-03-27T16:54:53.020 回答