diff --git a/endpoints/login.post.py b/endpoints/login.post.py index 0b121d9..908eeb8 100644 --- a/endpoints/login.post.py +++ b/endpoints/login.post.py @@ -6,24 +6,18 @@ 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) ): 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" - ) + ).order_by(Message.created_at.desc()).all() return messages ``` \ No newline at end of file