From 06012fda6e1b2453e9f0b879311659f6b45f6f00 Mon Sep 17 00:00:00 2001
From: ktianc <ktianc2000@163.com>
Date: Sun, 5 Mar 2023 23:07:52 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=EF=BC=88kinit-api=EF=BC=89?=
 =?UTF-8?q?=EF=BC=9A=E6=BC=94=E7=A4=BA=E7=8E=AF=E5=A2=83=E4=B8=8B=E5=8F=96?=
 =?UTF-8?q?=E6=B6=88=E7=99=BB=E5=BD=95=E5=A4=B1=E8=B4=A5=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../vadmin/auth/utils/validation/login.py     | 24 +++++++++----------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/kinit-api/apps/vadmin/auth/utils/validation/login.py b/kinit-api/apps/vadmin/auth/utils/validation/login.py
index 9c4b251..654e348 100644
--- a/kinit-api/apps/vadmin/auth/utils/validation/login.py
+++ b/kinit-api/apps/vadmin/auth/utils/validation/login.py
@@ -6,13 +6,11 @@
 # @IDE            : PyCharm
 # @desc           : 登录验证装饰器
 
-from fastapi import Request, Depends
+from fastapi import Request
 from pydantic import BaseModel, validator
 from sqlalchemy.ext.asyncio import AsyncSession
-
-from application.settings import DEFAULT_AUTH_ERROR_MAX_NUMBER
-from apps.vadmin.auth import models, crud, schemas
-from core.database import db_getter
+from application.settings import DEFAULT_AUTH_ERROR_MAX_NUMBER, DEMO
+from apps.vadmin.auth import crud, schemas
 from core.validator import vali_telephone
 from typing import Optional
 from utils.count import Count
@@ -70,18 +68,20 @@ class LoginValidation:
 
         if not result.status:
             self.result.msg = result.msg
-            number = await count.add(ex=86400)
-            if number >= DEFAULT_AUTH_ERROR_MAX_NUMBER:
-                await count.reset()
-                # 如果等于最大次数，那么就将用户 is_active=False
-                user.is_active = False
-                await db.flush()
+            if not DEMO:
+                number = await count.add(ex=86400)
+                if number >= DEFAULT_AUTH_ERROR_MAX_NUMBER:
+                    await count.reset()
+                    # 如果等于最大次数，那么就将用户 is_active=False
+                    user.is_active = False
+                    await db.flush()
         elif not user.is_active:
             self.result.msg = "此手机号已被冻结！"
         elif data.platform in ["0", "1"] and not user.is_staff:
             self.result.msg = "此手机号无权限！"
         else:
-            await count.delete()
+            if not DEMO:
+                await count.delete()
             self.result.msg = "OK"
             self.result.status = True
             self.result.user = schemas.UserSimpleOut.from_orm(user)