diff --git a/server.py b/server.py index e199903..afc84f8 100644 --- a/server.py +++ b/server.py @@ -358,8 +358,8 @@ def format_date(value, length: int = 10) -> str: # ============ Auth UI Endpoints ============ @app.get("/login", response_class=HTMLResponse) -async def ui_login_page(request: Request): - """Login page.""" +async def ui_login_page(request: Request, return_to: str = None): + """Login page. Accepts optional return_to URL for redirect after login.""" username = get_user_from_cookie(request) if username: return HTMLResponse(base_html("Already Logged In", f''' @@ -369,10 +369,14 @@ async def ui_login_page(request: Request):
''', username)) - content = ''' + # Hidden field for return_to URL + return_to_field = f'' if return_to else '' + + content = f'''