fm453/kinit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增（kinit-api）：演示环境下取消登录失败限制

Browse Source
This commit is contained in:
ktianc 2023-03-05 23:07:52 +08:00
parent ebcdac7796
commit 06012fda6e
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
kinit-api/apps/vadmin/auth/utils/validation/login.py
View File

@ -6,13 +6,11 @@
# @IDE : PyCharm # @IDE : PyCharm
# @desc : 登录验证装饰器 # @desc : 登录验证装饰器
from fastapi import Request, Depends from fastapi import Request
from pydantic import BaseModel, validator from pydantic import BaseModel, validator
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from application.settings import DEFAULT_AUTH_ERROR_MAX_NUMBER, DEMO
from application.settings import DEFAULT_AUTH_ERROR_MAX_NUMBER from apps.vadmin.auth import crud, schemas
from apps.vadmin.auth import models, crud, schemas
from core.database import db_getter
from core.validator import vali_telephone from core.validator import vali_telephone
from typing import Optional from typing import Optional
from utils.count import Count from utils.count import Count
@ -70,6 +68,7 @@ class LoginValidation:
if not result.status: if not result.status:
self.result.msg = result.msg self.result.msg = result.msg
if not DEMO:
number = await count.add(ex=86400) number = await count.add(ex=86400)
if number >= DEFAULT_AUTH_ERROR_MAX_NUMBER: if number >= DEFAULT_AUTH_ERROR_MAX_NUMBER:
await count.reset() await count.reset()
@ -81,6 +80,7 @@ class LoginValidation:
elif data.platform in ["0", "1"] and not user.is_staff: elif data.platform in ["0", "1"] and not user.is_staff:
self.result.msg = "此手机号无权限！" self.result.msg = "此手机号无权限！"
else: else:
if not DEMO:
await count.delete() await count.delete()
self.result.msg = "OK" self.result.msg = "OK"
self.result.status = True self.result.status = True