diff --git a/endpoints/login.post.py b/endpoints/login.post.py index a8ab4aa..0b121d9 100644 --- a/endpoints/login.post.py +++ b/endpoints/login.post.py @@ -1,37 +1,29 @@ -from fastapi import APIRouter, Depends, HTTPException -from pydantic import BaseModel -from datetime import timedelta -from core.database import get_db +# Entity: Message + +```python +from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session -from core.auth import verify_password, create_access_token -from models.user import User +from typing import List +from core.database import get_db +from models.message import Message +from schemas.message import MessageSchema, MessageCreate router = APIRouter() -class UserAuth(BaseModel): - username: str - password: str - -@router.post("/login") -async def login( - user_data: UserAuth, +@router.get("/messages/{user_id}", response_model=List[MessageSchema]) +async def get_user_messages( + user_id: int, db: Session = Depends(get_db) ): - """User authentication endpoint""" - user = db.query(User).filter(User.username == user_data.username).first() + messages = db.query(Message).filter( + (Message.sender_id == user_id) | (Message.receiver_id == user_id) + ).all() - if not user or not verify_password(user_data.password, user.hashed_password): - raise HTTPException(status_code=400, detail="Invalid credentials") + if not messages: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail="No messages found for this user" + ) - # Generate token with expiration - access_token = create_access_token( - data={"sub": user.id}, - expires_delta=timedelta(hours=1) - ) - - return { - "access_token": access_token, - "token_type": "bearer", - "user_id": user.id, - "username": user.username - } + return messages +``` \ No newline at end of file