bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement custom url_for to generating urls with ROOT_PATH prefix

Browse Source
This commit is contained in:
Iam54r1n4
2025-02-04 16:34:52 +00:00
parent 468f4a4abc
commit 360e6ac4ba
4 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
core/scripts/webpanel/routers/login/login.py
View File

@ -2,7 +2,7 @@ from fastapi import APIRouter, Depends, Form, Request
from fastapi.responses import RedirectResponse
from fastapi.templating import Jinja2Templates
from dependency import get_templates, get_session_manager
from dependency import get_templates, get_session_manager, url_for
from session import SessionManager
from config import CONFIGS
@ -18,7 +18,7 @@ async def login(request: Request, templates: Jinja2Templates = Depends(get_templ
async def login_post(
request: Request,
templates: Jinja2Templates = Depends(get_templates), session_manager: SessionManager = Depends(get_session_manager),
username: str = Form(), password: str = Form(),
username: str = Form(), password: str = Form()
):
ADMIN_USERNAME = CONFIGS.ADMIN_USERNAME
ADMIN_PASSWORD = CONFIGS.ADMIN_PASSWORD
@ -28,7 +28,7 @@ async def login_post(
session_id = session_manager.set_session(username)
res = RedirectResponse(url='/', status_code=302)
res = RedirectResponse(url=url_for(context={'request': request}, name='index'), status_code=302)
res.set_cookie(key='session_id', value=session_id)
return res
@ -40,6 +40,6 @@ async def logout(request: Request, session_manager: SessionManager = Depends(get
if session_id:
session_manager.revoke_session(session_id)
res = RedirectResponse(url='/', status_code=302)
res = RedirectResponse(url=url_for(context={'request': request}, name='index'), status_code=302)
res.delete_cookie('session_id')
return res