Django数据库迁移编写指南：从基础到高级实践-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00203/article/details/148323491

Django数据库迁移编写指南：从基础到高级实践

django django/django: 是一个用于 Python 的高级 Web 框架，可以用于快速开发安全和可维护的 Web 应用程序，提供了多种内置功能和扩展库，支持多种数据库和模板引擎。项目地址: https://gitcode.com/gh_mirrors/dj/django

引言

在Django开发过程中，数据库迁移是管理模型变更的核心机制。本文将深入探讨Django迁移系统的各种高级用法，帮助开发者处理复杂场景下的数据库迁移需求。

多数据库环境下的迁移策略

在分布式系统架构中，我们经常需要处理多个数据库的情况。Django提供了灵活的机制来控制迁移在不同数据库上的执行：

基于数据库别名的控制

from django.db import migrations

def forwards(apps, schema_editor):
    if schema_editor.connection.alias != "default":
        return
    # 迁移代码仅会在default数据库上执行

class Migration(migrations.Migration):
    operations = [
        migrations.RunPython(forwards),
    ]

使用数据库路由提示

通过定义自定义数据库路由，可以实现更精细的控制：

# 自定义路由
class MyRouter:
    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if "target_db" in hints:
            return db == hints["target_db"]
        return True

# 迁移文件中
class Migration(migrations.Migration):
    operations = [
        migrations.RunPython(forwards, hints={"target_db": "default"}),
    ]

添加唯一字段的最佳实践

为已有数据的表添加唯一非空字段需要特殊处理，以避免违反唯一性约束：

三阶段迁移法：
- 阶段一：添加可为空的字段
- 阶段二：为现有数据填充唯一值
- 阶段三：将字段改为非空唯一

# 阶段一：添加可为空字段
class Migration(migrations.Migration):
    operations = [
        migrations.AddField(
            model_name="mymodel",
            name="uuid",
            field=models.UUIDField(default=uuid.uuid4, null=True),
        ),
    ]

# 阶段二：填充数据
def gen_uuid(apps, schema_editor):
    MyModel = apps.get_model("myapp", "MyModel")
    for row in MyModel.objects.all():
        row.uuid = uuid.uuid4()
        row.save(update_fields=["uuid"])

# 阶段三：设为非空唯一
class Migration(migrations.Migration):
    operations = [
        migrations.AlterField(
            model_name="mymodel",
            name="uuid",
            field=models.UUIDField(default=uuid.uuid4, unique=True),
        ),
    ]

非原子迁移处理大数据

对于大型数据表，可能需要禁用事务以保证性能：

class Migration(migrations.Migration):
    atomic = False  # 禁用整个迁移的事务

    operations = [
        migrations.RunPython(large_data_migration),
    ]

也可以在迁移函数内部使用小事务块：

def large_data_migration(apps, schema_editor):
    with transaction.atomic():  # 小事务块
        # 处理部分数据
        pass

迁移顺序控制

Django通过dependencies和run_before属性控制迁移顺序：

class Migration(migrations.Migration):
    dependencies = [
        ("myapp", "0123_previous_migration"),
    ]
    
    run_before = [
        ("third_party_app", "0001_initial"),
    ]

第三方应用间数据迁移

安全地迁移第三方应用数据的方法：

def forwards(apps, schema_editor):
    try:
        OldModel = apps.get_model("old_app", "OldModel")
    except LookupError:
        return  # 旧应用未安装时安全退出
    
    # 执行数据迁移逻辑

class Migration(migrations.Migration):
    dependencies = [
        ("new_app", "0001_initial"),
    ]
    
    if global_apps.is_installed("old_app"):
        dependencies.append(("old_app", "0001_initial"))

ManyToManyField使用through模型

安全转换ManyToManyField为使用中间模型：

operations = [
    migrations.SeparateDatabaseAndState(
        database_operations=[
            migrations.RunSQL(
                sql="ALTER TABLE old_table RENAME TO new_table",
                reverse_sql="ALTER TABLE new_table RENAME TO old_table",
            ),
        ],
        state_operations=[
            # 创建新的中间模型
        ],
    ),
    # 添加额外字段
]

非托管模型转托管模型

将managed=False的模型转为托管模型时，应先单独处理此变更：

class Migration(migrations.Migration):
    operations = [
        migrations.AlterModelOptions(
            name="mymodel",
            options={"managed": True},
        ),
        # 其他模式变更应在后续迁移中
    ]