from rest_framework.viewsets import GenericViewSet
from utils.pagination import PageOrNot
from rest_framework.mixins import ListModelMixin
from .models import *
from .serializers import *

from apps.project.models import CertApp
# Create your views here.
class CertaccessViewSet(PageOrNot, ListModelMixin, GenericViewSet):
    """
    业务评定
    """
    perms_map = {'get': 'access_view'}
    serializer_class = CertappSerializerX
    filterset_fields = []
    ordering = ['-create_time']

    def get_queryset(self):
        queryset = CertApp.objects.filter(state__in=['产品检测', '评定', '出证'])
        if hasattr(self.get_serializer_class(), 'setup_eager_loading'):
            queryset = self.get_serializer_class().setup_eager_loading(queryset)
        return queryset