# Create your views here.
from rest_framework import viewsets
from .models import Menu
from .serializers import MenuSerializer
from django_filters import rest_framework
from utils.paginator import BasicSetPagination


class MenuFilter(rest_framework.FilterSet):
    name = rest_framework.CharFilter(field_name="name", lookup_expr="icontains")

    class Meta:
        model = Menu
        fields = "__all__"


class MenuViewSet(viewsets.ModelViewSet):
    queryset = Menu.objects.all()
    serializer_class = MenuSerializer
    pagination_class = BasicSetPagination
    filter_backends = [
        rest_framework.DjangoFilterBackend,
    ]  # 过滤器
    filterset_class = MenuFilter