WTOAAM/wtoaamapi/apps/views.py

from django.shortcuts import render

# Create your views here.
from rest_framework import viewsets
from drf_yasg.utils import swagger_auto_schema
from .models import *
from .serializers import *

class CategoryViewSet(viewsets.ModelViewSet):
    queryset = Category.objects.all()
    serializer_class = CategorySerializer
    
    # 应用于整个视图集的动作
    @swagger_auto_schema(tags=['Categories'])
    def get_serializer_class(self):
        return super().get_serializer_class()


class ProductViewSet(viewsets.ModelViewSet):
    queryset = Product.objects.all()
    serializer_class = ProductSerializer

    # 应用于整个视图集的动作
    @swagger_auto_schema(tags=['Products'])
    def get_serializer_class(self):
        return super().get_serializer_class()