Django DRF 如何序列化模型继承（读取/写入）

在本文中，我们将介绍如何使用Django Rest Framework（DRF）序列化模型继承，包括读取和写入操作。模型继承是一种常见的数据库设计模式，它允许在不重复编写相同字段的情况下创建多个相关模型。

什么是模型继承？

模型继承是指创建一个新的模型，它继承了一个或多个其他模型的字段和方法。父模型可以称为基类或抽象模型，子模型可以称为派生类或具体模型。模型继承的主要目的是实现代码重用，避免重复定义相同的字段和方法。

Django提供了三种类型的模型继承：抽象基类、多表继承和代理模型。在本文中，我们将重点关注多表继承的序列化方法。

多表继承序列化

在多表继承中，每个模型都有自己的数据库表，并且通过一个指向父模型的外键字段进行关联。为了序列化多表继承的模型，我们需要在DRF中定义适当的序列化器。

让我们以一个简单的示例开始。假设我们有两个模型：Person和Student，Student是Person的子类。每个模型都有一个名字和年龄字段，但Student还有一个major字段。

from django.db import models

class Person(models.Model):
    name = models.CharField(max_length=100)
    age = models.PositiveIntegerField()

class Student(Person):
    major = models.CharField(max_length=100)

读取操作

为了序列化多表继承的模型并进行读取操作，我们需要创建相应的序列化器。在本例中，我们将使用ModelSerializer。

from rest_framework import serializers
from .models import Person, Student

class PersonSerializer(serializers.ModelSerializer):
    class Meta:
        model = Person
        fields = ['name', 'age']

class StudentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Student
        fields = ['name', 'age', 'major']

在这里，我们定义了一个PersonSerializer和一个StudentSerializer，分别用于序列化Person和Student模型的字段。我们通过在Meta类中指定模型和字段来配置序列化器。

要进行读取操作，我们可以使用DRF提供的ListAPIView或RetrieveAPIView。例如，要获取所有学生的列表，我们可以创建一个StudentListView视图。

from rest_framework.generics import ListAPIView
from .serializers import StudentSerializer

class StudentListView(ListAPIView):
    queryset = Student.objects.all()
    serializer_class = StudentSerializer

我们在这里指定了查询集（所有学生）和序列化器（StudentSerializer）。然后，我们可以通过访问/students/来获取学生列表。

写入操作

要进行写入操作，我们需要定义一个用于创建和更新模型的序列化器，而不仅仅是用于读取。为了使major字段可写，我们需要添加它到StudentSerializer中，并覆盖create和update方法。

class StudentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Student
        fields = ['name', 'age', 'major']

    def create(self, validated_data):
        major = validated_data.pop('major')
        student = Student.objects.create(**validated_data)
        student.major = major
        student.save()
        return student

    def update(self, instance, validated_data):
        major = validated_data.pop('major')
        instance = super().update(instance, validated_data)
        instance.major = major
        instance.save()
        return instance

在这里，我们覆盖了create和update方法，以处理major字段的写入。在create方法中，我们先从validated_data中弹出major字段的值，并使用Student.objects.create创建一个新的学生。然后，我们将major字段的值赋给学生对象，并保存。在update方法中，我们先更新实例（学生对象），然后在后续的处理中再将major字段的值赋给实例并保存。