Spaces:
Running
Running
File size: 1,102 Bytes
c672e11 bfa9638 acdfb5c bfa9638 c672e11 bfa9638 c672e11 bfa9638 c672e11 bfa9638 c672e11 bfa9638 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from typing import Any
from fastapi import APIRouter, Depends, HTTPException
from utils.db import tinydb_helper # Ensure this import is correct based on our project structure
from dependencies import get_current_user, oauth2_scheme
router = APIRouter()
@router.post("/user/logout")
async def user_logout(token: str = Depends(oauth2_scheme), current_user: Any = Depends(get_current_user)):
try:
# Assuming `get_current_user` now also ensures and returns the full payload including `user_id`
user_id = current_user["user_id"]
# Invalidate the token by removing it from the database
if not tinydb_helper.query_token(user_id, token):
raise HTTPException(status_code=404, detail="Token not found.")
tinydb_helper.remove_token_by_value(token)
if tinydb_helper.query_token(user_id, token):
raise HTTPException(status_code=404, detail="Logout unsuccessful.")
return {"message": "User logged out successfully"}
except Exception as e:
raise HTTPException(status_code=400, detail=f"Error during logout: {str(e)}")
|