57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.permissions import AllowAny, IsAuthenticated
|
|
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from django.core.mail import EmailMessage
|
|
from django.db import transaction
|
|
|
|
from .serializers import *
|
|
from .services import *
|
|
from common.models.choiceModels import CertificateCodeUseType
|
|
from common.utils.codeManger import generate_code
|
|
|
|
|
|
certificate_use_type = {
|
|
"phone": SmsService,
|
|
# "email": EmailService
|
|
}
|
|
|
|
class CertificationAPIView(APIView):
|
|
permission_classes = [AllowAny]
|
|
|
|
# 인증 발송
|
|
@transaction.atomic
|
|
def post(self, request):
|
|
use_type = request.query_params.get("type")
|
|
serv = certificate_use_type[use_type]
|
|
serializer = CertificateCodeSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
create_code = generate_code(6)
|
|
if serv.save_certificate_info(use_type, create_code, serializer.validated_data['identifier']):
|
|
if serv.send(create_code, serializer.validated_data['identifier']):
|
|
return Response({'message': "success send and save"})
|
|
else: # 전송 실패
|
|
return Response({"message": "failed send"})
|
|
else: # 코드 저장 실패
|
|
return Response({'message': "failed save"}, status=status.HTTP_400_BAD_REQUEST)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
# 인증 확인
|
|
@transaction.atomic
|
|
def patch(self, request):
|
|
use_type = request.query_params.get("type")
|
|
serv = certificate_use_type[use_type]
|
|
code = request.data.get('code', None)
|
|
if not code:
|
|
return Response({"message": "no code"}, status=status.HTTP_400_BAD_REQUEST)
|
|
serializer = CertificateCodeSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
if serv.check_code(use_type, code, serializer.validated_data['identifier']):
|
|
return Response({"message": "certificated successfully"}, status=status.HTTP_200_OK)
|
|
return Response({"message": "wrong code, please retry"}, status=status.HTTP_400_BAD_REQUEST)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|