Kaynağa Gözat

更新日志

yf_yzl 2 yıl önce
ebeveyn
işleme
06295bc4a6
1 değiştirilmiş dosya ile 11 ekleme ve 11 silme
  1. 11 11
      smartfarming/views/user.py

+ 11 - 11
smartfarming/views/user.py

@@ -146,18 +146,18 @@ class LoginAPIView(APIView):
         else:
             captcha_low = captcha.lower()
         key = f"{code_str}_{captcha_low}"
-        # redis_tools = RedisPool().get_redis_pool(settings.redis_db["captcha"])
-        # is_captcha = redis_tools.get(key)
-        # if is_captcha:
-        #     if captcha_low == captcha_low:
-        #         pass
-        #     else:
-        #         return Response({"msg": "验证码校验失败", "code": 2})
-        # else:
-        #     return Response({"msg": "验证码已过期或验证码错误", "code": 2})
+        redis_tools = RedisPool().get_redis_pool(settings.redis_db["captcha"])
+        is_captcha = redis_tools.get(key)
+        if is_captcha:
+            if captcha_low == captcha_low:
+                pass
+            else:
+                return Response({"msg": "验证码校验失败", "code": 2})
+        else:
+            return Response({"msg": "验证码已过期或验证码错误", "code": 2})
         user = authenticate(username=username, password=password)
         if user:
-            # redis_tools.delete(key)
+            redis_tools.delete(key)
             device_user = DeviceUser.objects.filter(uid=user.id, state=1)
             device_user = device_user.first() if device_user else None
             msg = ""
@@ -233,7 +233,7 @@ class LoginInfoAPIView(APIView):
             }
             return Response({"code": 0, "msg": "success", "data":data})
         except Exception as e:
-            print(e)
+            logger.error(f"获取权限列表失败: {e}")
             return Response({"code": 3, "msg": "获取权限列表失败"})
         
 class UserListAPIView(APIView):