Restructure

This commit is contained in:
Iam54r1n4
2025-02-03 19:28:59 +00:00
parent 74feeab4c8
commit d1caf5c389
5 changed files with 6 additions and 5 deletions

View File

@ -1,2 +0,0 @@
from . import session
from . import auth

View File

@ -0,0 +1,2 @@
from .auth import AuthMiddleware
from .request import AfterRequestMiddleware

View File

@ -5,7 +5,7 @@ from starlette.types import ASGIApp
from typing import Awaitable, Callable
from datetime import datetime, timezone
from .session import SessionManager
from session import SessionManager
class AuthMiddleware(BaseHTTPMiddleware):
@ -43,13 +43,13 @@ class AuthMiddleware(BaseHTTPMiddleware):
return RedirectResponse(url='/login', status_code=302)
session_data = self.__session_manager.get_session(session_id)
if not session_data:
if is_api_request:
raise HTTPException(status_code=401, detail="The session is invalid.")
return RedirectResponse(url='/login', status_code=302)
if session_data.expires_at < datetime.now(timezone.utc):
if is_api_request:
raise HTTPException(status_code=401, detail="The session has expired.")

View File

@ -0,0 +1 @@
from .session import SessionData, SessionStorage, SessionManager