Update code in endpoints/login.post.py

This commit is contained in:
Backend IM Bot 2025-03-28 19:24:05 +00:00
parent 84b088ce58
commit 953c00c21b

View File

@ -1,30 +1,17 @@
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from typing import List
from core.database import get_db
from models.user import User
from schemas.user import UserLogin, UserResponse
from helpers.auth_helpers import authenticate_user
from helpers.token_helpers import create_access_token
from models.name import Name
from schemas.name import NameSchema, NameCreate
from helpers.name_helpers import get_all_names, create_name
router = APIRouter()
@router.post("/login", status_code=200, response_model=UserResponse)
async def login(
login_data: UserLogin,
@router.post("/login", status_code=201, response_model=List[NameSchema])
async def login_names(
name_data: NameCreate,
db: Session = Depends(get_db)
):
user = authenticate_user(db, login_data.userid, login_data.password)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Incorrect username or password",
headers={"WWW-Authenticate": "Bearer"},
)
access_token = create_access_token(data={"sub": user.userid})
return {
"access_token": access_token,
"token_type": "bearer",
"userid": user.userid
}
names = get_all_names(db)
return names