如何在序列化程序中返回正确的 model 字段？

Question

我正在实现一个返回用户和帖子的搜索栏。 我能够返回数据，但是当我清除搜索栏时，我得到返回的错误：

AttributeError：尝试在序列化程序SearchUserSerializer上获取字段username的值时出现 AttributeError。 序列化程序字段的命名可能不正确，并且与Post实例上的任何属性或键都不匹配。 原始异常文本为：“Post”object 没有属性“username”。

我的模特：

class User(AbstractUser):
    avi_pic = models.ImageField(
        max_length=400, null=True, blank=True, upload_to='images')
    name = models.CharField(max_length=50, blank=True, null=True)
    username = models.CharField(max_length=30, unique=True)

class Post(models.Model):
    user = models.ForeignKey(
        User, on_delete=models.CASCADE, default=None
    )
    cover = models.CharField(max_length=300, default='', blank=True)
    title = models.CharField(max_length=300, default='', blank=True)
    date = models.DateTimeField(editable=False, auto_now_add=True)

我的序列化程序：

class SearchPostSerializer(serializers.ModelSerializer):
    username = serializers.SerializerMethodField()

    class Meta:
        model = Post
        fields = ('id', 'title', 'user', 'username', 'cover')

    def get_username(self, post):
        return post.user.username

class SearchUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ('id', 'name', 'username', 'avi_pic')

和我的 Views.py：

class SearchView(generics.ListAPIView):

    def get_serializer_class(self):
        queryset = self.get_queryset()
        if len(queryset) == 0:
            return None
        if isinstance(queryset[0], User):
            return SearchUserSerializer
        elif isinstance(queryset[0], Post):
            return SearchPostSerializer
        else:
            return None

    def get_queryset(self):
        query = self.request.query_params.get('q', None)
        if query is not None:
            queryset_users = User.objects.filter(
                Q(name__icontains=query) | Q(username__icontains=query))
            queryset_posts = Post.objects.filter(
                Q(playlist_title__icontains=query))
            results = list(queryset_users) + list(queryset_posts)
            return results
        else:
            return User.objects.none()

我已经尝试过很多零运气的事情。

Answer 1

尝试这个：

class SearchPostSerializer(serializers.ModelSerializer):
    username = serializers.CharField(source='user.username')
    class Meta:
        model = Post
        fields = ('id', 'title', 'user', 'username', 'cover')

这应该可以解决问题。

您可以删除get_username function

您可以在这里查看解释：

https://www.django-rest-framework.org/api-guide/fields/#source