mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import json
|
|
from datetime import timedelta
|
|
|
|
import nonebot
|
|
from fastapi import APIRouter, Depends
|
|
from fastapi.security import OAuth2PasswordRequestForm
|
|
|
|
from ..models.model import Result
|
|
from ..utils import (
|
|
ACCESS_TOKEN_EXPIRE_MINUTES,
|
|
create_token,
|
|
get_user,
|
|
token_data,
|
|
token_file,
|
|
)
|
|
|
|
app = nonebot.get_app()
|
|
|
|
|
|
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:
|
|
return Result.fail("你滴配置文件里用户名密码配置项为空", 998)
|
|
access_token = create_token(
|
|
user=user, expires_delta=timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
|
|
)
|
|
token_data["token"].append(access_token)
|
|
if len(token_data["token"]) > 3:
|
|
token_data["token"] = token_data["token"][1:]
|
|
with open(token_file, "w", encoding="utf8") as f:
|
|
json.dump(token_data, f, ensure_ascii=False, indent=4)
|
|
return Result.ok(
|
|
{"access_token": access_token, "token_type": "bearer"}, "欢迎回家, 欧尼酱!"
|
|
)
|