sqlalchemy映射器事件问题

0 投票
1 回答
1211 浏览
提问于 2025-04-17 07:49

我在sqlalchemy中定义了以下几个类:

class User(Base):
    __tablename__ = 'user'
    userId = Column(Integer, primary_key=True)
    email = Column(String(60), unique=True,nullable=False)
    userName = Column(Unicode(16),unique=True)
    _password = Column('password', Unicode(60))
    groups = relationship("Group",secondary='userGroup') 
    profile = relationship("UserProfile",uselist=False,backref='user') 

class Group(Base):
        __tablename__ = 'group'
        groupId = Column(Integer,
                       Sequence('group_id_seq', optional=True),
                       primary_key=True)
        groupName = Column(Unicode(255), unique=True)

UserGroupTable = Table('userGroup', Base.metadata,
        Column('userId', Integer, ForeignKey(User.userId)),
        Column('groupId', Integer, ForeignKey(Group.groupId))
    )


class UserProfile(Base):
    __tablename__='userProfile'
    userId = Column(Integer, ForeignKey('user.userId'), primary_key=True)
    dateJoin = Column(Date,nullable=False,default=func.current_date())
    gender = Column(Integer)

这里的usergroup之间是多对多的关系。当我初始化这些表的时候,我添加了两个组,分别是'user'和'admin',如下所示。问题是,当我使用映射器事件来把一个用户添加到'user'组时,这个操作没有成功:

def before_insert_user_listener(mapper,connection,target):
    session=DBSession()
    ug=session.query(Group).filter(Group.groupName=='user').first()
    target.groups.append(ug)    
    target.profile=UserProfile()
    log.debug(">>>>>")

def initialize_sql(engine):
    DBSession.configure(bind=engine)
    Base.metadata.bind = engine
    Base.metadata.create_all(engine)
    try:
        session = DBSession()
        #event register
        event.listen(User,'before_insert',before_insert_user_listener)
        group1 = Group('admin')
        group2 = Group('user') 
        session.add(group1)
        session.add(group2)           
        transaction.commit()
    except IntegrityError:
        pass

当我尝试添加一个新用户时,用户确实被添加了,我也能在事件监听器中看到日志打印出来,但这个用户并没有被加入到'user'组里。我还尝试了'after_insert'事件,但同样没有效果。

补充说明:我添加了一个userProfile表,这个表和用户是一对一的关系,并且可以在上面的before_insert事件中创建一个带有默认加入日期的用户档案。不过,把用户添加到组里还是不成功。

我漏掉了什么吗?

谢谢!

1 个回答

1

根据before_insert文档的内容:

在这个方法中,可以修改基于列的属性,这样就会插入新的值。但是不能对整体的刷新计划进行更改,也就是说,直接操作会话(Session)不会产生预期的效果。如果想在扩展中操作会话,应该使用SessionExtension。

由此我得出的结论是,不能在正在提交或刷新会话时添加新对象,或者添加/设置关系。

我觉得更好的方法是监听before_commitbefore_flush这两个会话事件,检查所有新的User对象,并将它们添加到User组中。例如:

def my_before_commit(session):
    g=session.query(Group).filter(Group.groupName=='user').one()
    for target in session:
        if isinstance(target, User) and target in session.new:
            target.groups.append(ug)

撰写回答