如何将Django抽象基模型转非抽象模型并解决键冲突？

问题背景 #

现有抽象基模型Work，两个子类WorkMusic、WorkVisual分别存储音乐和视觉作品数据：

# 原抽象基模型
class Work(models.Model):
    creator = models.ForeignKey(Person, null=True, blank=True, on_delete=models.PROTECT)
    name = models.CharField(max_length=400, null=True, blank=True) 

    class Meta:
       abstract = True

# 子类模型
class WorkMusic(MPTTModel, Work):
    key = models.CharField(max_length=10, null=True, blank=True)
    tonality = models.CharField(max_length=20, null=True, blank=True)
    tempo = models.CharField(max_length=500, null=True, blank=True)

class WorkVisual(Work):
    material = models.CharField(max_length=100, null=True, blank=True)
    material_original = models.CharField(max_length=100, null=True, blank=True)

需求是将Work转为非抽象模型，把两类作品的数据整合到统一结构中，尝试移除abstract=True后加载fixture出现唯一键冲突，需解决该问题并完成模型转换。

方案一：单表继承（整合所有数据到一张表） #

适合需要将所有作品数据存储在同一表的场景，通过work_type字段区分作品类型：

1. 创建过渡模型 #

先创建包含所有字段的非抽象模型，用于迁移数据：

class BaseWork(models.Model):
    creator = models.ForeignKey(Person, null=True, blank=True, on_delete=models.PROTECT)
    name = models.CharField(max_length=400, null=True, blank=True)
    # 标记作品类型
    TYPE_CHOICES = (
        ('music', '音乐作品'),
        ('visual', '视觉作品'),
    )
    work_type = models.CharField(max_length=20, choices=TYPE_CHOICES)
    # 子类特有字段，允许空值
    key = models.CharField(max_length=10, null=True, blank=True)
    tonality = models.CharField(max_length=20, null=True, blank=True)
    tempo = models.CharField(max_length=500, null=True, blank=True)
    material = models.CharField(max_length=100, null=True, blank=True)
    material_original = models.CharField(max_length=100, null=True, blank=True)

2. 执行迁移 #

生成并应用迁移，创建新表：

python manage.py makemigrations
python manage.py migrate

3. 迁移现有数据 #

通过Django Shell将旧模型数据导入新模型：

# 迁移音乐作品数据
from app.models import WorkMusic, BaseWork
for item in WorkMusic.objects.all():
    BaseWork.objects.create(
        creator=item.creator,
        name=item.name,
        work_type='music',
        key=item.key,
        tonality=item.tonality,
        tempo=item.tempo
    )

# 迁移视觉作品数据
from app.models import WorkVisual
for item in WorkVisual.objects.all():
    BaseWork.objects.create(
        creator=item.creator,
        name=item.name,
        work_type='visual',
        material=item.material,
        material_original=item.material_original
    )

4. 替换原模型结构 #

删除旧的抽象Work模型，将BaseWork重命名为Work，并创建代理子类保持原业务逻辑：

class Work(models.Model):
    creator = models.ForeignKey(Person, null=True, blank=True, on_delete=models.PROTECT)
    name = models.CharField(max_length=400, null=True, blank=True)
    TYPE_CHOICES = (
        ('music', '音乐作品'),
        ('visual', '视觉作品'),
    )
    work_type = models.CharField(max_length=20, choices=TYPE_CHOICES)
    key = models.CharField(max_length=10, null=True, blank=True)
    tonality = models.CharField(max_length=20, null=True, blank=True)
    tempo = models.CharField(max_length=500, null=True, blank=True)
    material = models.CharField(max_length=100, null=True, blank=True)
    material_original = models.CharField(max_length=100, null=True, blank=True)

class WorkMusic(Work):
    class Meta:
        proxy = True

    def save(self, *args, **kwargs):
        self.work_type = 'music'
        super().save(*args, **kwargs)

class WorkVisual(Work):
    class Meta:
        proxy = True

    def save(self, *args, **kwargs):
        self.work_type = 'visual'
        super().save(*args, **kwargs)

5. 清理旧模型 #

验证数据无误后，删除旧的WorkMusic、WorkVisual模型，生成并执行迁移清理旧表。

方案二：多表继承（保留子类表，父表存储公共字段） #

适合需要保留子类独立表结构，仅将公共字段提取到父表的场景，解决唯一键冲突的核心是手动关联父类与子类实例：

1. 转换Work为非抽象模型 #

移除Work的abstract=True配置：

class Work(models.Model):
    creator = models.ForeignKey(Person, null=True, blank=True, on_delete=models.PROTECT)
    name = models.CharField(max_length=400, null=True, blank=True) 

2. 生成并执行迁移 #

此时Django会创建Work表，同时为WorkMusic、WorkVisual添加work_ptr_id外键字段：

python manage.py makemigrations
python manage.py migrate

3. 迁移公共字段并关联实例 #

通过脚本为每个子类实例创建对应的父类实例，避免id冲突：

from app.models import Work, WorkMusic, WorkVisual

# 处理音乐作品
for music in WorkMusic.objects.all():
    # 用子类id创建父类实例，确保外键关联正确
    if not Work.objects.filter(id=music.id).exists():
        work = Work(id=music.id, creator=music.creator, name=music.name)
        work.save()
    # 关联子类到父类实例
    music.work_ptr_id = music.id
    music.save()

# 处理视觉作品（处理可能的id冲突）
for visual in WorkVisual.objects.all():
    if not Work.objects.filter(id=visual.id).exists():
        work = Work(id=visual.id, creator=visual.creator, name=visual.name)
        work.save()
        visual.work_ptr_id = visual.id
        visual.save()
    else:
        # 若id重复，创建新的父类实例并关联
        work = Work(creator=visual.creator, name=visual.name)
        work.save()
        visual.work_ptr_id = work.id
        visual.save()

4. 验证数据 #

检查Work表是否包含所有公共字段数据，子类表的work_ptr_id是否正确关联父表id。

内容的提问来源于stack exchange，提问作者H C