動態設置serializer和permission獲取用戶信息

1、用戶詳情的序列化

users/serializers.py

class UserDetailSerializer(serializers.ModelSerializer):
    """
    用戶詳情
    """
    class Meta:
        model = User
        fields = ['name', 'birthday', 'sex', 'phone', 'email']

2、users/views.py

class UserViewSet(mixins.CreateModelMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
    """
    create:
        建立用戶
    retrieve:
        用戶詳情
    """
    serializer_class = UserRegSerializer
    queryset = User.objects.all()
    fields = ['username', 'password', 'phone']

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = self.perform_create(serializer)
        re_dict = serializer.data
        payload = jwt_payload_handler(user)
        re_dict["token"] = jwt_encode_handler(payload)
        re_dict["name"] = user.name if user.name else user.username

        headers = self.get_success_headers(serializer.data)
        return Response(re_dict, status=status.HTTP_201_CREATED, headers=headers)

    def perform_create(self, serializer):
        return serializer.save()

    def get_permissions(self):
        """
        動態權限配置
        :return: 空列表/IsAuthenticated
        """
        if self.action == 'retrieve':
            return [IsAuthenticated()]
        return []

    def get_serializer_class(self):
        """
        動態選擇序列化的方式
        :return: 序列化對象
        """
        if self.action == 'create':
            return UserRegSerializer
        return UserDetailSerializer

    def get_object(self):
        """
        獲取具體的那個用戶
        :return: 用戶
        """
        return self.request.user

 

如今就能夠從數據庫獲取用戶的信息了