Soulter
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
from .. import Route, Response
|
|
from quart import Quart, request
|
|
from type.types import Context
|
|
|
|
class AuthRoute(Route):
|
|
def __init__(self, context: Context, app: Quart) -> None:
|
|
super().__init__(context, app)
|
|
self.routes = {
|
|
'/auth/login': ('POST', self.login),
|
|
'/auth/password/reset': ('POST', self.reset_password),
|
|
}
|
|
self.register_routes()
|
|
|
|
async def login(self):
|
|
username = self.context.config_helper.dashboard.username
|
|
password = self.context.config_helper.dashboard.password
|
|
post_data = await request.json
|
|
if post_data["username"] == username and post_data["password"] == password:
|
|
return Response().ok({
|
|
"token": "astrbot-test-token",
|
|
"username": username
|
|
}).__dict__
|
|
else:
|
|
return Response().error("用户名或密码错误").__dict__
|
|
|
|
async def reset_password(self):
|
|
password = self.context.config_helper.dashboard.password
|
|
post_data = await request.json
|
|
if post_data["password"] == password:
|
|
self.context.config_helper.dashboard.password = post_data['new_password']
|
|
return Response().ok(None).__dict__
|
|
else:
|
|
return Response().error("原密码错误").__dict__ |