89 lines
3.5 KiB
Python
89 lines
3.5 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
|
|
|
|
# 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
|
|
@action(methods=['post'], detail=False, perms_map = {'post':'task_createself'})
|
|
def createself(self, request):
|
|
"""
|
|
主动报送
|
|
"""
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save(is_self=True, create_by=request.user, belong_dept=request.user.dept)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
@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)
|
|
class RecordViewSet(CreateUpdateCustomMixin,ModelViewSet):
|
|
perms_map = {'get': '*'}
|
|
queryset = Record.objects.all()
|
|
serializer_class = RecordSerializer
|
|
search_fields = ['content','task','up_date','state']
|
|
ordering = ['-create_time']
|
|
def get_queryset(self):
|
|
queryset = self.queryset
|
|
if hasattr(self.get_serializer_class(), 'setup_eager_loading'):
|
|
queryset = self.get_serializer_class().setup_eager_loading(queryset) # 性能优化
|
|
return queryset
|
|
|