62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
from drf_spectacular.utils import extend_schema
|
|
from rest_framework.permissions import AllowAny
|
|
from rest_framework.decorators import action
|
|
from django_core.mixins.base import BaseViewSetMixin
|
|
from rest_framework.viewsets import GenericViewSet
|
|
from core.services import UserService, SmsService
|
|
from ..serializers import LoginSerializer, ConfirmSerializer
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from django.contrib.auth import get_user_model
|
|
from rest_framework.exceptions import PermissionDenied
|
|
from django_core import exceptions
|
|
|
|
|
|
@extend_schema(tags=["Login"])
|
|
class LoginView(BaseViewSetMixin, GenericViewSet, UserService):
|
|
permission_classes = [AllowAny]
|
|
|
|
def get_serializer_class(self):
|
|
match self.action:
|
|
case "send_code":
|
|
return LoginSerializer
|
|
case "send_confirm":
|
|
return ConfirmSerializer
|
|
case _:
|
|
return LoginSerializer
|
|
|
|
@action(detail=False, methods=["post"], url_path="send_code")
|
|
def send_code(self, request):
|
|
ser = self.get_serializer(data=request.data)
|
|
ser.is_valid(raise_exception=True)
|
|
data = ser.validated_data
|
|
phone = data.get('phone')
|
|
self.send_confirmation(phone)
|
|
return Response(
|
|
{"detail": _("Sms %(phone)s raqamiga yuborildi") % {"phone": phone}},
|
|
status=status.HTTP_202_ACCEPTED,
|
|
)
|
|
|
|
@action(detail=False, methods=["post"], url_path="send_confirm")
|
|
def send_confirm(self, request):
|
|
ser = self.get_serializer(data=request.data)
|
|
ser.is_valid(raise_exception=True)
|
|
data = ser.validated_data
|
|
phone = data.get("phone")
|
|
code = data.get("code")
|
|
try:
|
|
if SmsService.check_confirm(phone, code=code):
|
|
token = self.get_token(get_user_model().objects.filter(phone=phone).first())
|
|
return Response(
|
|
data={
|
|
"detail": _("Tasdiqlash ko'di qabul qilindi"),
|
|
"token": token,
|
|
},
|
|
status=status.HTTP_202_ACCEPTED,
|
|
)
|
|
except exceptions.SmsException as e:
|
|
raise PermissionDenied(e) # Response exception for APIException
|
|
except Exception as e:
|
|
raise PermissionDenied(e) # Api exception for APIException
|