Django移除模型继承时保留原有ID序列与外键的解决方案

问题背景 #

现有两个存在多表继承关系的Django模型：

class Drink(models.Model):
    # 其他业务字段
    ....

class Juice(Drink):
    # 其他业务字段
    ....

当前采用Django默认多表继承机制：父表Drink持有自增主键ID，子表Juice通过隐式生成的drink_ptr_id外键字段关联父表对应记录。目标是移除二者的继承关系，将Juice改为直接继承models.Model的独立模型，同时完整保留原有ID序列、存量数据和关联外键关系，全程不触发字段冲突、主键报错，无需清空存量数据。

分步操作流程 #

操作前必须对数据库做全量备份，避免操作失误导致数据丢失。

• 第一步：临时调整模型定义，避免自动生成冲突字段
  不要直接将Juice的父类修改为models.Model后就执行makemigrations，先手动显式声明主键字段，把隐式生成的关联指针字段摆到明面上，模型临时调整为如下形式：

  class Juice(models.Model):
      id = models.OneToOneField(Drink, on_delete=models.CASCADE, parent_link=True, primary_key=True)
      # 原样保留Juice类下所有原有业务字段
      ....
  

• 第二步：生成初始迁移文件并手动修改字段逻辑
  执行python manage.py makemigrations生成第一个迁移文件，不要直接执行migrate，打开生成的迁移文件做两处修改：
  1. 把自动生成的id字段从OneToOneField类型改为和Drink表主键完全一致的自增字段类型（默认是AutoField，如果项目全局配置了BigAutoField就对应替换）：

     # 自动生成的原字段定义
     migrations.AddField(
         model_name='juice',
         name='id',
         field=models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to='对应app名.drink'),
         preserve_default=False,
     ),
     # 修改为自增主键定义
     migrations.AddField(
         model_name='juice',
         name='id',
         field=models.AutoField(primary_key=True, serialize=False),
         preserve_default=False,
     ),
     

  2. 在AddField操作之后追加一个RunPython操作，批量把原drink_ptr_id的值同步到新的id主键字段：

     from django.db.models import F
     
     def copy_origin_id(apps, schema_editor):
         Juice = apps.get_model('对应app名', 'Juice')
         Juice.objects.update(id=F('drink_ptr_id'))
     
     # 追加到operations列表中
     migrations.RunPython(copy_origin_id),
     

• 第三步：清理冗余字段，校准自增序列起始值
  继续在迁移文件的operations列表追加两个操作：
  1. 删除原继承关系生成的冗余drink_ptr字段

     migrations.RemoveField(
         model_name='juice',
         name='drink_ptr',
     ),
     

  2. 根据使用的数据库类型，追加对应SQL把Juice表的自增ID序列起始值调整为当前表内最大ID+1，避免后续新增数据触发主键冲突：

     # PostgreSQL用这个
     migrations.RunSQL("SELECT setval(pg_get_serial_sequence('对应app名_juice','id'), COALESCE((SELECT MAX(id) FROM 对应app名_juice), 1));"),
     # MySQL用这个
     # migrations.RunSQL("ALTER TABLE 对应app名_juice AUTO_INCREMENT = (SELECT MAX(id)+1 FROM 对应app名_juice);"),
     # SQLite用这个
     # migrations.RunSQL("UPDATE sqlite_sequence SET seq = (SELECT MAX(id) FROM 对应app名_juice) WHERE name='对应app名_juice';"),
     

• 第四步：执行迁移，完成模型最终清理
  执行python manage.py migrate，确认迁移无报错执行完成后，回到models.py，删掉Juice类里临时添加的id字段定义，只保留class Juice(models.Model)和Juice自身的业务字段即可：

  class Juice(models.Model):
      # 仅保留Juice自身的业务字段，无需显式定义id，Django会自动生成默认自增主键
      ....
  

  最后再执行一次python manage.py makemigrations，此时Django只会检测到id字段从显式定义转为隐式自动生成，生成的迁移不会涉及表结构变更，执行python manage.py migrate后全部操作完成。

关联数据说明 #

项目中其他模型如果存在指向Juice的外键，由于Juice的主键值和原drink_ptr_id完全一致，外键关联关系不会错乱，无需额外修改数据。如果存在指向Drink模型、实际关联Juice记录的外键，可根据业务需求手动调整关联指向即可。
如果存量数据量超过百万级，RunPython同步ID的步骤可以改成分批迭代更新，避免单事务锁表时间过长影响线上业务。

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