22 lines
595 B
Python
22 lines
595 B
Python
# Entity: Message
|
|
|
|
```python
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
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
|
|
|
|
router = APIRouter()
|
|
|
|
@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()
|
|
return messages
|
|
``` |