Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 06:12:53 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

perf👌: webui登录细节优化

Browse Source
This commit is contained in:
HibiKier 2024-01-22 05:33:41 +08:00
parent 428f374a8c
commit 66c9b8045d
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
plugins/web_ui/auth/__init__.py
View File

@ -5,6 +5,8 @@ import nonebot
from fastapi import APIRouter, Depends
from fastapi.security import OAuth2PasswordRequestForm
from configs.config import Config
from ..base_model import Result
from ..utils import (
ACCESS_TOKEN_EXPIRE_MINUTES,
@ -22,13 +24,14 @@ router = APIRouter()
@router.post("/login")
async def login_get_token(form_data: OAuth2PasswordRequestForm = Depends()):
if user := get_user(form_data.username):
if user.password != form_data.password:
return Result.fail("真笨, 密码都能记错!", 999)
else:
username = Config.get_config("web-ui", "username")
password = Config.get_config("web-ui", "password")
if not username or not password:
return Result.fail("你滴配置文件里用户名密码配置项为空", 998)
if username != form_data.username or password != form_data.password:
return Result.fail("真笨, 账号密码都能记错!", 999)
access_token = create_token(
user=user, expires_delta=timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
user=get_user(form_data.username), expires_delta=timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
)
token_data["token"].append(access_token)
if len(token_data["token"]) > 3: