from apps.certificate.serializers import CertificateSerializer
import re
from django.http import request
from rest_framework.views import APIView
from django.shortcuts import render
from rest_framework.serializers import ModelSerializer
from rest_framework.viewsets import ModelViewSet, GenericViewSet
from rest_framework.response import Response
from rest_framework import status
from .models import *
from apps.system.models import Dict
from apps.system.permission_data import RbacFilterSet
from apps.system.mixins import CreateUpdateCustomMixin, OptimizationMixin
import random
from rest_framework.decorators import action
from utils.pagination import PageOrNot
from .serializers import *
from rest_framework.exceptions import ParseError, NotAuthenticated
from django.utils import timezone
from apps.project.models import Certappunit
# Create your views here.

class CertificateViewSet(PageOrNot, OptimizationMixin, ModelViewSet):
    perms_map = {'get': 'certificate_view', 'post':'certificate_create', 'put':'certificate_update'}
    queryset = Certificate.objects.all()
    serializer_class = CertificateSerializer
    ordering = ['-create_time']
    filterset_fields = ['number', 'certunit__unittype']
    search_fields = ['number', 'certunit__name', 'certunit__enterprise__name']

    def create(self, request, *args, **kwargs):
        """
        证书颁发
        """
        serializer = CertificateCreateSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        print(request.data)
        if Certificate.objects.filter(certunit__id=request.data['certunit']).exists():
            return Response('该认证单元已存在证书', status = status.HTTP_400_BAD_REQUEST)
        if Certappunit.objects.get(pk=request.data['id']).decision is None:
            return Response('该认证单元未认证决定', status = status.HTTP_400_BAD_REQUEST)
        serializer.save(create_by = self.request.user, number = 'ZS' + str(random.randrange(2000,3000)), status = Dict.objects.get(code='validate_01'))
        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)