如何为Django迁移的forwards和backwards函数添加类型提示？

要给你的迁移代码加上类型提示，需要导入Django内部提供的对应类型，再给函数参数、模型变量添加类型标注。以下是具体实现步骤和完整代码：

1. 导入必要的类型 #

先引入迁移过程中用到的核心类型：

from django.db import migrations
from django.db.migrations.operations.base import Apps
from django.db.backends.base.schema import BaseSchemaEditor
from django.db.models import Model
from typing import Type

2. 为forwards和backwards函数添加类型提示 #

• apps参数的类型是Apps，它是迁移中用于获取历史版本模型的核心对象
• schema_editor参数的类型是BaseSchemaEditor，对应数据库Schema编辑器的基类
• 通过apps.get_model()获取的模型类，类型标注为Type[Model]

这里要注意：迁移中不要直接引用当前代码里的模型类（比如你原代码里的FixedResponse.objects.create），必须用apps.get_model()获取历史版本的模型，否则可能因模型结构变更导致迁移失败。

完整的带类型提示的代码 #

from django.db import migrations
from django.db.migrations.operations.base import Apps
from django.db.backends.base.schema import BaseSchemaEditor
from django.db.models import Model
from typing import Type


def forwards(apps: Apps, schema_editor: BaseSchemaEditor) -> None:
    Foo: Type[Model] = apps.get_model('app', 'Foo')
    FixedResponse: Type[Model] = apps.get_model('app', 'FixedResponse')
    FixedResponse.objects.create(name='Bar')


def backwards(apps: Apps, schema_editor: BaseSchemaEditor) -> None:
    Foo: Type[Model] = apps.get_model('app', 'Foo')
    Foo.objects.filter(name='Bar').delete()


class Migration(migrations.Migration):
    dependencies = [
        ('app', '0035_fixedresponse'),
    ]

    operations = [
        migrations.RunPython(forwards, backwards)
    ]

额外说明 #

• 这些类型都是Django内部公开可用的类型，虽然官方文档没有专门针对迁移的类型提示说明，但使用这些基类做类型标注是安全且符合Django设计逻辑的
• 如果需要更精确的模型类型，可以用Type[YourModel]，但迁移中不推荐直接引用当前模型，所以用Type[Model]是更通用的选择

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