如何在Django的RunPython中条件执行迁移操作？

你当前的写法存在问题：直接在RunPython的函数内调用migrations.AddField()是无效的，因为迁移操作类必须定义在Migration的operations列表中，RunPython的函数是用来执行自定义Python逻辑，需通过schema_editor来实际修改数据库结构。以下是两种可行的实现方案：

方案1：通过schema_editor在运行时执行条件操作 #

适合需要基于数据库数据、当前数据库类型等运行时条件判断的场景，核心是利用schema_editor提供的方法完成字段操作：

from django.db import migrations, models

def forward_migration(apps, schema_editor):
    # 替换为你的实际条件判断逻辑，比如检查数据库中是否存在特定数据
    some_condition = True
    
    if some_condition:
        # 必须用apps.get_model获取历史版本的模型，不能直接导入当前模型
        MyModel = apps.get_model('some_app', 'MyModel')
        # 调用schema_editor的add_field方法，对应AddField迁移操作
        schema_editor.add_field(
            model=MyModel,
            field=models.CharField(max_length=100, null=True, blank=True, name='new_field')
        )

def reverse_migration(apps, schema_editor):
    # 反向迁移需对应处理，保证迁移可回滚
    some_condition = True
    if some_condition:
        MyModel = apps.get_model('some_app', 'MyModel')
        schema_editor.remove_field(
            model=MyModel,
            field_name='new_field'
        )

class Migration(migrations.Migration):
    dependencies = [
        ('some_app', '0002_auto_...'),
    ]

    operations = [
        migrations.RunPython(forward_migration, reverse_migration),
    ]

关键说明： #

• schema_editor支持多种操作：add_field、alter_field、remove_field、rename_field等，对应常规迁移中的AddField、AlterField等操作。
• 必须使用apps.get_model()获取模型，避免直接导入当前模型类，因为迁移依赖的是历史版本的模型定义。

方案2：动态生成operations列表 #

适合条件可提前确定的场景（比如基于环境变量、配置开关），直接动态构建迁移操作列表，写法更贴近常规迁移：

from django.db import migrations, models

# 替换为你的静态条件判断，比如读取环境变量
some_condition = True

# 动态构建操作列表
operations = []
if some_condition:
    operations.append(
        migrations.AddField(
            model_name='mymodel',
            name='new_field',
            field=models.CharField(max_length=100, null=True, blank=True),
        )
    )

class Migration(migrations.Migration):
    dependencies = [
        ('some_app', '0002_auto_...'),
    ]

    operations = operations

适用场景： #

当你不需要在迁移运行时访问数据库，仅需基于静态配置决定是否执行迁移操作时，这种方案更简洁易读。

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