from django.shortcuts import render from rest_framework.mixins import ListModelMixin from rest_framework.viewsets import GenericViewSet, ModelViewSet from .models import * from .serializers import * from rest_framework.decorators import action from django.conf import settings from rest_framework import status from rest_framework.response import Response from apps.system.models import Organization from django.db.models import Count from utils.pagination import PageOrNot from apps.system.mixins import CreateUpdateCustomMixin from apps.system.permission_data import RbacFilterSet from django.utils import timezone from .filters import RecordFilter from django.db.models.query import QuerySet from django.core.cache import cache from apps.system.permission import get_permission_list # Create your views here. class ContentViewSet(CreateUpdateCustomMixin, ModelViewSet): """ 资料清单：增删改查 """ perms_map = {'get': '*', 'post': 'content_create', 'put': 'content_update', 'delete': 'content_delete'} queryset = Content.objects.all() serializer_class = ContentSerializer pagination_class = None search_fields = ['name', 'desc'] filterset_fields = ['type'] ordering = ['type__sort', 'create_time'] class TaskViewSet(CreateUpdateCustomMixin, ModelViewSet): perms_map = {'get': '*', 'post': 'task_create', 'put': 'task_update', 'delete': 'task_delete'} queryset = Task.objects.all() search_fields = ['name'] ordering = ['-create_time'] def get_serializer_class(self): if self.action in ['create', 'update']: return TaskCreateUpdateSerializer elif self.action == 'retrieve': return TaksDetailSerializer return TaskListSerializer def destroy(self, request, *args, **kwargs): instance = self.get_object() instance.delete() Record.objects.filter(task=instance).delete() return Response(status=status.HTTP_204_NO_CONTENT) @action(methods=['post'], detail=True, perms_map = {'post':'task_init'}) def init(self, request, *args, **kwargs): """ 初始化任务,生成记录 """ obj = self.get_object() depts = request.data['depts'] contents = request.data['contents'] if depts and contents: record_list = [] for m in depts: for n in contents: data={} data['content'] = Content.objects.get(pk=n) data['belong_dept'] = Organization.objects.get(pk=m) data['task'] = obj data['end_date'] = obj.end_date data['create_by'] = request.user record_list.append(Record(**data)) Record.objects.bulk_create(record_list) obj.state = '执行中' obj.save() return Response(status=status.HTTP_200_OK) return Response('单位或清单不能为空', status=status.HTTP_400_BAD_REQUEST) @action(methods=['get'], detail=True, perms_map = {'get':'*'}) def contents(self, request, *args, **kwargs): """ 聚合 """ obj = self.get_object() serializer = ContentListSerializer(obj.contents.distinct(), many=True) return Response(serializer.data) @action(methods=['get'], detail=True, perms_map = {'get':'*'}) def depts(self, request, *args, **kwargs): """ 聚合 """ obj = self.get_object() serializer = OrganizationSimpleSerializer(obj.depts.distinct(), many=True) return Response(serializer.data) def cal_task_rate(task): """ 计算任务上报率, 确认率 """ objs = Record.objects.filter(task=task) if objs.exists(): count = objs.count() up_rate = objs.filter(state__in=['已确认','已上报']).count()/count confirm_rate = objs.filter(state='已确认').count()/count task.up_rate = int(up_rate*100) task.confirm_rate = int(confirm_rate*100) if task.confirm_rate == 100: task.state ='已完成' task.save() class RecordViewSet(RbacFilterSet, PageOrNot, CreateUpdateCustomMixin, ModelViewSet): perms_map = {'get': '*', 'post': 'record_create', 'put': 'record_update', 'delete': 'record_delete'} queryset = Record.objects.all() serializer_class = RecordSerializer search_fields = ['content__name','task__name'] ordering = ['create_time'] filter_class = RecordFilter # 过滤类 def update(self, request, *args, **kwargs): instance = self.get_object() instance.note = request.data.get('note', None) instance.save() return Response(status=status.HTTP_200_OK) @action(methods=['post'], detail=True, perms_map = {'post':'record_up'}) def createself(self, request, *args, **kwargs): contents = request.data['contents'] rlist = [] for i in contents: data = {'content':Content.objects.get(pk=i), 'belong_dept':request.user.dept, 'is_self':True} rlist.append(Record(**data)) Record.objects.bulk_create(rlist) return Response(status=status.HTTP_200_OK) @action(methods=['put'], detail=True, perms_map = {'post':'record_up'}) def up(self, request, *args, **kwargs): """ 上报 """ obj = self.get_object() if obj.state == '待上报' or obj.state == '待整改': if 'files' in request.data: obj.files.clear() obj.files.add(*request.data['files']) if request.data.get('noteb', None): obj.noteb = request.data['noteb'] if request.data.get('is_yes', True): obj.is_yes = request.data.get('is_yes') obj.state = '已上报' obj.up_user = request.user obj.up_date = timezone.now() obj.save() if obj.task: cal_task_rate(obj.task) return Response(status=status.HTTP_200_OK) return Response('记录状态错误', status=status.HTTP_400_BAD_REQUEST) @action(methods=['put'], detail=True, perms_map = {'post':'record_reject'}) def reject(self, request, *args, **kwargs): """ 驳回 """ obj = self.get_object() if obj.state == '已上报': if request.data.get('note', None): obj.note = request.data['note'] obj.state = '待整改' obj.save() if obj.task: cal_task_rate(obj.task) return Response(status=status.HTTP_200_OK) else: return Response('请填写备注', status=status.HTTP_400_BAD_REQUEST) return Response('记录状态错误', status=status.HTTP_400_BAD_REQUEST) @action(methods=['put'], detail=True, perms_map = {'post':'record_confirm'}) def confirm(self, request, *args, **kwargs): """ 确认 """ obj = self.get_object() if obj.state == '已上报': obj.state = '已确认' obj.save() if obj.task: cal_task_rate(obj.task) return Response(status=status.HTTP_200_OK) return Response('记录状态错误', status=status.HTTP_400_BAD_REQUEST)