Add POST endpoint for /login

This commit is contained in:
Backend IM Bot 2025-03-26 07:48:48 +00:00
parent b1a1aa0a7d
commit ba6462bd3f

View File

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