Django：如何从 ManyToMany 迁移到 ForeignKey？

Question

我正在使用 Django 和 Django Rest Framework 构建一个 REST API 和服务器。我们正在使用 postgres 数据库。

我需要简化一个设计糟糕的关系。我们有一个模型 (House) 与另一个模型 (City) 有 ManyToMany 关系。实际上，当这是ForeignKey 关系时就足够了。

我用谷歌搜索并找不到任何博客文章或文档如何正确地朝这个方向迁移。我只能找到另一种方式（FK 到 M2M）。

我 98% 确定服务器上的所有数据都将与 FK 关系一致（这意味着我很确定所有房屋只有一个城市）。我们需要改变关系有几个原因，并且无法保留 M2M。

恐怕只是更改模型并运行makemigrations 和migrate。我想知道，您如何正确地从 M2M 迁移到 FK？有什么需要注意的地方吗？如果令人惊讶的是有多个城市的房屋，我该如何处理数据？如果重要的话，数据集仍然非常小（少于 10k 个条目）。

非常感谢。

Answer 1

EDIT首先创建数据库备份

首先创建一个新的临时FK关系

_city = models.ForeignKey(...)

并进行迁移

python manage.py makemigration
python manage.py migrate

然后您需要通过创建一个空的迁移文件在相关应用中创建new data migration：

python manage.py makemigration --empty myhouseapp

并在该文件中手动将新关系从 M2M 分配给 FK。它看起来像：

from django.db import migrations


def save_city_fk(apps, schema):
    City = apps.get_model('myapp', 'City')
    House = apps.get_model('myapp', 'House')
    for house in House.objects.all():
        house._city = house.cities.all().first()  # Or whatever criterea you want
        house._city.save()


def save_city_m2m(apps, schema):
    # This should do the reverse
    City = apps.get_model('myapp', 'City')
    House = apps.get_model('myapp', 'House')
    for house in House.objects.all():
        if house._city:
            house.cities.add(house._city)


class Migration(migrations.Migration):

    dependencies = [
    ]

    operations = [
        migrations.RunPython(save_city_fk, save_city_m2m)
    ]

删除 M2M 字段，然后创建另一个迁移。

python manage.py makemigrations

将 FK 从 _city 重命名为 city 并创建最终迁移

python manage.py makemigrations

然后迁移：

python manage.py migrate

Answer 2

基于Timmy's answer 这是我所做的：

1. 我添加了一个类似 city = models.ForeignKey(City, related_name='has_houses', blank=True, null=True) 的字段，以避免 related_name 用于反向关系并将 FK 留空。然后我跑了makemigrations 和migrate。

2. 接下来，我运行了python manage.py makemigrations --empty houses，因为我的应用名为houses。

3. 我添加了迁移代码（见下文）。然后我跑了migrate。

4. 我删除了 M2M 字段和 related_name、null 和 blank 约束，最后一次运行了 makemigrations 和 migrate。

迁移代码：

# -*- coding: utf-8 -*-
# Generated by Django 1.11.15 on 2019-02-15 09:09
from __future__ import unicode_literals

from django.db import migrations


def save_city_fk(apps, schema):
    City = apps.get_model('cities', 'City')
    House = apps.get_model('houses', 'House')
    for house in House.objects.all():
        house.city = house.cities.all().first()
        house.save()


class Migration(migrations.Migration):

    dependencies = [
        ('houses', '0003_auto_20190215_0949'),
    ]

    operations = [
        migrations.RunPython(save_city_fk),
    ]

Answer 3

首先，显然您需要确保（通过进行适当的查询）您确实每栋房屋只有一个城市。如果有房屋拥有多个城市，则需要通过从关系中删除城市、拆分房屋等方式来解决冲突。

之后，你可以分步进行：

• 在 House 中创建一个新的 FK，使用旧 m2m 关系中的一个城市 ID 迁移并填充它
• 重命名 m2m 字段，迁移，然后根据需要为新 FK 指定旧 m2m 字段的名称并再次迁移
• 检查您的查询是否有效并根据需要进行调整
• 当对一切正常感到满意时，删除旧的 m2m 字段 - 只有在迁移此步骤后，您才会失去回滚 db 的能力

Answer 4

对于 M2M 关系，最好建立一个新的模型来表示关系。下面是一个`员工拥有多个职称和多个员工获得的职称的例子

class Designation(models.Model):
    desig_id = models.AutoField(primary_key=True)
    title = models.CharField(max_length=60, unique=True)
    job_discription = models.CharField(max_length=2000)

    def __str__(self):
        return self.title

class Employee(AbstractUser):
    middle_name = models.CharField(max_length=20, blank=True,  null=True)
    basic_salary = models.FloatField(default=1)
    designation = models.ManyToManyField(Designation, default=None, blank=True,
                                     through='EmployeeDesignation')
    certification = models.ManyToManyField(Certification, default=None, null=True, blank=True,
                                        through='EmployeeCertification')
    emp_img = models.FileField(default=None,upload_to='employees')
    leaves_allowed = models.IntegerField(default=25)
    leave_balance = models.IntegerField(default=25)
    leave_count = models.IntegerField(default=0)
    objects = EmployeeManager()

    def __str__(self):
        return self.first_name+' '+self.last_name

class EmployeeDesignation(models.Model):
    desig = models.ForeignKey(Designation, on_delete=models.CASCADE)
    emp = models.ForeignKey(Employee, on_delete=models.CASCADE)

    class Meta:
        unique_together = (('emp', 'desig'),)