如何为Django现有MovieTicket模型添加中间父类实现多表继承？

问题背景 #

我需要给现有Django模型添加一个中间父类实现逻辑分组：

• 初始模型结构：

from django.db import models

class Ticket(PolymorphicModel):
    name = models.CharField(max_length=50)
    company = models.CharField(max_length=80)
    price = models.CharField(max_length=10)

class MovieTicket(Ticket):
    # 业务逻辑

• 期望最终结构：

class Ticket(PolymorphicModel):
    name = models.CharField(max_length=50)
    address = models.CharField(max_length=80)
    price = models.CharField(max_length=10)

class EntertainmentTicket(Ticket):
    # 抽取的通用逻辑

class MovieTicket(EntertainmentTicket):
    # MovieTicket专属逻辑

限制条件：

• 子类仅包含函数逻辑，与Ticket字段完全一致
• 已有其他模型通过外键关联Ticket，无法将其设为抽象类
• 使用django-polymorphic实现多态查询

目前已生成EntertainmentTicket的迁移文件，下一步需要让现有MovieTicket关联的Ticket实例对应创建EntertainmentTicket实例，同时保证数据关联正确。

解决方案 #

步骤1：运行EntertainmentTicket的初始迁移 #

先执行已生成的迁移，创建EntertainmentTicket数据表：

python manage.py migrate

步骤2：编写数据迁移脚本，关联现有数据 #

生成一个空迁移文件用于处理数据转换：

python manage.py makemigrations --empty your_app_name

打开生成的迁移文件，替换为以下代码（注意替换your_app_name为实际应用名，迁移依赖编号改为EntertainmentTicket对应的迁移编号）：

from django.db import migrations

def prepare_entertainment_tickets(apps, schema_editor):
    # 通过apps获取历史模型，避免直接导入导致的兼容性问题
    Ticket = apps.get_model('your_app_name', 'Ticket')
    EntertainmentTicket = apps.get_model('your_app_name', 'EntertainmentTicket')
    MovieTicket = apps.get_model('your_app_name', 'MovieTicket')

    # 遍历所有现有MovieTicket实例，创建对应EntertainmentTicket
    for movie_ticket in MovieTicket.objects.all():
        original_ticket = movie_ticket.ticket_ptr
        # 用get_or_create避免重复创建
        EntertainmentTicket.objects.get_or_create(ticket_ptr_id=original_ticket.id)

def link_movie_to_entertainment(apps, schema_editor):
    MovieTicket = apps.get_model('your_app_name', 'MovieTicket')
    EntertainmentTicket = apps.get_model('your_app_name', 'EntertainmentTicket')

    # 将MovieTicket关联到对应的EntertainmentTicket
    for movie_ticket in MovieTicket.objects.all():
        ent_ticket = EntertainmentTicket.objects.get(ticket_ptr_id=movie_ticket.ticket_ptr_id)
        movie_ticket.entertainmentticket_ptr = ent_ticket
        movie_ticket.save()

class Migration(migrations.Migration):
    dependencies = [
        # 替换为实际的EntertainmentTicket迁移编号，比如('your_app_name', '0002_entertainmentticket')
        ('your_app_name', '0002_entertainmentticket'),
    ]

    operations = [
        migrations.RunPython(prepare_entertainment_tickets),
        migrations.RunPython(link_movie_to_entertainment),
    ]

运行这个数据迁移：

python manage.py migrate

步骤3：修改MovieTicket的继承关系 #

将MovieTicket的父类从Ticket改为EntertainmentTicket：

class MovieTicket(EntertainmentTicket):
    # MovieTicket专属逻辑

步骤4：生成并运行继承关系修改的迁移 #

生成新的迁移文件，Django会自动处理表结构变化（添加entertainmentticket_ptr_id字段）：

python manage.py makemigrations

运行最终迁移：

python manage.py migrate

关键注意事项 #

• 备份数据库：迁移前务必备份数据，避免意外数据丢失
• 多态验证：迁移完成后可通过Ticket.objects.all()查询，验证返回实例类型是否正确
• 迁移依赖：确保数据迁移的依赖项是EntertainmentTicket的初始迁移，避免执行顺序错误

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