迁移操作（如AddField）如何检测执行方向？自定义同类迁移功能

Django的迁移操作（比如AddField）之所以能自动处理应用和回滚，是因为它的父类（FieldOperation继承自MigrationOperation）已经实现了反向操作逻辑——不是在你看到的__init__方法里，而是通过reverse()方法或者专门的database_forwards/database_backwards方法来区分执行方向的。

要实现自己的支持双向执行的迁移操作，你需要自定义一个继承自migrations.Operation的类，分别实现应用和回滚时的逻辑：

步骤1：自定义操作类 #

创建一个继承自migrations.Operation的类，核心要实现以下方法：

• database_forwards：应用迁移时，在数据库上执行的逻辑
• database_backwards：回滚迁移时，在数据库上执行的逻辑
• describe：返回操作的描述文本，用于迁移命令输出
• （可选）state_forwards/state_backwards：如果操作需要修改模型状态元数据（比如像AddField那样影响模型字段），则需要实现这两个方法

步骤2：示例实现 #

比如我们做一个操作，应用时给Foo模型的bar字段批量添加测试数据，回滚时清除这些数据：

from django.db import migrations, models

class AddTestData(migrations.Operation):
    def __init__(self, model_name, field_name):
        self.model_name = model_name
        self.field_name = field_name

    def database_forwards(self, app_label, schema_editor, from_state, to_state):
        # 应用迁移：给所有Foo实例的bar字段设置测试值
        model = to_state.apps.get_model(app_label, self.model_name)
        model.objects.update(**{self.field_name: "mig_test_data"})

    def database_backwards(self, app_label, schema_editor, from_state, to_state):
        # 回滚迁移：清除测试值（设为None，可根据需求调整）
        model = from_state.apps.get_model(app_label, self.model_name)
        model.objects.update(**{self.field_name: None})

    def state_forwards(self, app_label, state):
        # 不需要修改模型状态，直接pass
        pass

    def describe(self):
        return f"Add test data to {self.model_name}.{self.field_name}"

步骤3：在迁移中使用自定义操作 #

像用AddField一样，把自定义操作加入迁移文件的operations列表：

from django.db import migrations, models
from .operations import AddTestData  # 假设自定义类在operations.py中

class Migration(migrations.Migration):
    dependencies = [
        ('yourapp', '0001_initial'),  # 依赖的前序迁移
    ]

    operations = [
        migrations.AddField(
            model_name="foo",
            name="bar",
            field=models.TextField(blank=True, null=True),
        ),
        AddTestData(
            model_name="foo",
            field_name="bar",
        ),
    ]

关键说明 #

• database_forwards和database_backwards中的from_state/to_state参数：分别代表迁移执行前后的模型状态，通过它们获取模型类，能避免因模型代码变更导致的不一致问题
• 内置操作比如AddField的反向逻辑是返回RemoveField实例，这也是回滚时自动删除字段的核心原因

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