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 fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from typing import List
from core.database import get_db from core.database import get_db
from models.user import User from models.name import Name
from schemas.user import UserLogin, UserResponse from schemas.name import NameSchema, NameCreate
from helpers.auth_helpers import authenticate_user from helpers.name_helpers import get_all_names, create_name
from helpers.token_helpers import create_access_token
router = APIRouter() router = APIRouter()
@router.post("/login", status_code=200, response_model=UserResponse) @router.post("/login", status_code=201, response_model=List[NameSchema])
async def login( async def login_names(
login_data: UserLogin, name_data: NameCreate,
db: Session = Depends(get_db) db: Session = Depends(get_db)
): ):
user = authenticate_user(db, login_data.userid, login_data.password) names = get_all_names(db)
if not user: return names
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
}