Django为ABC模型新增字段并根据关联GHI表字段值填充历史数据

步骤1：给ABC模型新增Current_Status字段 #

首先修改你的ABC模型代码，新增和GHI的variable_3规则完全一致的字段，初始设置允许为空避免首次迁移报错：

class ABC(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    variable_1 = models.DecimalField(max_digits=8, decimal_places=2, blank=True, null=True)
    # 新增字段
    Current_Status = models.CharField(
        choices=Status.choices(),
        default=Status.Status1,
        max_length=50,
        blank=True,
        null=True
    )

修改完成后执行命令生成首次迁移文件，注意替换为你自己的App名称：

python manage.py makemigrations 你的App名称

步骤2：创建自定义数据迁移文件，填充已有记录的字段值 #

执行命令创建空的自定义迁移文件：

python manage.py makemigrations 你的App名称 --empty

打开新生成的空迁移文件，添加如下逻辑，注意替换占位的App名称、模型路径：

from django.db import migrations, models
# 替换为你自己的Status导入路径
from your_app_path.models import Status

def populate_current_status(apps, schema_editor):
    # 从迁移上下文中获取模型，禁止直接导入自己写的模型，避免后续修改模型导致迁移失效
    ABC = apps.get_model('你的App名称', 'ABC')
    GHI = apps.get_model('你的App名称', 'GHI')
    from django.db.models import Count

    for abc_item in ABC.objects.all():
        # 统计当前ABC关联的所有GHI记录中各状态的出现频次，按频次倒序、状态正序排序（频次相同取字典序靠前的状态）
        status_rank = GHI.objects.filter(def__abc=abc_item)\
                                 .values('variable_3')\
                                 .annotate(count=Count('variable_3'))\
                                 .order_by('-count', 'variable_3')
        if status_rank.exists():
            abc_item.Current_Status = status_rank.first()['variable_3']
        else:
            # 无关联GHI记录时使用默认状态
            abc_item.Current_Status = Status.Status1.value
        abc_item.save()

class Migration(migrations.Migration):
    # 这里保留自动生成的依赖配置即可，不需要手动修改
    dependencies = [
        ('你的App名称', 'xxxx_上一个迁移文件的编号'),
    ]

    operations = [
        migrations.RunPython(populate_current_status),
    ]

编辑完成后执行迁移命令，即可完成所有已有ABC记录的字段自动填充：

python manage.py migrate 你的App名称

步骤3（可选）：调整字段约束 #

如果业务不允许Current_Status为空，你可以把ABC模型里该字段的blank=True, null=True参数删除，再执行一次makemigrations和migrate即可，Django会自动用你设置的默认值处理可能的空值情况。

后续自动更新说明 #

如果需要后续GHI记录新增、修改、删除时自动更新关联ABC的Current_Status，可以给GHI模型添加post_save和post_delete信号，触发对应ABC的状态重算即可。

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