PostgreSQL转SQLite迁移适配：自定义逻辑实现多数据库兼容

你可以通过在迁移文件中添加数据库引擎判断，让同一个迁移在PostgreSQL和SQLite环境下执行不同操作，无需重新创建迁移或迁移表。具体步骤如下：

1. 生成基础迁移文件 #

先正常生成针对字段调整的迁移文件：

python manage.py makemigrations

生成后不要直接执行migrate，打开这个新迁移文件进行修改。

2. 修改迁移文件，添加条件分支逻辑 #

在迁移文件中导入django.db.connection判断当前数据库引擎，然后在operations中根据引擎类型执行对应操作：

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

class Migration(migrations.Migration):

    dependencies = [
        # 填写你的旧迁移依赖，例如('your_app', '000X_old_migration'),
    ]

    def apply_postgresql(apps, schema_editor):
        # PostgreSQL环境下不做变更，保留原有数组字段
        pass

    def apply_sqlite(apps, schema_editor):
        # SQLite环境下替换数组字段为JSONField
        MyModel = apps.get_model('your_app', 'MyModel')
        # 1. 添加临时JSON字段
        schema_editor.add_field(MyModel, 'temp_json', models.JSONField(null=True))
        # 2. 迁移原数组字段的数据到临时字段（无数据可省略此步）
        for instance in MyModel.objects.all():
            instance.temp_json = instance.your_array_field
            instance.save()
        # 3. 删除原数组字段
        schema_editor.remove_field(MyModel, 'your_array_field')
        # 4. 将临时字段重命名为原字段名
        schema_editor.alter_field(MyModel, 'temp_json', models.JSONField(null=True, name='your_array_field'))

    operations = [
        migrations.RunPython(
            apply_sqlite,
            reverse_code=migrations.RunPython.noop,
            condition=lambda se: se.connection.vendor == 'sqlite'
        ),
        migrations.RunPython(
            apply_postgresql,
            reverse_code=migrations.RunPython.noop,
            condition=lambda se: se.connection.vendor == 'postgresql'
        ),
    ]

关键细节： #

• schema_editor.connection.vendor会返回当前数据库标识：PostgreSQL返回postgresql，SQLite返回sqlite。
• 若无需迁移数据，可简化SQLite操作：直接删除原数组字段，再添加同名JSONField。

3. 模型层兼容处理（可选） #

为让模型在两种数据库下都正常工作，可动态指定字段类型：

from django.db import models
from django.db import connection

class MyModel(models.Model):
    if connection.vendor == 'postgresql':
        your_array_field = models.ArrayField(models.CharField(max_length=255), blank=True, default=list)
    else:
        your_array_field = models.JSONField(blank=True, default=list)

4. 执行迁移 #

直接运行迁移命令，系统会自动适配当前数据库环境：

python manage.py migrate

切换数据库时，仅需修改settings.py中的数据库配置，再次执行migrate即可，无需删除迁移表或重新生成迁移文件。

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