使用sqlalchemy关联代理正确级联删除



我在sqlalchemy中有一个自引用关系,它在很大程度上基于这个答案中的示例。

我有一个用户表,以及一个将主用户链接到辅助用户的关联表。用户A可以是用户B的主要用户,B也可以不是用户A的主要用户。它的工作原理与我上面链接的答案中的twitter类比完全相同。

这很好,只是我不知道如何为关联代理建立级联规则。目前,如果我删除了一个用户,关联记录会保留下来,但它会为被删除的用户清空任何FK。我希望删除级联到关联表并删除记录。

我还需要能够解除用户的关联,这只会删除关联记录,但会传播到用户的"is_primary_of"one_answers"is_secondary_of"关联代理。

有人能帮我找出如何将这些行为整合到我的模型中吗?代码如下。谢谢

import sqlalchemy
import sqlalchemy.orm
import sqlalchemy.ext.declarative
import sqlalchemy.ext.associationproxy

# This is the base class from which all sqlalchemy table objects must inherit
SAModelBase = sqlalchemy.ext.declarative.declarative_base()

class UserAssociation(SAModelBase):
__tablename__ = 'user_associations'
# Columns
id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True)
# Foreign key columns
primary_user_id = sqlalchemy.Column(sqlalchemy.Integer,
sqlalchemy.ForeignKey('users.id', name='user_association_primary_user_fk'))
secondary_user_id = sqlalchemy.Column(sqlalchemy.Integer,
sqlalchemy.ForeignKey('users.id', name='user_association_secondary_user_fk'))
# Foreign key relationships
primary_user = sqlalchemy.orm.relationship('User',
foreign_keys=primary_user_id,
backref='secondary_users')
secondary_user = sqlalchemy.orm.relationship('User',
foreign_keys=secondary_user_id,
backref='primary_users')
def __init__(self, primary, secondary, **kwargs):
self.primary_user = primary
self.secondary_user = secondary
for kw,arg in kwargs.items():
setattr(self, kw, arg)

class User(SAModelBase):
__tablename__ = 'users'
# Columns
id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True)
first_name = sqlalchemy.Column(sqlalchemy.String)
last_name = sqlalchemy.Column(sqlalchemy.String)
is_primary_of = sqlalchemy.ext.associationproxy.association_proxy('secondary_users', 'secondary_user')
is_secondary_of = sqlalchemy.ext.associationproxy.association_proxy('primary_users', 'primary_user')
def associate(self, user, **kwargs):
UserAssociation(primary=self, secondary=user, **kwargs)

结果非常简单。原始代码中的backref只是字符串,但它们可以是backref对象。这允许您设置级联行为。请参阅有关backref参数的sqlalchemy文档。

这里只需要在UserAssociation对象中进行更改。现在显示:

# Foreign key columns
primary_user_id = sqlalchemy.Column(sqlalchemy.Integer,
sqlalchemy.ForeignKey('users.id',
name='user_association_primary_user_fk'),
nullable=False)
secondary_user_id = sqlalchemy.Column(sqlalchemy.Integer,
sqlalchemy.ForeignKey('users.id',
name='user_association_associated_user_fk'),
nullable=False)
# Foreign key relationships
primary_user = sqlalchemy.orm.relationship('User',
foreign_keys=primary_user_id,
backref=sqlalchemy.orm.backref('secondary_users',
    cascade='all, delete-orphan'))
secondary_user = sqlalchemy.orm.relationship('User',
foreign_keys=secondary_user_id,
backref=sqlalchemy.orm.backref('primary_users',
      cascade='all, delete-orphan'))

backref关键字参数现在是backref对象,而不是字符串。我还能够使外键列不可为null,因为它现在级联已删除的用户,这样关联也会被删除。

最新更新