Update code in endpoints/logout.post.py

endpoints/logout.post.py

@@ -1,17 +1,19 @@
 # Entity: User
 
 from fastapi import APIRouter, Depends, status
+from fastapi.security import OAuth2PasswordBearer
 from sqlalchemy.orm import Session
 from core.database import get_db
-from models.user import User
-from schemas.user import UserSchema
+from schemas.user import UserToken
 from helpers.user_helpers import logout_user
 
 router = APIRouter()
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
 @router.post("/logout", status_code=200)
 async def logout(
+    token: str = Depends(oauth2_scheme),
     db: Session = Depends(get_db)
 ):
-    logout_user(db)
-    return {"message": "Successfully logged out"}
+    logout_user(token, db)
+    return {"message": "Logout successful"}