Django中继承模型的外键反向查询实现方法

你现在用的是Django默认的多表继承模式：Child1、Child2这类子类会单独生成自己的数据表，靠隐式创建的一对一字段和Parent表关联，而指向Foo的外键是存在Parent表里的。默认的反向接口foo.parent_set只会查Parent自身表的记录，不会自动关联子类表做类型转换，自然拿不到子类的实例。

方案1：纯原生Django实现，无需额外依赖 #

核心思路是先查出所有和当前Foo关联的Parent记录，提前预取好子类的关联数据避免N+1查询，再逐条判断每条记录实际对应的子类类型：

# 先预取所有子类的反向关联，把多次单条查询合并成批量查询
related_parents = foo.parent_set.prefetch_related("child1", "child2").all()

result = []
for parent in related_parents:
    # 遍历所有子类匹配当前记录的实际类型
    for child_cls in [Child1, Child2]:
        rel_attr = child_cls.__name__.lower()
        if hasattr(parent, rel_attr):
            result.append(getattr(parent, rel_attr))
            break
    else:
        # 匹配不到子类就是Parent本身的直接实例
        result.append(parent)

注意：这个方案的缺点是后续新增子类的时候，必须手动更新预取字段列表和子类判断列表，子类越多维护越麻烦。

方案2：用InheritanceManager实现自动类型转换（推荐） #

django-model-utils封装的InheritanceManager专门处理多表继承的查询问题，不用自己写子类判断逻辑：

1. 先给Parent模型挂载对应的管理器：

from model_utils.managers import InheritanceManager

class Parent(models.Model):
    foreign_key = models.ForeignKey(Foo, on_delete=models.CASCADE)
    objects = InheritanceManager()

2. 查询的时候链式调用select_subclasses()就行，返回的结果会自动转换成对应的子类实例，Parent本身的实例也会正常返回：

# 一行代码拿到所有关联的Parent、Child1、Child2实例
result = list(foo.parent_set.select_subclasses())

这个方法会自动关联所有子类的表，后续新增子类不用改现有查询代码。如果只需要拿特定子类的结果，给select_subclasses()传对应子类类名当参数就行，比如select_subclasses(Child1)就只会返回Parent和Child1类型的对象。

如果你的业务里不需要实例化Parent本身，可以把Parent改成抽象基类：

class Parent(models.Model):
    foreign_key = models.ForeignKey(Foo, on_delete=models.CASCADE)
    class Meta:
        abstract = True

这种模式下外键会直接存在每个子类自己的表里，不会生成Parent的独立数据表，缺点是没有统一的parent_set入口，需要分别查询每个子类的反向结果再合并，适合不需要操作Parent通用实例的场景。

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