python - 如何在 Django Rest Framework 中获取模型方法的值？

Question

所以基本上我有一个 Django 模型，它有一个 ManyToManyField friend 和两个在它上面运行的方法。这是我的文件:

模型.py:

from django.db import models
from django.contrib.auth.models import User


class Profile(models.Model):
    first_name = models.CharField(max_length=50, blank=True)
    last_name = models.CharField(max_length=50, blank=True)
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    friends = models.ManyToManyField(User, blank=True, related_name='friends')

    def friends_list(self):
        return self.friends.all()

    def number_of_friends(self):
        return self.friends.all().count()

序列化器.py:

from rest_framework import serializers
from .models import Profile


class ProfileSerializer(serializers.ModelSerializer):
    class Meta:
        model = Profile
        fields = '__all__'

Views.py:

from rest_framework import viewsets, permissions
from .models import Profile
from .serializers import ProfileSerializer


class ProfileViewSet(viewsets.ModelViewSet):
    queryset = Profile.objects.all()
    permission_classes = [
        permissions.AllowAny
    ]
    serializer_class = ProfileSerializer

问题是在 Api 中，方法的返回值不存在。例如，friends_list 方法应该返回您拥有的 friend 列表，即使这在传统的 django 项目中确实有效，但 Django Rest Framework 并未显示此方法的任何值(value)。我该如何解决这个问题并让这两种方法的返回值显示在 API 中？

Answer 1

由于模型序列化程序只为序列化程序字段选择模型字段，因此您不会自动复制任何方法。

通过引用模型方法显式添加两个字段，您仍然可以通过 API 发送此只读数据

class ProfileSerializer(serializers.ModelSerializer):
    class Meta:
        model = Profile
        fields = [
            # need to explicitly define all fields I believe
            'friends_list',
            'number_of_friends',
        ]

既然声明了两个字段(匹配方法名，DRF 应该为每个创建SerializerMethodField 或ReadOnly 字段(不确定是哪一个，但它们是相似的)他们中的。它之所以有效，是因为它将这些字段的源设置为相同的名称，并且如果在模型上找到某些属性(在本例中为方法)。

如果不行，你可以

class ProfileSerializer(serializers.ModelSerializer):
    friends_list = serializers.SerializerMethodField()
    number_of_friends = serializers.SerializerMethodField()

    class Meta:
        model = Profile
        fields = [
            # need to explicitly define all fields I believe
            'friends_list',
            'number_of_friends',
        ]

    def get_friends_list(self, instance):
        return instance.friends_list()

    def get_number_of_friends(self, instance):
        return instance.number_of_friends()