Question

这是我的Flask-SQLAlchemy声明代码：

from sqlalchemy.ext.associationproxy import association_proxy
from my_flask_project import db


tagging = db.Table('tagging',
    db.Column('tag_id', db.Integer, db.ForeignKey('tag.id', ondelete='cascade'),
              primary_key=True),
    db.Column('role_id', db.Integer, db.ForeignKey('role.id', ondelete='cascade'),
              primary_key=True)
)


class Tag(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), unique=True, nullable=False)

    def __init__(self, name=None):
        self.name = name


class Role(db.Model):

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id', ondelete='cascade'))
    user = db.relationship('User', backref=db.backref('roles', cascade='all',
                           lazy='dynamic'))
    ...
    tags = db.relationship('Tag', secondary=tagging, cascade='all',
                           backref=db.backref('roles', cascade='all'))
    tag_names = association_proxy('tags', 'name')

    __table_args__ = (
        db.UniqueConstraint('user_id', 'check_id'),
    )

我认为这是非常标准的多对多标记解决方案。现在，我想获取角色的所有标签，并为角色设置新的标签集。

第一个很简单：

print role.tags
print role.tag_names

然而，第二个让我整天绊倒我的Python代码:-(我以为我可以这样做：

role.tag_names[:] = ['red', 'blue', 'white']

...或至少使用role.tags[:] = ...的类似内容，但我发明的所有内容都引发了许多完整性错误，因为SQLAlchemy没有检查是否存在任何现有标记并试图将所有标记作为全新实体插入

我的最终解决方案是：

# cleanup input
tag_names = set(filter(None, tag_names))

# existings tags to be updated
to_update = [t for t in role.tags if t.name in tag_names]

# existing tags to be added
to_add = list(
    Tag.query.filter(Tag.name.in_(tag_names - set(role.tag_names)))
)

# tags to be created
existing_tags = to_update + to_add
to_create = [Tag(name) for name in tag_names - set([t.name for t in existing_tags])]

# assign new tags
role.tags[:] = existing_tags + to_create

# omitted bonus: find a way how to get rid of orphan tags

问题是：这真的是正确的解决方案吗？有没有更优雅的方法如何解决这个琐碎的问题？我认为整件事与this question有关。也许我只是愚蠢，也许我让事情变得过于复杂......无论如何，谢谢你的任何建议！

Answer 1

实际上，SQLAlchemy通过调用Session.merge()来检查对象是否存在。但它是通过身份 - 它的主要关键来实现的。最简单的解决方案是生成name主键，一切都会正常工作。当然，在这种情况下，三个表链将变得多余，除非您要在Tag中添加一些额外的字段（例如计数器）。

在SQLAlchemy中通过association_proxy分配一组新的多对多相关标签的最优雅方法是什么？

1 个答案: