80 lines
3.7 KiB
Python
80 lines
3.7 KiB
Python
from django.shortcuts import render
|
|
from rest_framework.viewsets import ModelViewSet
|
|
from .models import Standard, UnitType, ImplementRule,EvaluationItem,Evaluations
|
|
from .serializers import StandardSerializer, ImplementRuleSerializer, UnitTypeSerializer,EvaluationItemListSerializer, ImplementRuleListSerializer,EvaluationItemSerializer,EvaluationsSerializer
|
|
from apps.system.permission_data import RbacFilterSet
|
|
from apps.system.mixins import CreateUpdateCustomMixin, OptimizationMixin
|
|
# Create your views here.
|
|
|
|
class StandardViewSet(CreateUpdateCustomMixin, ModelViewSet):
|
|
perms_map = {'get': '*', 'post': 'standard_create',
|
|
'put': 'standard_update', 'delete': 'standard_delete'}
|
|
queryset = Standard.objects.all()
|
|
serializer_class = StandardSerializer
|
|
search_fields = ['name', 'code', 'status', 'implement_date']
|
|
filterset_fields = ['status']
|
|
ordering = ['-create_time']
|
|
|
|
class ImplementRuleViewSet(CreateUpdateCustomMixin, OptimizationMixin, ModelViewSet):
|
|
perms_map = {'get': '*', 'post': 'implementrule_create',
|
|
'put': 'implementrule_update', 'delete': 'implementrule_delete'}
|
|
queryset = ImplementRule.objects.all()
|
|
serializer_class = ImplementRuleSerializer
|
|
search_fields = ['name', 'code', 'cert_field']
|
|
filterset_fields = ['pv_scope', 'cert_field', 'cert_field__code', 'ccc_list']
|
|
ordering = ['-create_time']
|
|
|
|
def get_serializer_class(self):
|
|
# 根据请求类型动态变更serializer
|
|
if self.action == 'list':
|
|
return ImplementRuleListSerializer
|
|
return ImplementRuleSerializer
|
|
|
|
class UnitTypedViewSet(CreateUpdateCustomMixin, OptimizationMixin, ModelViewSet):
|
|
perms_map = {'get': '*', 'post': 'unittype_create',
|
|
'put': 'unittype_update', 'delete': 'unittype_delete'}
|
|
queryset = UnitType.objects.all()
|
|
serializer_class = UnitTypeSerializer
|
|
search_fields = ['name']
|
|
filterset_fields = ['name','implementrule']
|
|
ordering = ['-create_time']
|
|
|
|
def paginate_queryset(self, queryset):
|
|
"""
|
|
如果查询参数里没有page且含有参数implementrule则不分页,否则请求分页
|
|
"""
|
|
if ((not self.request.query_params.get('page', None)) and (self.request.query_params.get('implementrule', None))) or (self.paginator is None):
|
|
return None
|
|
return self.paginator.paginate_queryset(queryset, self.request, view=self)
|
|
class EvaluationItemViewSet(CreateUpdateCustomMixin, ModelViewSet):
|
|
perms_map = {'get': '*', 'post': 'evaluationitem_create',
|
|
'put': 'evaluationitem_update', 'delete': 'evaluationitem_delete'}
|
|
queryset = EvaluationItem.objects.all()
|
|
serializer_class = EvaluationItemSerializer
|
|
search_fields = ['cert_field']
|
|
ordering = ['-create_time']
|
|
def get_serializer_class(self):
|
|
# 根据请求类型动态变更serializer
|
|
if self.action == 'list':
|
|
return EvaluationItemListSerializer
|
|
return EvaluationItemSerializer
|
|
class EvaluationsViewSet(CreateUpdateCustomMixin, ModelViewSet):
|
|
"""
|
|
培训记录-增删改查
|
|
"""
|
|
|
|
perms_map = {'get': '*', 'post': 'evaluations_create',
|
|
'put': 'evaluations_update', 'delete': 'evaluations_delete'}
|
|
queryset = Evaluations.objects.all()
|
|
serializer_class = EvaluationsSerializer
|
|
search_fields = ['content']
|
|
filterset_fields = ['evItem','content']
|
|
ordering = ['-create_time']
|
|
def paginate_queryset(self, queryset):
|
|
|
|
if ((not self.request.query_params.get('page', None)) and (self.request.query_params.get('evItem', None))) or (self.paginator is None):
|
|
return None
|
|
return self.paginator.paginate_queryset(queryset, self.request, view=self)
|
|
|
|
|
|
|