Django修改ForeignKey关联模型时触发IntegrityError问题

这个问题我之前处理过，本质是数据库现有数据和新的外键约束不匹配导致的——原来Company关联的是User的ID，现在要改成关联LucyGuide的ID，但这两个ID大概率不是一一对应的（毕竟LucyGuide是独立模型，主键默认是自增的）。直接修改外键会让数据库检查约束时找不到对应的LucyGuide记录，从而抛出IntegrityError。

下面是一步步的解决流程：

1. 先确保所有关联的User都有对应的LucyGuide记录 #

首先要保证每一条Company记录里的lucy_guide（原User对象）都已经创建了对应的LucyGuide实例，否则迁移时必然会出现找不到关联的错误。

你可以打开Django Shell执行以下代码：

from django.contrib.auth.models import User
from your_app.models import LucyGuide, Company  # 替换成你的应用名称

# 遍历所有Company记录，为关联的User创建LucyGuide（如果不存在的话）
for company in Company.objects.all():
    target_user = company.lucy_guide
    LucyGuide.objects.get_or_create(user=target_user)

2. 修改模型并生成初始迁移文件 #

先把Company模型的外键改成关联LucyGuide：

class Company(models.Model):
    lucy_guide = models.ForeignKey(LucyGuide, on_delete=models.CASCADE)  # 替换原来的User关联

然后运行生成迁移的命令：

python manage.py makemigrations

这时候Django会自动生成一个迁移文件，但不要直接执行迁移——自动生成的迁移逻辑是直接替换外键，会触发IntegrityError，我们需要手动修改这个迁移文件。

3. 手动调整迁移文件的逻辑 #

打开刚生成的迁移文件（比如your_app/migrations/00XX_update_company_lucy_guide.py），替换里面的内容为以下逻辑：

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

def migrate_user_to_lucyguide(apps, schema_editor):
    # 获取迁移时的模型（不能直接导入当前模型，要通过apps获取）
    Company = apps.get_model('your_app', 'Company')
    LucyGuide = apps.get_model('your_app', 'LucyGuide')
    
    # 遍历所有Company记录，把原User关联映射到对应的LucyGuide
    for company in Company.objects.all():
        # 通过原User找到对应的LucyGuide
        lucy_guide_instance = LucyGuide.objects.get(user=company.lucy_guide)
        # 把临时字段设置为找到的LucyGuide
        company.lucy_guide_new = lucy_guide_instance
        company.save()

class Migration(migrations.Migration):

    dependencies = [
        ('your_app', '00XX_previous_migration'),  # 替换成你之前的迁移文件编号
    ]

    operations = [
        # 第一步：添加一个临时字段，用来存储新的LucyGuide外键（允许为空）
        migrations.AddField(
            model_name='company',
            name='lucy_guide_new',
            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='your_app.LucyGuide'),
        ),
        # 第二步：运行数据迁移函数，把旧的User关联转到新的LucyGuide关联
        migrations.RunPython(migrate_user_to_lucyguide),
        # 第三步：删除原来的lucy_guide字段（关联User的那个）
        migrations.RemoveField(
            model_name='company',
            name='lucy_guide',
        ),
        # 第四步：把临时字段重命名为原来的字段名
        migrations.RenameField(
            model_name='company',
            old_name='lucy_guide_new',
            new_name='lucy_guide',
        ),
        # 第五步：如果原来的字段是必填的，把新字段设置为非空
        migrations.AlterField(
            model_name='company',
            name='lucy_guide',
            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='your_app.LucyGuide'),
        ),
    ]

注意替换代码里的your_app为你的应用名称，00XX_previous_migration为迁移文件里依赖的上一个迁移编号（自动生成的迁移文件里会有，直接保留即可）。

4. 执行迁移命令 #

现在可以安全地执行迁移了：

python manage.py migrate

这样就能顺利完成外键的迁移，同时保证现有数据的完整性。

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