Parcourir la source

优化登录接口逻辑

yf_fyh il y a 2 ans
Parent
commit
080a04daf0
1 fichiers modifiés avec 8 ajouts et 6 suppressions
  1. 8 6
      apps/UserApp/views.py

+ 8 - 6
apps/UserApp/views.py

@@ -20,12 +20,14 @@ class LoginView(APIView):
 
         user = authenticate(**request_data)
         if user is not None and user.is_active:
-            if request_data["modules"] == "pestanalysis":
-                if user.is_superuser or user.user_modules == 1:
-                    token = get_token(user)
-                    return Response(token)
-                else:
-                    return Response(data={"msg": "账户无权限进入该模块", "data": ""}, status=status.HTTP_401_UNAUTHORIZED)
+            if user.is_superuser:
+                token = get_token(user)
+                return Response(token)
+            if request_data["modules"] == "pestanalysis" and user.user_modules == 1:
+                token = get_token(user)
+                return Response(token)
+            else:
+                return Response(data={"msg": "账户无权限进入该模块", "data": ""}, status=status.HTTP_401_UNAUTHORIZED)
         else:
             return Response(data={"msg": "登录验证失败", "data": ""}, status=status.HTTP_401_UNAUTHORIZED)