from rest_framework import viewsets
from .models import Role
from .serializers import RoleSerializer
from django_filters import rest_framework
from utils.paginator import BasicSetPagination


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

    class Meta:
        model = Role
        fields = "__all__"


class RoleViewSet(viewsets.ModelViewSet):
    queryset = Role.objects.all()  # 查询集
    serializer_class = RoleSerializer  # 序列化器
    pagination_class = BasicSetPagination
    filter_backends = [ 
        rest_framework.DjangoFilterBackend,
    ]  # 过滤器
    filterset_class = RoleFilter