Django loaddata前如何临时禁用PostgreSQL的唯一性与外键约束

首先得明确：你用的ALTER TABLE ... DISABLE TRIGGER ALL确实只能禁用触发器（包括外键关联的触发器），但唯一性约束（UNIQUE、PRIMARY KEY）是基于数据库索引实现的，触发器禁用对它们无效，这就是你还收到IntegrityError的原因。针对PostgreSQL 9.6/10和Django 1.11.9，我给你一套可行的操作流程：

步骤1：备份数据库（关键！） #

在做任何结构修改前，务必备份你的目标数据库，避免操作失误导致数据丢失：

pg_dump -U your_username -d your_database_name > backup.sql

步骤2：批量生成并执行删除约束的SQL #

我们需要先移除所有主键和唯一约束，这样导入时就不会触发唯一性检查。

生成删除主键约束的语句 #

执行以下SQL查询，会输出所有需要执行的删除主键的命令：

SELECT 'ALTER TABLE ' || table_name || ' DROP CONSTRAINT ' || constraint_name || ';'
FROM information_schema.table_constraints
WHERE constraint_type = 'PRIMARY KEY' AND table_schema = 'public';

生成删除唯一约束的语句 #

同样，执行这个查询得到删除唯一约束的命令：

SELECT 'ALTER TABLE ' || table_name || ' DROP CONSTRAINT ' || constraint_name || ';'
FROM information_schema.table_constraints
WHERE constraint_type = 'UNIQUE' AND table_schema = 'public';

把上面两个查询输出的所有SQL语句复制出来，在PostgreSQL客户端（比如psql）中执行，完成约束删除。

步骤3：禁用所有表的外键触发器 #

继续批量生成并执行禁用触发器的命令，确保外键关联不会干扰导入：

SELECT 'ALTER TABLE ' || table_name || ' DISABLE TRIGGER ALL;'
FROM information_schema.tables
WHERE table_schema = 'public' AND table_type = 'BASE TABLE';

执行输出的所有ALTER语句。

步骤4：执行Django的loaddata导入数据 #

现在可以安全执行数据导入了：

python manage.py loaddata your_dump_file.json

步骤5：重建所有约束和触发器 #

导入完成后，需要把之前移除的约束重新建回来，同时启用触发器。

生成重建主键约束的语句 #

SELECT 'ALTER TABLE ' || tc.table_name || ' ADD CONSTRAINT ' || tc.constraint_name || ' PRIMARY KEY (' || string_agg(kcu.column_name, ', ') || ');'
FROM information_schema.table_constraints tc
JOIN information_schema.key_column_usage kcu
ON tc.constraint_name = kcu.constraint_name
WHERE tc.constraint_type = 'PRIMARY KEY' AND tc.table_schema = 'public'
GROUP BY tc.table_name, tc.constraint_name;

生成重建唯一约束的语句 #

SELECT 'ALTER TABLE ' || tc.table_name || ' ADD CONSTRAINT ' || tc.constraint_name || ' UNIQUE (' || string_agg(kcu.column_name, ', ') || ');'
FROM information_schema.table_constraints tc
JOIN information_schema.key_column_usage kcu
ON tc.constraint_name = kcu.constraint_name
WHERE tc.constraint_type = 'UNIQUE' AND tc.table_schema = 'public'
GROUP BY tc.table_name, tc.constraint_name;

执行这两个查询输出的重建语句，然后启用所有触发器：

SELECT 'ALTER TABLE ' || table_name || ' ENABLE TRIGGER ALL;'
FROM information_schema.tables
WHERE table_schema = 'public' AND table_type = 'BASE TABLE';

重要注意事项 #

• 如果导入的数据中存在重复值，重建约束时会报错。这时候你必须先清理重复数据（比如用DELETE语句去重），否则无法恢复约束——毕竟唯一性约束的存在是有业务意义的，最终还是要保证数据合规。
• 以上操作都是针对public schema的，如果你的表在其他schema下，记得修改查询中的table_schema参数。
• Django 1.11的loaddata没有忽略重复的参数（这个功能在Django 2.2+才加入），所以只能通过数据库层面的约束移除来绕过唯一性检查。

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