如何在添加新的唯一字段时正确进行迁移

Question

我为我的一个模型添加了一个新字段:

class Agency(models.Model):
    email = models.EmailField(unique=True, verbose_name=_("e-mail"))

由于此字段不能为空,django-admin makemigrations请求我提供一次性默认,我这样做.这是生成的迁移:

# Generated by Django 1.9.4 on 2016-03-20 10:38
from __future__ import unicode_literals

from django.db import migrations, models


class Migration(migrations.Migration):

    dependencies = [
        ('accounts', '0008_auto_20160226_1226'),
    ]

    operations = [
        migrations.AddField(
            model_name='agency',
            name='email',
            field=models.EmailField(default='example@example.fr', max_length=254, unique=True, verbose_name='e-mail'),
            preserve_default=False,
        ),
    ]

正如所料,django-admin migrate抛出一个错误:

psycopg2.IntegrityError: could not create unique index "accounts_agency_email_key"
DETAIL:  Key (email)=(example@example.fr) is duplicate.

我想我可以在使字段唯一之前编辑迁移以设置唯一值.所以我尝试过:

# -*- coding: utf-8 -*-
# Generated by Django 1.9.4 on 2016-03-20 10:38
from __future__ import unicode_literals

from django.db import migrations, models
from django.utils.text import slugify


def set_email(apps, schema_editor):
    Agency = apps.get_model('accounts', 'Agency')
    for agency in Agency.objects.all():
        agency.email = '{}@example.fr'.format(slugify(agency.name))
        agency.save()


class Migration(migrations.Migration):

    dependencies = [
        ('accounts', '0008_auto_20160226_1226'),
    ]

    operations = [
        migrations.AddField(
            model_name='agency',
            name='email',
            field=models.EmailField(default='', max_length=254, blank=True, verbose_name='e-mail'),
            preserve_default=False,
        ),
        migrations.RunPython(set_email),
        migrations.AlterField(
            model_name='agency',
            name='email',
            field=models.EmailField(max_length=254, unique=True, verbose_name='e-mail'),
            preserve_default=False,
        ),
    ]

不幸的是,我在运行时遇到此错误django-admin migrate:

django.db.utils.OperationalError: cannot ALTER TABLE "accounts_agency" because it has pending trigger events

我的猜测是operations没有同步执行.

我想我可以通过将迁移分成两个迁移来解决问题,但我想知道我是否只能在一次迁移中完成.在模型中添加新的唯一字段时,创建迁移的常用方法是什么？

PS:我也尝试使用F表达式作为默认值(default=models.F('name') + '@example.fr')但它失败了:

django.db.utils.IntegrityError: could not create unique index "accounts_agency_email_key"
DETAIL:  Key (email)=(F(name) + Vallu(@example.fr)) is duplicated.

Answer 1

也许为时已晚，但也许对其他人有用

您可以通过使用migrations.RunSQL方法在一次迁移中完成此操作

对于将新字段添加到模型并运行python manage.py makemigrations命令后的示例代码（此处，如果表命令中有现有行想要选择默认值，则可以选择“立即提供一次性默认值”选项并给出一些字符串值这并不重要，因为实际上我们没有使用它）然后转到迁移文件并使用此更改操作部分（注意我使用 postgresql 您可以更改数据库的 SQL）

operations = [
        migrations.RunSQL(
        'ALTER TABLE "agency" ADD COLUMN "email" varchar(254) NULL;ALTER TABLE "agency" ALTER COLUMN "email" DROP DEFAULT;COMMIT;',
        ),
        migrations.RunSQL(
        "UPDATE agency SET email= Concat(country_code, '@example.fr');COMMIT;",
        ),
        migrations.RunSQL(
        'ALTER TABLE "agency" ALTER COLUMN "email" SET NOT NULL;ALTER TABLE "agency" ADD CONSTRAINT "agency_email_b551ad2a_uniq" UNIQUE ("email");ALTER TABLE "agency" ALTER COLUMN "email" DROP DEFAULT;CREATE INDEX "agency_email_b551ad2a_like" ON "agency" ("email" varchar_pattern_ops);COMMIT;'
        )
    ]

然后运行“ python manage.py migrate ”命令就是这样。