多对多关系：get-or-create答案

【问题标题】：Many-to-many relationship: get-or-create多对多关系：get-or-create
【发布时间】：2016-11-13 01:42:42
【问题描述】：

我正在为博客开发一个标签系统。这是创建 Flask 应用程序对象和相关 Post 和 Tag 模型的代码的精简版本。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.ext.associationproxy import association_proxy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.sqlite'
db = SQLAlchemy(app)

post_tags = db.Table('post_tags',
                     db.Column('post_id', db.Integer,
                               db.ForeignKey('posts.id'),
                               nullable=False),
                     db.Column('tag_id', db.Integer,
                               db.ForeignKey('tags.id'),
                               nullable=False),
                     db.PrimaryKeyConstraint('post_id', 'tag_id'))

class Tag(db.Model):
    __tablename__ = 'tags'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(30), nullable=False, unique=True)

    @classmethod
    def get_or_create(cls, name):
        return cls.query.filter_by(name=name).scalar() or cls(name=name)

class Post(db.Model):
    __tablename__ = 'posts'

    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(80), nullable=False)
    content = db.Column(db.Text, nullable=False)

    _tags = db.relationship('Tag', secondary=post_tags)
    tags = association_proxy('_tags', 'name', creator=Tag.get_or_create)

    def __init__(self, title, content, tags=None):
        self.title = title
        self.content = content
        self.tags = tags

我正在使用association_proxy 来传递字符串列表并将其转换为Tag 对象列表。请注意，字符串到Tag 的转换发生在Post 对象上设置tags 属性时（例如，在实例化Post 对象时）。

从上述模块导入所有内容后，以下内容可在 Python 控制台中运行：

>>> app.app_context().push()
>>> db.create_all()
>>> post1 = Post('Test', 'A test post', tags=['Test', 'Foo'])
>>> db.session.add(post1)
>>> db.session.commit()
>>> post2 = Post('A second test', 'Another test post', tags=['Test'])
>>> db.session.add(post2)
>>> db.session.commit()

但是，以下失败：

>>> app.app_context().push()
>>> db.create_all()
>>> post1 = Post('Test', 'A test post', tags=['Test', 'Foo'])
>>> post2 = Post('A second test', 'Another test post', tags=['Test'])
>>> db.session.add(post1)
>>> db.session.add(post2)
>>> db.session.commit()

最后一行抱怨Tag.name 上的UNIQUE 约束失败：

sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) UNIQUE constraint failed:
  tag.name [SQL: 'INSERT INTO tag (name) VALUES (?)'] [parameters: ('Test',)]

我理解为什么会发生这种情况：在第一种情况下，在创建 post2 时，数据库中已经存在名称为 Test 的 Tag；在第二个中，db.session.new 包含两个具有该名称的 Tag 对象，这些对象在提交时尚未持久化。

我不知道如何解决它。我曾想过使用before_flush SQLAlchemy 事件来整合db.session.new 中的Tag 对象，但我无法使其工作。我不确定这是否是正确的策略。

StackOverflow 集体智慧有什么见解或建议吗？

【问题讨论】：

标签： python sqlalchemy many-to-many flask-sqlalchemy

【解决方案1】：

你的 get_or_create 需要将创建的标签添加到会话中，以便后续调用它可以找到会话中未提交的标签实例并返回相同的实例。

@classmethod
def get_or_create(cls, name):
    tag = cls.query.filter_by(name=name).scalar()
    if not tag:
        tag = cls(name=name)
        db.session.add(tag)
    return tag

【讨论】：

如此简单！谢谢！