fix(login)

core/apps/accounts/views/login.py

@@ -1,4 +1,4 @@
-from drf_spectacular.utils import extend_schema
+from drf_spectacular.utils import OpenApiResponse, extend_schema
 from rest_framework.permissions import AllowAny
 from rest_framework.decorators import action
 from django_core.mixins.base import BaseViewSetMixin
@@ -12,6 +12,8 @@ from django.contrib.auth import get_user_model
 from rest_framework.exceptions import PermissionDenied
 from django_core import exceptions
 
+User = get_user_model()
+
 
 @extend_schema(tags=["Login"])
 class LoginView(BaseViewSetMixin, GenericViewSet, UserService):
@@ -21,25 +23,50 @@ class LoginView(BaseViewSetMixin, GenericViewSet, UserService):
         match self.action:
             case "send_code":
                 return LoginSerializer
-            case "send_confirm":
+            case "confirm":
                 return ConfirmSerializer
             case _:
                 return LoginSerializer
 
-    @action(detail=False, methods=["post"], url_path="send_code")
+    @extend_schema(
+        summary="Tasdiqlash ko'dini olish",
+        responses={
+            200: OpenApiResponse(
+                response={
+                    "type": "object",
+                    "properties": {"detail": {"type": "string"}},
+                }
+            )
+        },
+    )
+    @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')
+        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):
+    @extend_schema(
+        summary="Tasdiqlash ko'dini kiritish",
+        responses={
+            200: OpenApiResponse(
+                response={
+                    "type": "object",
+                    "properties": {
+                        "token": {"type": "string"},
+                        "ask_name": {"type": "boolean"},
+                    },
+                }
+            )
+        },
+    )
+    @action(detail=False, methods=["post"], url_path="confirm")
+    def confirm(self, request):
         ser = self.get_serializer(data=request.data)
         ser.is_valid(raise_exception=True)
         data = ser.validated_data
@@ -47,15 +74,18 @@ class LoginView(BaseViewSetMixin, GenericViewSet, UserService):
         code = data.get("code")
         try:
             if SmsService.check_confirm(phone, code=code):
-                token = self.get_token(get_user_model().objects.filter(phone=phone).first())
+                user = User.objects.filter(phone=phone).first()
+                if user is None:
+                    user = User.objects.create_user(phone)
+                token = self.get_token(user)
                 return Response(
                     data={
-                        "detail": _("Tasdiqlash ko'di qabul qilindi"),
                         "token": token,
+                        "ask_name": True if user.first_name is None or user.first_name == "" else False,
                     },
                     status=status.HTTP_202_ACCEPTED,
                 )
         except exceptions.SmsException as e:
-            raise PermissionDenied(e)  # Response exception for APIException
+            raise PermissionDenied(e)
         except Exception as e:
-            raise PermissionDenied(e)  # Api exception for APIException
+            raise PermissionDenied(e)