diff --git a/endpoints/login.post.py b/endpoints/login.post.py index 0c4c62a..56d61a6 100644 --- a/endpoints/login.post.py +++ b/endpoints/login.post.py @@ -1,19 +1,17 @@ from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from core.database import get_db -from schemas.user import UserSchema, UserCreate -from helpers.user_helpers import get_user_by_username +from schemas.user import UserLogin +from helpers.user_helpers import authenticate_user router = APIRouter() -@router.post("/login", status_code=status.HTTP_200_OK, response_model=UserSchema) +@router.post("/login", status_code=status.HTTP_200_OK) async def login( - user_data: UserCreate, + user_login: UserLogin, db: Session = Depends(get_db) ): - user = get_user_by_username(db, username=user_data.username) - if not user: - raise HTTPException(status_code=404, detail="User not found") - if not user.verify_password(user_data.password): - raise HTTPException(status_code=401, detail="Invalid password") - return user \ No newline at end of file + authenticated_user = authenticate_user(db, user_login) + if not authenticated_user: + raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid credentials") + return authenticated_user \ No newline at end of file