如何使用 SQLAlchemy 将新记录插入到有关每个现有数据的关联表中

Question

环境：python3.7、SQLAlchemy1.3、SQLite3.22

我定义了 2 个表和 1 个关联表，如下所示。

# associative table
taggings = Table('taggings', Base.metadata, 
    Column('id', Integer, primary_key=True),
    Column('article_id', Integer, ForeignKey('articles.id'), nullable=False),
    Column('tag_name', Integer, ForeignKey('tags.name'), nullable=False),
)

# mapper about article master
class Article(Base):
    __tablename__ = 'articles'
    id = Column(Integer, primary_key=True)
    ...
    tags = relationship('Tag', secondary=taggings, backref='articles')

# mapper about tag master
class Tag(Base):
    __tablename__ = 'tags'
    name = Column(String, primary_key=True)

现在我希望能够创建带有任何现有标签的新文章。 所以，我试着...

new_one = Article(**payload.article)
new_one.tags = [Tag(name=x) for x in payload.tags]  # these tags are already stored in database.
sesion.add(new_one)
session.flush()

并对其进行测试，但我收到有关 UNIQUE 约束的错误。 详情在这里。

*** sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) UNIQUE 约束失败：tags.name [SQL: 'INSERT INTO tags (name) VALUES (?)'] [parameters: (('x',) , ('a',))]（此错误的背景：http://sqlalche.me/e/gkpj）

我猜session.add 总是在父元素有子元素时尝试插入，而这里呈现的“子元素”可能是Tag。 但我真正想做的只是针对引用新文章数据和任何现有标签的taggings 表创建新记录。

我阅读了有关cascade in the official document 的信息，并尝试从relationship() 中删除save-update 选项，但未解决。

请问如何解决这个问题。

Answer 1

考虑到您对我的问题的回答，您应该添加检查 payload 中的标签是否不在数据库中。所以应该是这样的：

new_one = Article(**payload.article)
tags = []
for tag_name in payload.tags:
    tag = session.query(Tag).filter_by(name=tag_name).first()
    # tag will be None if it does not exist in the database
    tags.append(tag if tag else Tag(name=tag_name))
new_one.tags = tags
sesion.add(new_one)
session.flush()

所以现在你将使用现有标签，如果它已经在数据库中，或者创建一个新标签。