222 lines
8.6 KiB
Python
222 lines
8.6 KiB
Python
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, has_permission
|
|
from django.db.models import Q
|
|
# Create your views here.
|
|
|
|
class ContentViewSet(CreateUpdateCustomMixin, ModelViewSet):
|
|
"""
|
|
资料清单：增删改查
|
|
"""
|
|
perms_map = {'get': '*', 'post': 'content',
|
|
'put': 'content', 'delete': 'content'}
|
|
queryset = Content.objects.all()
|
|
serializer_class = ContentSerializer
|
|
pagination_class = None
|
|
search_fields = ['name', 'desc']
|
|
filterset_fields = ['type','can_doself']
|
|
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
|
|
data['state'] = '待发布'
|
|
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)
|
|
|
|
@action(methods=['put'], detail=True, perms_map = {'put':'task_start'})
|
|
def start(self, request, *args, **kwargs):
|
|
"""
|
|
发布任务
|
|
"""
|
|
obj = self.get_object()
|
|
if obj.state == '待发布':
|
|
obj.state = '执行中'
|
|
obj.save()
|
|
Record.objects.filter(task=obj).update(state='待上报')
|
|
return Response(status=status.HTTP_200_OK)
|
|
return Response('任务状态错误', status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
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 # 过滤类
|
|
|
|
@action(methods=['get'], detail=False, perms_map = {'get':'*'})
|
|
def todos(self, request, *args, **kwargs):
|
|
ret={}
|
|
dept = request.user.dept
|
|
objs = Record.objects.filter(is_deleted=False, belong_dept=dept)
|
|
ret['toup'] = objs.filter(state='待上报').count()
|
|
ret['tozg'] = objs.filter(state='待整改').count()
|
|
if has_permission('record_confirm', request.user):
|
|
ret['toconfirm'] = Record.objects.filter(is_deleted=False, state='已上报').count()
|
|
return Response(ret)
|
|
|
|
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=False, perms_map = {'post':'record_createself'})
|
|
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 = {'put':'record_up'})
|
|
def up(self, request, *args, **kwargs):
|
|
"""
|
|
上报
|
|
"""
|
|
obj = self.get_object()
|
|
if obj.state in ['待整改','待上报']:
|
|
if 'files' in request.data and len(request.data['files'])==0:
|
|
if request.data.get('noteb', None) and request.data.get('is_yes', True):
|
|
pass
|
|
else:
|
|
return Response('请上传文件或选择不适用并填写上报说明', status=status.HTTP_400_BAD_REQUEST)
|
|
obj.is_yes = request.data.get('is_yes')
|
|
obj.files.clear()
|
|
obj.files.add(*request.data['files'])
|
|
obj.noteb = request.data['noteb']
|
|
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 = {'put':'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 = {'put':'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)
|
|
|
|
|