如何在SQLAlchemy中跳过已存在标签，仅添加新标签至数据库？

查询Tag表并将结果存储到变量：

all_tags = Tag.query.all() # <- Query all existing tags

输出：

>>> all_tags
[<Tag>: STM32, <Tag>: Linux, <Tag>: Unix, <Tag>: Skype, <Tag>: MCU, <Tag>: CPU, <Tag>: Silk, <Tag>: WAN]

从JSON客户端接收标签值，希望跳过已存在的标签，仅添加新标签到数据库，编写的代码如下：

for tag in json_data['tags']:
      tag = Tag(tag_name=tag)
      if tag in all_tags: # <- If tag from json query does exists in table skip
        pass
      myPost.tags.append(tag) # < - or add it

但代码无法正常运行，抛出错误：

sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) UNIQUE constraint failed: tags.tag_name

问题出在对象实例的比较：你每次循环都新建了一个Tag对象，哪怕tag_name和数据库里的一样，这个新对象和all_tags里的数据库实例在Python中是不同的对象，所以if tag in all_tags永远返回False，导致你不断尝试添加重复标签，触发唯一约束错误。

正确的实现步骤：

1. 先提取所有已存在的标签名称到一个集合（集合的查找效率远高于列表）：

existing_tag_names = {tag.tag_name for tag in all_tags}

2. 遍历JSON中的标签时，先判断名称是否已存在：
   • 若存在：直接从数据库获取对应的Tag实例
   • 若不存在：创建新的Tag实例
   • 最后将实例关联到myPost.tags

完整代码示例：

# 提取已存在的标签名称集合
existing_tag_names = {tag.tag_name for tag in all_tags}

for tag_name in json_data['tags']:
    if tag_name in existing_tag_names:
        # 获取已存在的标签实例
        tag = Tag.query.filter_by(tag_name=tag_name).first()
    else:
        # 创建新标签实例
        tag = Tag(tag_name=tag_name)
        # 若需立即存入数据库，可添加 db.session.add(tag)
    # 将标签关联到帖子
    myPost.tags.append(tag)

# 最后统一提交数据库操作（若之前未单独提交）
# db.session.commit()

关键说明 #

• 用标签名称的集合做存在性判断，比遍历all_tags列表高效得多，尤其是标签数量多的时候。
• 关联已存在的标签实例，而非新建重复对象，避免触发唯一约束错误。
• 若数据库会话需要手动提交，记得最后执行db.session.commit()保存所有变更。

内容的提问来源于stack exchange，提问作者ussrback