41 lines
1.5 KiB
Python
41 lines
1.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 openpyxl import Workbook, load_workbook
|
|
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
|
|
return TaskListSerializer
|
|
|