Django模型继承下relative_name反向关联与序列化扩展问题

一、修正模型定义顺序（关键前置） #

原代码中A继承B但B定义在A之后，会触发Python语法错误，需调整顺序并使用字符串引用避免循环依赖：

from django.db import models

class B(models.Model):
    owner = models.ForeignKey('A', null=True, blank=True, related_name='sites')
    name = models.CharField(max_length=255)
    street1 = models.CharField(max_length=255)

class A(B):
    designation = models.CharField(max_length=255)

class C(models.Model):
    name = models.CharField(max_length=255)
    relation_a = models.ForeignKey(A, related_name='all_elements', null=True, blank=True)
    relation_b = models.ForeignKey(B, related_name='elements', null=True, blank=True)

二、解决A实例直接获取关联C的问题 #

Django多表继承下，A和B是两张独立表，A通过b_ptr字段与对应的B实例一对一关联。在A模型中添加属性代理B的反向关联：

class A(B):
    designation = models.CharField(max_length=255)

    @property
    def elements(self):
        # 直接代理对应B实例的elements反向查询集
        return self.b_ptr.elements.all()

现在可直接通过A.objects.get(id=xxx).elements.all()获取A作为B实例关联的所有C对象，无需间接查询B表。

三、DRF序列化器支持扩展elements字段 #

基于你使用的expandable_fields（推测为DRF扩展功能，如drf-extensions的可扩展序列化器），调整序列化器定义，让A_Serializer继承B_Serializer后保留elements的扩展能力：

1. 基础B序列化器 #

from rest_framework import serializers
from rest_framework_extensions.serializers import ExpandableSerializerMixin

class C_Serializer(serializers.ModelSerializer):
    class Meta:
        model = C
        fields = ['id', 'name']

class B_Serializer(ExpandableSerializerMixin, serializers.ModelSerializer):
    class Meta:
        model = B
        fields = ['id', 'name', 'street1', 'owner', 'elements']
        expandable_fields = {
            'elements': (C_Serializer, {'many': True})
        }

2. A序列化器继承并扩展 #

class A_Serializer(ExpandableSerializerMixin, B_Serializer):
    # 重新绑定elements字段到A模型的elements属性
    elements = serializers.PrimaryKeyRelatedField(
        many=True,
        read_only=True,
        source='elements'
    )

    class Meta(B_Serializer.Meta):
        model = A
        # 合并B的字段与A的专属字段
        fields = B_Serializer.Meta.fields + ['designation', 'all_elements', 'sites']
        # 合并扩展字段并添加A的专属扩展
        expandable_fields = {
            **B_Serializer.Meta.expandable_fields,
            'sites': (B_Serializer, {'many': True}),
            'all_elements': (C_Serializer, {'many': True}),
            'elements': (C_Serializer, {'many': True})
        }

四、实现嵌套扩展能力 #

现在可以通过请求参数实现多级扩展：

• 获取A对象并扩展自身关联的C：GET /a/xxx/?expand=elements
• 获取A对象并扩展所有站点（B）：GET /a/xxx/?expand=sites
• 获取A对象并扩展所有站点及每个站点关联的C：GET /a/xxx/?expand=sites.elements
• 同时扩展多种关联：GET /a/xxx/?expand=all_elements,elements,sites.elements

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