Fix login errors, add get /users and get /user (username or id) routes.

This commit is contained in:
Hannah dagemark 2026-09-09 17:08:28 +02:00
commit fbf731acb6
5 changed files with 59 additions and 24 deletions

View file

@ -6,12 +6,17 @@ from fastapi import Depends, HTTPException, status
from fastapi.security import OAuth2PasswordBearer
from pwdlib import PasswordHash
from app.config import Settings
from app.database import add_to_database, get_from_database_where
from app.config import settings
from app.database import (
add_to_database,
get_from_database,
get_from_database_where,
get_from_database_where_id,
)
from app.models.token import TokenData
from app.models.user import User, UserIn, UserInDB
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/users/auth")
password_hash = PasswordHash.recommended()
@ -26,25 +31,37 @@ def get_password_hash(password: str):
return password_hash.hash(password)
def get_users():
return get_from_database(UserInDB)
def get_user(username: str):
user = get_from_database_where(
User,
User.username, # type: ignore (Type correct on runtime)
UserInDB,
UserInDB.username, # type: ignore (Type correct on runtime)
username,
True,
limit=1,
)
if user is not None:
return UserInDB.model_validate(user[0])
return user[0]
def get_user_by_id(id: int):
user = get_from_database_where_id(UserInDB, id)
if user is not None:
return user
def save_user(user: UserIn):
print("Got save user call")
user_in_db = UserInDB(
username=user.username,
authority=user.authority,
hashed_password=get_password_hash(user.password),
)
add_to_database(user_in_db)
return user_in_db
print("Saved user")
return User.model_validate(get_user(user.username))
def authenticate_user(username: str, password: str):
@ -69,8 +86,8 @@ def create_access_token(
to_encode.update({"exp": expire})
encoded_jwt = jwt.encode( # pyright: ignore[reportUnknownMemberType] (Third party skill issue)
to_encode,
Settings.secret_key,
algorithm=Settings.algorithm,
settings.secret_key,
algorithm=settings.algorithm,
)
return encoded_jwt
@ -84,8 +101,8 @@ async def get_current_user(token: Annotated[str, Depends(oauth2_scheme)]):
try:
payload = jwt.decode( # pyright: ignore[reportUnknownMemberType] (Third party skill issue)
token,
Settings.secret_key,
algorithms=[Settings.algorithm],
settings.secret_key,
algorithms=[settings.algorithm],
)
username = payload.get("sub")
if username is None:
@ -96,7 +113,7 @@ async def get_current_user(token: Annotated[str, Depends(oauth2_scheme)]):
user = get_user(token_data.username) # type: ignore Look just above here idiot!
if user is None:
raise credentials_exception
return user
return User.model_validate(user)
async def get_current_active_user(