from rest_framework import viewsets, mixins
from .models import AbilityReview, QualityCommendation, QualityActivities, Contact, ExternalAuditors
from .serializers import AbilityReviewSerializer, \
    QualityCommendationSerializer, \
        QualityActivitiesSerializer,\
             ContactSerializer, \
                ExternalAuditorsSerializer 


class AbilityReviewViewSet( mixins.CreateModelMixin, 
                            mixins.ListModelMixin, 
                            mixins.DestroyModelMixin, 
                            mixins.UpdateModelMixin, 
                            viewsets.GenericViewSet):
    queryset = AbilityReview.objects.all()
    serializer_class = AbilityReviewSerializer

    #自定义查询
    def get_queryset(self):
        pass


class QualityCommendationBViewSet(mixins.CreateModelMixin, 
                        mixins.ListModelMixin, 
                        mixins.DestroyModelMixin, 
                        mixins.UpdateModelMixin, 
                        viewsets.GenericViewSet):
    queryset = QualityCommendation.objects.all()
    serializer_class = QualityCommendationSerializer


class QualityActivitiesViewSet(mixins.CreateModelMixin, 
                        mixins.ListModelMixin, 
                        mixins.DestroyModelMixin, 
                        mixins.UpdateModelMixin, 
                        viewsets.GenericViewSet):
    queryset = QualityActivities.objects.all()
    serializer_class = QualityActivitiesSerializer


class ContactViewSet(mixins.CreateModelMixin, 
            mixins.ListModelMixin, 
            mixins.DestroyModelMixin, 
            mixins.UpdateModelMixin, 
            viewsets.GenericViewSet):
    queryset = Contact.objects.all()
    serializer_class = ContactSerializer


class ExternalAuditorsViewSet(mixins.CreateModelMixin, 
                        mixins.ListModelMixin, 
                        mixins.DestroyModelMixin, 
                        mixins.UpdateModelMixin, 
                        viewsets.GenericViewSet):
    queryset = ExternalAuditors.objects.all()
    serializer_class = ExternalAuditorsSerializer