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

View File

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