GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-13 01:30:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Devaev Maxim 2020-08-30 21:00:35 +03:00
parent a7de2184a7
commit 2175338470
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
kvmd/apps/kvmd/api/auth.py
View File

@ -43,18 +43,15 @@ _COOKIE_AUTH_TOKEN = "auth_token"
async def check_request_auth(auth_manager: AuthManager, exposed: HttpExposed, request: Request) -> None: async def check_request_auth(auth_manager: AuthManager, exposed: HttpExposed, request: Request) -> None:
if exposed.auth_required and auth_manager.is_auth_enabled(): if exposed.auth_required and auth_manager.is_auth_enabled():
user = request.headers.get("X-KVMD-User", "") if (user := request.headers.get("X-KVMD-User", "")):
passwd = request.headers.get("X-KVMD-Passwd", "")
token = request.cookies.get(_COOKIE_AUTH_TOKEN, "")
if user:
user = valid_user(user) user = valid_user(user)
passwd = request.headers.get("X-KVMD-Passwd", "")
set_request_auth_info(request, f"{user} (xhdr)") set_request_auth_info(request, f"{user} (xhdr)")
if not (await auth_manager.authorize(user, valid_passwd(passwd))): if not (await auth_manager.authorize(user, valid_passwd(passwd))):
raise ForbiddenError() raise ForbiddenError()
return return
elif token: elif (token := request.cookies.get(_COOKIE_AUTH_TOKEN, "")):
user = auth_manager.check(valid_auth_token(token)) user = auth_manager.check(valid_auth_token(token))
if not user: if not user:
set_request_auth_info(request, "- (token)") set_request_auth_info(request, "- (token)")