diff --git a/endpoints/login.post.py b/endpoints/login.post.py index 0b121d9..c0646d5 100644 --- a/endpoints/login.post.py +++ b/endpoints/login.post.py @@ -6,11 +6,11 @@ from sqlalchemy.orm import Session from typing import List from core.database import get_db from models.message import Message -from schemas.message import MessageSchema, MessageCreate +from schemas.message import MessageSchema router = APIRouter() -@router.get("/messages/{user_id}", response_model=List[MessageSchema]) +@router.get("/messages", status_code=200, response_model=List[MessageSchema]) async def get_user_messages( user_id: int, db: Session = Depends(get_db) @@ -18,12 +18,5 @@ async def get_user_messages( messages = db.query(Message).filter( (Message.sender_id == user_id) | (Message.receiver_id == user_id) ).all() - - if not messages: - raise HTTPException( - status_code=status.HTTP_404_NOT_FOUND, - detail="No messages found for this user" - ) - return messages ``` \ No newline at end of file