sqlalchemy映射器事件问题
我在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)
这里的user
和group
之间是多对多的关系。当我初始化这些表的时候,我添加了两个组,分别是'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_commit或before_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)