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 OptimizationMixin
from apps.system.mixins import CreateUpdateCustomMixin
from django.utils import timezone
# 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)
    
    @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)

class RecordViewSet(OptimizationMixin,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','task','up_date','state']
    ordering = ['-create_time']
    filterset_fields = ['content', 'belong_dept', 'task']

    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=['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 request.data.get('files', None):
                obj.files = 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()
            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.save()
                return Response(status=status.HTTP_200_OK)
            else:
                return Response('请填写备注', status=status.HTTP_200_OK)
        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()
            return Response(status=status.HTTP_200_OK)
        return Response('记录状态错误', status=status.HTTP_400_BAD_REQUEST)