40 lines
1.9 KiB
Python
40 lines
1.9 KiB
Python
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
|
|
# 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)
|
|
if Certificate.objects.filter(certunit__id=request.data['certunit']).exists():
|
|
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) |