72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
from apps.dpm.serializers import RiskPointCreateUpdateSerializer, RiskPointSerializer, RiskCreateUpdateSerializer, RiskSerializer, CheckTaskSetCreateUpdateSerializer, CheckTaskSetSerializer, CheckWork_10_CreateUpdateSerializer, CheckWorkSerializer
|
|
from apps.utils.viewsets import CustomModelViewSet
|
|
from apps.dpm.models import CheckTaskSet, CheckWork, RiskPoint, Risk
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from django.db import transaction
|
|
|
|
|
|
# Create your views here.
|
|
class RiskPointViewSet(CustomModelViewSet):
|
|
queryset = RiskPoint.objects.all()
|
|
create_serializer_class = RiskPointCreateUpdateSerializer
|
|
update_serializer_class = RiskPointCreateUpdateSerializer
|
|
serializer_class = RiskPointSerializer
|
|
select_related_fields = ['area', 'dept_res', 'user_res']
|
|
search_fields = ['name']
|
|
filterset_fields = ['area', 'type', 'level', 'manage_level']
|
|
|
|
|
|
class RiskViewSet(CustomModelViewSet):
|
|
queryset = Risk.objects.all()
|
|
create_serializer_class = RiskCreateUpdateSerializer
|
|
update_serializer_class = RiskCreateUpdateSerializer
|
|
serializer_class = RiskSerializer
|
|
select_related_fields = ['riskpoint']
|
|
prefetch_related_fields = ['results']
|
|
search_fields = ['name']
|
|
filterset_fields = ['results']
|
|
|
|
|
|
class CheckTaskSetViewSet(CustomModelViewSet):
|
|
queryset = CheckTaskSet.objects.all()
|
|
create_serializer_class = CheckTaskSetCreateUpdateSerializer
|
|
update_serializer_class = CheckTaskSetCreateUpdateSerializer
|
|
serializer_class = CheckTaskSetSerializer
|
|
select_related_fields = ['periodictask']
|
|
search_fields = ['note']
|
|
filterset_fields = ['periodictask__enabled']
|
|
|
|
@action(methods=['put'], detail=True, perms_map={'put': 'checktaskset.update'})
|
|
def toggle(self, request, pk=None):
|
|
"""修改启用禁用状态
|
|
|
|
修改启用禁用状态
|
|
"""
|
|
obj = self.get_object()
|
|
periodictask = obj.periodictask
|
|
periodictask.enabled = False if periodictask.enabled else True
|
|
periodictask.save()
|
|
return Response()
|
|
|
|
def perform_destroy(self, instance):
|
|
with transaction.atomic():
|
|
periodictask = instance.periodictask
|
|
instance.delete()
|
|
periodictask.enabled = False
|
|
periodictask.save()
|
|
|
|
|
|
class CheckWorkViewSet(CustomModelViewSet):
|
|
queryset = CheckWork.objects.all()
|
|
create_serializer_class = CheckWork_10_CreateUpdateSerializer
|
|
update_serializer_class = CheckWork_10_CreateUpdateSerializer
|
|
serializer_class = CheckWorkSerializer
|
|
search_fields = ['name', 'description']
|
|
select_related_fields = ['category', 'user_duty', 'riskpoint']
|
|
filterset_fields = ['type', 'category', 'riskpoint']
|
|
|
|
|
|
class HazardViewSet(CustomModelViewSet):
|
|
pass
|