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


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

    class Meta:
        model = MenuButton
        fields = "__all__"


class MenuButtonViewSet(viewsets.ModelViewSet):
    queryset = MenuButton.objects.all()
    serializer_class = MenuButtonSerializer
    pagination_class = BasicSetPagination
    filter_backends = [
        rest_framework.DjangoFilterBackend,
    ]  # 过滤器
    filterset_class = MenuButtonFilter