sqlalchemy 列更新事件

Question

我在更新 User.name 时注册了一个事件，以便在更新之前或之后应用一些规则。

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy import event
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
engine = create_engine('sqlite:///:memory:', echo=True)
Session = sessionmaker(bind=engine)   # engine part

Base = declarative_base()


class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    fullname = Column(String)
    password = Column(String)
    rule = Column(String)
Base.metadata.create_all(engine)


session = Session()
u1 = User(id=1, name="hello world", fullname="hello kitty")
u2 = User(id=2, name="hello world2", fullname="hello kitty2")
session.add(u1)
session.add(u2)


@event.listens_for(User, 'before_update')
def User_before_update(target, value, initiator):
    print ":::::received before insert event for target"


@event.listens_for(User.name, 'set')
def name_set(target, value, old_value, initiator):
    print ":::::set before insert event for target"


### option A
user = session.query(User).get(2)
user.name = u"wawamsma"
session.merge(user)

### option B
session.query(User).filter(User.id == 2).update({User.name: u"eenimenee"})

选项 A

一切都运行得很好，但我有一些更新方法需要使用很多字段来更新数据库。

选项 B

虽然更新了，但没有触发两个打印函数。

所以我在想，是注册事件的方式不对，还是更新的方式不对呢？

Answer 1

当你使用query.update()方法时，如果想要接收事件，你需要用到after_bulk_update这个事件。因为这个操作不是针对单个对象，而是直接发出一个UPDATE语句，所以在内存中可能受到影响的实际对象并不在你手边；你可能需要重新查询一下这些对象。如果你把"synchronize_session"设置为"evaluate"，那么上下文中会有一个matched_objects属性。