zhenxun_bot/plugins/web_ui/auth/__init__.py

import json
from datetime import timedelta

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,
    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()):
    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=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:
        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"}, "欢迎回家, 欧尼酱!"
    )