26 lines
826 B
Python
26 lines
826 B
Python
|
from fastapi import HTTPException
|
||
|
|
from fastapi.responses import RedirectResponse
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
from starlette.requests import Request
|
||
|
|
|
||
|
|
from web.models.user import User, UserStatusEnum
|
||
|
|
|
||
|
|
|
||
|
|
def get_session_user_id(request: Request) -> int | None:
|
||
|
|
return request.session.get("user_id")
|
||
|
|
|
||
|
|
|
||
|
|
def get_current_user(request: Request, db: Session) -> User:
|
||
|
|
user_id = get_session_user_id(request)
|
||
|
|
if not user_id:
|
||
|
|
raise HTTPException(status_code=307, headers={"Location": "/login"})
|
||
|
|
user = db.get(User, user_id)
|
||
|
|
if not user or user.status == UserStatusEnum.suspended:
|
||
|
|
request.session.clear()
|
||
|
|
raise HTTPException(status_code=307, headers={"Location": "/login"})
|
||
|
|
return user
|
||
|
|
|
||
|
|
|
||
|
|
def login_redirect() -> RedirectResponse:
|
||
|
|
return RedirectResponse("/login", status_code=303)
|