如何在 django 迁移时 set/provide 默认值？

Question

场景:
我有一个模型，Customer

class Customer(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()
    company = models.CharField(max_length=100)

现在我用 ForeignKey 关系更新了 company 属性，如下所示，

class Company(models.Model):
    name = models.CharField(max_length=100)
    location = models.CharField(max_length=100)


class Customer(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()
    company = models.ForeignKey(Company)

我需要的是，当新的迁移应用于数据库时，相应的 Company 实例必须自动生成并映射到 Customer 实例的 company 属性。

这可能吗？我怎样才能做到这一点？

Answer 1

当然可以，但是您必须执行三个迁移，并且字段不能命名为相同的东西，因为两者需要同时存在。如果您已经删除了真实数据库中的公司字段，那么您就是 SOL 并且必须手动修复它们。

首先，在正常的数据库迁移中添加 Company 模型，然后进行数据迁移并在第一次数据库迁移后进行数据迁移运行，然后进行另一次数据库迁移，从中删除公司字段Customer 模型。

您可以像往常一样使用 manage.py makemigrations 进行数据库迁移，只需在它们之间的迁移文件中添加如下内容，这里我将新公司 ForeignKey 字段命名为 company_obj

def fix_companies(apps, schema_editor):
    Company = apps.get_model("myapp", "Company")
    Customer = apps.get_model("myapp", "Customer")
    for c in Customer.objects.all():
        company, _ = Company.objects.get_or_create(name=c.name)
        c.company_obj = company
        c.save()


def rev(apps, schema_editor):
    # the reverse goes here if you want to copy company names into customer again if you migrate backwards.
    pass

class Migration(migrations.Migration):

    dependencies = [
        ('myapp', 'XXXX_migration_that_added_company_model'),
    ]

    operations = [
        migrations.RunPython(fix_companies, rev),
    ]

Answer 2

让我们从你原来的模型开始，一步一步来。

class Customer(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()
    company = models.CharField(max_length=100)

首先，您必须保留原始字段并创建一个新字段，以便之后能够恢复旧数据。

class Customer(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()
    company = models.CharField(max_length=100)
    _company = models.ForeignKey(Company)

现在您可以使用 manage.py makemigrations 创建第一个迁移。然后你将不得不创建一个data migration。使用 manage.py makemigrations yourapp --empty 创建迁移并更新生成的文件：

from django.db import migrations

def export_customer_company(apps, schema_editor):
    Customer = apps.get_model('yourapp', 'Customer')
    Company = apps.get_model('yourapp', 'Company')
    for customer in Customer.objects.all():
        customer._company = Company.objects.get_or_create(name=customer.company)[0]
        customer.save()

def revert_export_customer_company(apps, schema_editor):
    Customer = apps.get_model('yourapp', 'Customer')
    Company = apps.get_model('yourapp', 'Company')
    for customer in Customer.objects.filter(_company__isnull=False):
        customer.company = customer._company.name
        customer.save()

class Migration(migrations.Migration):

    dependencies = [
        ('yourapp', 'xxxx_previous_migration'),  # Note this is auto-generated by django
    ]

    operations = [
        migrations.RunPython(export_customer_company, revert_export_customer_company),
    ]

以上迁移将根据 Customer.company.

填充您的 Company 模型和 Customer._company 字段

现在您可以删除旧的 Customer.company 并重命名 Customer._company。

class Customer(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()
    company = models.ForeignKey(Company)

最终 manage.py makemigrations 和 manage.py migrate。

Answer 3

需要注意的一点是，如果您正在经历一个重命名/弃用字段的循环，使用 RunPython 会让您指向旧模型字段，这些字段在您完成字段更改后将不再存在.

为避免这种情况，您可能需要使用 RunSQL。

# Generated by Django 3.2.3 on 2022-02-09 04:55

from django.db import migrations


class Migration(migrations.Migration):

    dependencies = [
        ("<your_app>", "<0006_migration_name>"),
    ]

    operations = [
        migrations.RunSQL(f"""
update public.<table_name> set new_field = old_field + some_magic;
"""
        ),
    ]

Docs.

如何在 django 迁移时 set/provide 默认值？

How can I set/provide a default value while django migration?

django

django-models

django-migrations