PostgreSQL中删除users表数据时跳过被外键引用行的解决方案

我尝试从users表中删除一批行，但部分user_id在user_plan表中被作为外键（FK）引用。我希望在for循环中跳过无法删除的行（即处理外键约束冲突错误）。

当前使用的代码如下：

x = [1,31,32,33,34]
for i in x:
    cursor.execute("DELETE FROM users WHERE id=%s", (str(i),))

执行时触发了如下错误：

psycopg2.errors.ForeignKeyViolation: update or delete on table "users" violates foreign key constraint "user_fk_id" on table "user_plan"
DETAIL: Key (id)=(1) is still referenced from table "user_plan".

请问如何修改查询语句，使其在外键被其他表引用时自动跳过对应行的删除操作？

解决方案 #

我给你两种可行的方案，你可以根据自己的实际场景选择：

方案1：单条SQL批量删除（推荐） #

没必要用循环一条条删，直接用一条DELETE语句就能搞定。我们可以通过子查询筛选出不在user_plan表中被引用的用户ID，然后批量删除，这样自动跳过有外键约束的行：

x = [1,31,32,33,34]
cursor.execute("""
    DELETE FROM users 
    WHERE id = ANY(%s) 
    AND id NOT IN (SELECT user_id FROM user_plan)
""", (x,))
# 记得最后提交事务（如果不是自动提交模式）
# connection.commit()

这种方法效率更高，因为只需要和数据库交互一次，适合批量操作的场景。

方案2：循环中捕获异常跳过 #

如果你坚持要保留循环的方式，可以捕获psycopg2的外键约束异常，遇到冲突时跳过当前行：

import psycopg2
from psycopg2 import errors

x = [1,31,32,33,34]
for i in x:
    try:
        cursor.execute("DELETE FROM users WHERE id=%s", (i,))
    except errors.ForeignKeyViolation:
        print(f"跳过用户ID {i}：存在外键引用，无法删除")
        # 如果不是自动提交模式，需要回滚当前失败的操作
        connection.rollback()
# 最后统一提交成功的删除操作
# connection.commit()

注意这里要确保你能拿到数据库连接对象（connection），非自动提交模式下，捕获异常后一定要回滚当前失败的操作，不然会影响后续的数据库操作。

额外提醒 #

如果你的外键约束设置了ON DELETE CASCADE，删除用户时会自动连带删除user_plan表中对应的记录——但这显然不是你想要的（你只是想跳过删除），所以如果要保留关联的user_plan数据，别用这个外键选项。

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