2

这是我的 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

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

4

1 回答 1

1

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

于 2012-02-15T10:20:35.937 回答