個人中心功能開發

drf文件註釋
http://www.django-rest-framework.org/topics/documenting-your-api/

動態設定serializer和permission獲取使用者資訊

獲取詳情只需要新增一個mixins.RetrieveModelMixin,就行了

使用者詳情的序列化
users/serializers.py

class UserDetailSerializer(serializers.ModelSerializer):
    """
    使用者詳情
    """
    class Meta:
        model = User
        fields = ("name", "gender", "birthday", "email","mobile")
 

views.py

class UserViewset(mixins.CreateModelMixin, mixins.RetrieveModelMixin, mixins.UpdateModelMixin, viewsets.GenericViewSet):
    '''
    使用者資訊管理
    create:
        使用者註冊
    retrieve:
        個人資訊
    update:
        修改個人資訊
    '''
    queryset = User.objects.all()
    authentication_classes = (JSONWebTokenAuthentication, authentication.SessionAuthentication)

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = self.perform_create(serializer)

        # 註冊成功直接生成token,自動登陸
        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)

        # 返回的不是serializer.data,而是我們自己寫的re_dict
        return Response(re_dict, status=status.HTTP_201_CREATED, headers=headers)

    # 這裡需要動態許可權配置
    # 1.使用者註冊的時候不應該有許可權限制
    # 2.當想獲取使用者詳情資訊的時候，必須登入才行
    def get_permissions(self):
        if self.action == "retrieve":
            return [permissions.IsAuthenticated(), ]
        elif self.action == "create":
            return []

        return []

    # 這裡需要動態選擇用哪個序列化方式
    # 1.UserRegSerializer（使用者註冊），只返回username和mobile，會員中心頁面需要顯示更多欄位，所以要建立一個UserDetailSerializer
    # 2.問題又來了，如果註冊的使用userdetailSerializer，又會導致驗證失敗，所以需要動態的使用serializer
    def get_serializer_class(self):
        if self.action == "retrieve":
            return UserDetailSerializer
        elif self.action == "create":
            return UserRegSerializer

        return UserDetailSerializer

    # 雖然繼承了Retrieve可以獲取使用者詳情，但是並不知道使用者的id，所有要重寫get_object方法
    # 重寫get_object方法，就知道是哪個使用者了
    def get_object(self):
        return self.request.user

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

使用者個人資訊修改,只需要繼承mixins.UpdateModelMixin就可以了

使用者收藏

user_operation/serializer.py

class UserFavDetailSerializer(serializers.ModelSerializer):
    '''
    使用者收藏詳情
    '''

    # 通過商品id獲取收藏的商品，需要巢狀商品的序列化
    goods = GoodsSerializer()

    class Meta:
        model = UserFav
        fields = ("goods", "id")
 

views.py

class UserFavViewset(viewsets.GenericViewSet, mixins.ListModelMixin, mixins.CreateModelMixin, mixins.DestroyModelMixin):
    '''
    list:
        獲取使用者的所有收藏
    create:
        新增收藏
    destroy:
        取消收藏
    '''
    # permission是用來做許可權判斷的
    # IsAuthenticated：必須登入使用者；IsOwnerOrReadOnly：必須是當前登入的使用者
    permission_classes = (IsAuthenticated, IsOwnerOrReadOnly)
    # auth使用來做使用者認證的
    authentication_classes = (JSONWebTokenAuthentication, SessionAuthentication)
    # 搜尋的欄位
    lookup_field = 'goods_id'

    def get_queryset(self):
        # 只能檢視當前登入使用者的收藏，不會獲取所有使用者的收藏
        return UserFav.objects.filter(user=self.request.user)

    # 動態選擇serializer
    def get_serializer_class(self):
        if self.action == "list":
            return UserFavDetailSerializer
        elif self.action == "create":
            return UserFavSerializer
        return UserFavSerializer

使用者留言功能

user_operation/serializers.py

class LeavingMessageSerializer(serializers.ModelSerializer):
    '''
    使用者留言
    '''
    # 獲取當前登入的使用者
    user = serializers.HiddenField(
        default=serializers.CurrentUserDefault()
    )
    # read_only:只返回，post時候可以不用提交，format：格式化輸出
    add_time = serializers.DateTimeField(read_only=True, format='%Y-%m-%d %H:%M')

    class Meta:
        model = UserLeavingMessage
        fields = ("user", "message_type", "subject", "message", "file", "id", "add_time")

views.py

class LeavingMessageViewset(mixins.ListModelMixin, mixins.DestroyModelMixin, mixins.CreateModelMixin,
                            viewsets.GenericViewSet):
    """
    list:
        獲取使用者留言
    create:
        新增留言
    delete:
        刪除留言功能
    """

    permission_classes = (IsAuthenticated, IsOwnerOrReadOnly)
    authentication_classes = (JSONWebTokenAuthentication, SessionAuthentication)
    serializer_class = LeavingMessageSerializer

    # 只能看到自己的留言
    def get_queryset(self):
        return UserLeavingMessage.objects.filter(user=self.request.user)

urls.py

# 配置使用者留言的url
router.register(r'messages', LeavingMessageViewset, base_name="messages")

使用者收穫地址

user_operation/serializers.py

class AddressSerializer(serializers.ModelSerializer):
    user = serializers.HiddenField(
        default=serializers.CurrentUserDefault()
    )
    add_time = serializers.DateTimeField(read_only=True, format='%Y-%m-%d %H:%M')

    class Meta:
        model = UserAddress
        fields = ("id", "user", "province", "city", "district", "address", "signer_name", "add_time", "signer_mobile")

views.py

class AddressViewset(viewsets.ModelViewSet):
    """
    收貨地址管理
    list:
        獲取收貨地址
    create:
        新增收貨地址
    update:
        更新收貨地址
    delete:
        刪除收貨地址
    """
    permission_classes = (IsAuthenticated, IsOwnerOrReadOnly)
    authentication_classes = (JSONWebTokenAuthentication, SessionAuthentication)
    serializer_class = AddressSerializer

    def get_queryset(self):
        return UserAddress.objects.filter(user=self.request.user)

urls.py

# 配置收貨地址
router.register(r'address',AddressViewset , base_name="address")