Fix login errors, add get /users and get /user (username or id) routes.
This commit is contained in:
parent
97afe5447a
commit
fbf731acb6
5 changed files with 59 additions and 24 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in a new issue