Django如何实现MyModel外键指向A或B模型的多态关系

单字段可关联A、B两个不同模型的需求有两种成熟实现方式，可根据项目场景选择：

方案1：Django原生通用外键（无第三方依赖） #

基于Django内置的contenttypes框架实现，不需要安装额外包，是官方推荐的跨模型关联方案。
实现逻辑是同时存储关联模型的类型、关联记录的主键ID，通过虚拟字段完成关联对象的读写。

1. 先确认配置：settings.py的INSTALLED_APPS中已加入django.contrib.contenttypes，Django默认生成的项目会自动开启该配置。
2. 修改模型代码如下：

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.fields import GenericForeignKey
from django.db.models import Q


class A(models.Model):
    email = models.EmailField()
    national_id = models.IntegerField()


class B(models.Model):
    name = models.CharField(max_length=255)
    age = models.IntegerField()


class MyModel(models.Model):
    name = models.CharField(max_length=255)
    # 存储关联的模型类型（标记当前关联的是A还是B）
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    # 存储关联记录的主键值
    object_id = models.PositiveIntegerField()
    # 虚拟操作字段，直接通过该字段读写关联的A/B实例
    provider = GenericForeignKey('content_type', 'object_id')

    class Meta:
        # 加约束限制只能关联A、B两个模型，避免关联到其他无关模型
        constraints = [
            models.CheckConstraint(
                check=Q(content_type__model__in=['a', 'b']),
                name='restrict_provider_model_type'
            )
        ]

使用示例 #

# 关联A模型实例
a_instance = A.objects.create(email="user@example.com", national_id=110101199001011234)
m1 = MyModel.objects.create(name="记录1", provider=a_instance)

# 关联B模型实例
b_instance = B.objects.create(name="张三", age=28)
m2 = MyModel.objects.create(name="记录2", provider=b_instance)

# 读取关联对象直接调用字段即可
print(m1.provider.email)  # 输出 user@example.com
print(m2.provider.name)   # 输出 张三

方案2：django-polymorphic多态模型（ORM体验更优） #

如果需要和普通外键一致的查询体验，比如直接跨关联表过滤字段，可以使用专门的多态模型库实现。

1. 先安装依赖：pip install django-polymorphic
2. 在settings.py的INSTALLED_APPS中加入polymorphic
3. 修改模型代码如下：

from django.db import models
from polymorphic.models import PolymorphicModel


# 定义多态基类，所有可被关联的模型都继承该类
class BaseProvider(PolymorphicModel):
    pass


class A(BaseProvider):
    email = models.EmailField()
    national_id = models.IntegerField()


class B(BaseProvider):
    name = models.CharField(max_length=255)
    age = models.IntegerField()


class MyModel(models.Model):
    name = models.CharField(max_length=255)
    # 外键直接指向多态基类，即可自动关联所有子类实例
    provider = models.ForeignKey(to=BaseProvider, on_delete=models.CASCADE)

该方案的使用方式和普通外键完全一致，支持ORM的跨表查询，比如直接执行MyModel.objects.filter(provider__name="张三")就能筛选出关联了对应B实例的记录，不需要额外做类型判断。缺点是需要引入第三方依赖，且多态基类会生成额外的数据表存储映射关系。

选型建议 #

• 项目不想引入额外依赖、关联逻辑简单，选原生通用外键即可满足需求
• 关联查询场景多、想要更顺滑的ORM使用体验，选django-polymorphic方案

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