Django+DRF+Vue+Mysql+Redis OUC软件工程作业

from django_filters.rest_framework import DjangoFilterBackend
from drf_haystack.viewsets import HaystackViewSet
from rest_framework.filters import OrderingFilter
from rest_framework.generics import ListAPIView, CreateAPIView, DestroyAPIView, RetrieveAPIView
from rest_framework.pagination import PageNumberPagination
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework.viewsets import GenericViewSet
from MyApps.goods import models
from MyApps.goods.models import SKU, GoodsCategory, Collect, Comment1, SKUImage
from MyApps.goods.serializers import SKUSerializer, SKUIndexSerializer, CollectSerializer, Comment1Serializer
from rest_framework.permissions import IsAuthenticated
from django_redis import get_redis_connection
from rest_framework.filters import BaseFilterBackend
from django_filters import FilterSet, filters
class SKUListView(ListAPIView):
    """
    sku列表数据
    """
    serializer_class = SKUSerializer
    # 分页与排序
    filter_backends = (OrderingFilter,)
    ordering_fields = ("create_time", "price", "sales")
    def get_queryset(self):
        category_id = self.kwargs["category_id"]
        return SKU.objects.filter(category_id=category_id, is_launched=True)
class SKUSearchViewSet(HaystackViewSet):
    """
    SKU搜索
    """
    index_models = [SKU]
    serializer_class = SKUIndexSerializer
    # 分页与排序
    # filter_backends = (OrderingFilter,)
    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset())
        ordering_fields = ("create_time", "price", "sales")
        od = request.query_params.get("ordering", None)
        if od in ordering_fields:
            queryset = queryset.order_by(od)
        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.get_serializer(page, many=True)
            return self.get_paginated_response(serializer.data)
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)
class CategoriesView(APIView):
    """获取当前分类信息"""
    def get(self, request, pk):
        """
        1.获取前端数据
        2. 查询当前三级分类信息
        3.通过三级分类信息获取一二集分类
        4. 返回
        :param request:
        :return:
        """
        cat3 = GoodsCategory.objects.get(id=pk)  # 获取三级
        cat2 = cat3.parent  # 自关联获取二级,
        cat1 = cat2.parent  # 自关联获取一级
        # 返回数据
        return Response({
            "cat1": cat1.name,
            "cat2": cat2.name,
            "cat3": cat3.name
        })
class GetGoodsView(APIView):
    """获取当前分类信息"""
    def get(self, request, pk):
        SKU1 = models.SKU.objects.filter(pk=pk).first()
        comment_count = SKU1.goods.comment_count
        collect_count = SKU1.goods.collect_count
        goods_id = SKU1.goods_id
        # 返回数据
        return Response({"goods_id": goods_id, 'comment_count': comment_count, 'collect_count': collect_count})
class IsCollectedView(APIView):
    def post(self, request):
        print(request.data)
        sku_id = int(request.data.get("sku_id", -5))
        user_id = request.data.get("user_id", 1)
        if sku_id == -5:
            return Response({"res_data": 0})
        collecs = models.Collect.objects.filter(user_id=user_id)
        sku_id_list = [i.sku_id for i in collecs]
        print(sku_id in sku_id_list)
        res = 0
        if sku_id in sku_id_list:
            res = 1
        return Response({"res_data": res})
##################################
class MyPageNumberPagination(PageNumberPagination):
    page_size_query_param = 'size'
    page_size = 10
    max_page_size = 100
class CommentFilterSet(FilterSet):
    goods = filters.NumberFilter(field_name='goods', required=True)
    class Meta:
        model = models.Comment1
        fields = ['goods']
class Comment1View(ListAPIView, GenericViewSet):
    filter_backends = [DjangoFilterBackend]
    filterset_class = CommentFilterSet
    queryset = models.Comment1.objects
    serializer_class = Comment1Serializer
################################################
class SelfFilterBackend(BaseFilterBackend):
    def filter_queryset(self, request, queryset, view):
        return queryset.filter(user=request.user)
class CollectViewSet(ListAPIView, CreateAPIView, GenericViewSet):
    permission_classes = [IsAuthenticated]
    filter_backends = [SelfFilterBackend, DjangoFilterBackend]
    pagination_class = MyPageNumberPagination
    # 当前登录用户的所有收藏记录
    queryset = models.Collect.objects
    serializer_class = CollectSerializer
    def perform_create(self, serializer):
        user = self.request.user
        instance = Collect.objects.filter(user=user, **serializer.validated_data).first()
        if not instance:
            instance = serializer.save(user=user)
            instance.sku.goods.collect_count += 1
            instance.sku.goods.save()
            return Response({"message": 'ok', 'data': {'active': True}})
        else:
            instance.delete()
            instance.sku.goods.collect_count -= 1
            instance.sku.goods.save()
            return Response({"message": 'ok', 'data': {'active': False}})
class HotSkus(ListAPIView):
    serializer_class = SKUSerializer
    # 分页与排序
    filter_backends = (OrderingFilter,)
    ordering_fields = ("sales", "create_time", "price")
    def get_queryset(self):
        category_id = self.kwargs["category_id"]
        return SKU.objects.filter(category_id=category_id, is_launched=True).order_by('-sales')[:2]
class ImgListView(APIView):
    def get(self, request):
        s_id = request.query_params.get("sku_id")
        img_list = SKUImage.objects.filter(sku_id=s_id)
        img_list = [i.image.path.split("BMyStore\\")[1] for i in img_list]
        return Response({"img_list": img_list})