這一節(jié)介紹你可能遇到的在不同情況下如何分析和編寫數(shù)據(jù)庫(kù)遷移. 有關(guān)遷移的入門資料，請(qǐng)查看 the topic guide.

數(shù)據(jù)遷移和多數(shù)據(jù)庫(kù)

在使用多個(gè)數(shù)據(jù)庫(kù)時(shí)，需要解決是否針對(duì)某個(gè)特定數(shù)據(jù)庫(kù)運(yùn)行遷移。例如，你可能 只 想在某個(gè)特定數(shù)據(jù)庫(kù)上運(yùn)行遷移。

為此你可以在RunPython中通過查看schema_editor.connection.alias 屬性來檢查數(shù)據(jù)庫(kù)連接別名：

from django.db import migrations

def forwards(apps, schema_editor):
    if not schema_editor.connection.alias == 'default':
        return
    # Your migration code goes here

class Migration(migrations.Migration):

    dependencies = [
        # Dependencies to other migrations
    ]

    operations = [
        migrations.RunPython(forwards),
    ]

Django 1.8 中新增。

你也可以提供一個(gè)提示作為 **hints參數(shù)傳遞到數(shù)據(jù)庫(kù)路由的allow_migrate() 方法：

myapp/dbrouters.py
class MyRouter(object):

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if 'target_db' in hints:
            return db == hints['target_db']
        return True

然后，要在你的遷移中利用，執(zhí)行以下操作：

from django.db import migrations

def forwards(apps, schema_editor):
    # Your migration code goes here

class Migration(migrations.Migration):

    dependencies = [
        # Dependencies to other migrations
    ]

    operations = [
        migrations.RunPython(forwards, hints={'target_db': 'default'}),
    ]

如果你的RunPython或者RunSQL操作只對(duì)一個(gè)模型有影響，最佳實(shí)踐是將model_name作為提示傳遞，使其盡可能對(duì)路由可見。這對(duì)可復(fù)用的和第三方應(yīng)用極其重要。

添加唯一字段的遷移

如果你應(yīng)用了一個(gè)“樸素”的遷移，向表中一個(gè)已存在的行中添加了一個(gè)唯一的非空字段，會(huì)產(chǎn)生錯(cuò)誤，因?yàn)槲挥谝汛嬖谛兄械闹抵粫?huì)生成一次。所以需要移除唯一性的約束。

所以，應(yīng)該執(zhí)行下面的步驟。在這個(gè)例子中，我們會(huì)以默認(rèn)值添加一個(gè)非空的UUIDField字段。你可以根據(jù)你的需要修改各個(gè)字段。

• 把default=...和unique=True參數(shù)添加到你模型的字段中。在這個(gè)例子中，我們默認(rèn)使用uuid.uuid4。
• 運(yùn)行 makemigrations 命令。
• 編輯創(chuàng)建的遷移文件。

生成的遷移類看上去像這樣：

class Migration(migrations.Migration):

    dependencies = [
        ('myapp', '0003_auto_20150129_1705'),
    ]

    operations = [
        migrations.AddField(
            model_name='mymodel',
            name='uuid',
            field=models.UUIDField(max_length=32, unique=True, default=uuid.uuid4),
        ),
    ]

你需要做三處更改：

• 從已生成的遷移類中復(fù)制，添加第二個(gè)AddField操作，并改為AlterField。
• 在第一個(gè)AddField操作中，把unique=True改為 null=True，這會(huì)創(chuàng)建一個(gè)中間的null字段。
• 在兩個(gè)操作之間，添加一個(gè)RunPython或RunSQL操作為每個(gè)已存在的行生成一個(gè)唯一值（例如UUID）。

最終的遷移類應(yīng)該看起來是這樣：

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.db import migrations, models
import uuid

def gen_uuid(apps, schema_editor):
    MyModel = apps.get_model('myapp', 'MyModel')
    for row in MyModel.objects.all():
        row.uuid = uuid.uuid4()
        row.save()

class Migration(migrations.Migration):

    dependencies = [
        ('myapp', '0003_auto_20150129_1705'),
    ]

    operations = [
        migrations.AddField(
            model_name='mymodel',
            name='uuid',
            field=models.UUIDField(default=uuid.uuid4, null=True),
        ),
        # omit reverse_code=... if you don't want the migration to be reversible.
        migrations.RunPython(gen_uuid, reverse_code=migrations.RunPython.noop),
        migrations.AlterField(
            model_name='mymodel',
            name='uuid',
            field=models.UUIDField(default=uuid.uuid4, unique=True),
        ),
    ]

現(xiàn)在你可以像平常一樣使用migrate命令應(yīng)用遷移。

注意如果你在這個(gè)遷移運(yùn)行時(shí)讓對(duì)象被創(chuàng)建，就會(huì)產(chǎn)生競(jìng)爭(zhēng)條件(race condition)。在AddField之后， RunPython之前創(chuàng)建的對(duì)象會(huì)覆寫他們?cè)嫉膗uid。