如何为已有数据的Django表添加unique_together约束

1. 迁移失败的核心原因 #

没错，这次迁移失败完全是因为你的表中已存在测试数据违反了要添加的unique_together约束。

unique_together要求frt_group_id、rec_loc_id、dis_loc_id这三个字段的组合在表中必须唯一，但你现有的测试数据里，肯定存在至少两组数据的这三个字段值完全相同——数据库在创建唯一约束时会检查所有现有数据，一旦发现违反约束的记录，就会抛出IntegrityError，这也是你之前空表能成功迁移的原因（空表没有数据可检查）。

2. 能否通过修改迁移文件解决？当然可以！ #

不过前提是你得先处理掉违反约束的现有数据，然后调整迁移文件的执行逻辑，具体步骤如下：

第一步：先定位并清理重复数据

首先你得知道哪些记录违反了约束，最简单的方式是通过Django Shell查询：

# 运行`python manage.py shell`后执行以下代码
from django.db.models import Count
from movt.models import Frt  # 替换成你的app名和模型类名

# 查询所有重复的字段组合
duplicate_combinations = Frt.objects.values(
    'frt_group_id', 'rec_loc_id', 'dis_loc_id'
).annotate(record_count=Count('id')).filter(record_count__gt=1)

print("重复的字段组合：", list(duplicate_combinations))

查出来后，你可以选择：

• 删除重复记录（比如保留每组的第一条，删除其余）
• 修改重复记录的某个字段值，让组合变得唯一

第二步：修改自动生成的迁移文件

当你运行makemigrations后，会生成一个包含AlterUniqueTogether操作的迁移文件。现在我们需要给这个文件加一步“前置操作”——先清理重复数据，再添加约束。

修改后的迁移文件示例：

from django.db import migrations, models
from django.db.models import Count

def clean_duplicate_frt_records(apps, schema_editor):
    # 获取迁移时的模型版本（不能直接导入当前模型，避免版本问题）
    Frt = apps.get_model('movt', 'Frt')
    # 找到所有重复的字段组合
    duplicate_groups = Frt.objects.values(
        'frt_group_id', 'rec_loc_id', 'dis_loc_id'
    ).annotate(count=Count('id')).filter(count__gt=1)
    
    for group in duplicate_groups:
        # 保留每组的第一条记录，删除后续重复的
        duplicate_records = Frt.objects.filter(**group).order_by('id')[1:]
        duplicate_records.delete()

class Migration(migrations.Migration):

    dependencies = [
        ('movt', '00xx_previous_migration'),  # 保持原来的依赖不变
    ]

    operations = [
        # 先执行数据清理操作
        migrations.RunPython(clean_duplicate_frt_records),
        # 再添加unique_together约束
        migrations.AlterUniqueTogether(
            name='frt',
            unique_together={('frt_group_id', 'rec_loc_id', 'dis_loc_id')},
        ),
    ]

⚠️ 注意：clean_duplicate_frt_records里的逻辑可以根据你的需求调整，比如如果不想删除，而是修改字段值，就把删除逻辑改成更新逻辑即可。

第三步：执行迁移

修改完迁移文件后，直接运行：

python manage.py migrate

这次就能成功完成迁移了。

额外提醒 #

• 执行数据清理前，一定要先备份数据库，避免误删重要数据；
• 以后添加约束前，最好先检查现有数据是否符合约束要求，能提前避免这类问题；
• 如果暂时不想修改数据，也可以先跳过添加约束，但这违背了你设置unique_together的初衷，不推荐。

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