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
        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)
class RecordViewSet(CreateUpdateCustomMixin,ModelViewSet):
    perms_map = {'get': '*', 'post': 'record_create',
                 'put': 'record_update', 'delete': 'record_delete'}
    queryset = Record.objects.all()
    serializer_class = RecordSerializer
    search_fields = ['name']
    ordering = ['-create_time']