rename user.py to users.py and clean up code

This commit is contained in:
maxDorninger
2025-02-23 11:38:56 +01:00
parent aca686d1e0
commit 606d233ec4
6 changed files with 12 additions and 12 deletions

View File

@@ -8,8 +8,8 @@ from jwt.exceptions import InvalidTokenError
from pydantic import BaseModel
import database
import database.user
from database.user import UserInternal
import database.users
from database.users import UserInternal
class Token(BaseModel):
access_token: str
@@ -52,7 +52,7 @@ async def get_current_user(token: str = Depends(oauth2_scheme)) -> UserInternal:
except InvalidTokenError:
log.warning("received invalid token: " + token)
raise credentials_exception
user = database.user.get_user(uid=token_data.uid)
user = database.users.get_user(uid=token_data.uid)
if user is None:
log.debug("user not found")
raise credentials_exception

View File

@@ -7,8 +7,8 @@ from fastapi.security import OAuth2PasswordRequestForm
import database
from auth import ACCESS_TOKEN_EXPIRE_MINUTES, create_access_token, Token, router
from database import user
from database.user import User, UserInternal
from database import users
from database.users import UserInternal
def verify_password(plain_password, hashed_password):
@@ -32,7 +32,7 @@ def authenticate_user(email: str, password: str) -> bool | UserInternal:
:param password: password of the user
:return: if authentication succeeds, returns the user object with added name and lastname, otherwise or if the user doesn't exist returns False
"""
user = database.user.get_user(email=email)
user = database.users.get_user(email=email)
if not user:
return False
if not verify_password(password, user.hashed_password):

View File

@@ -7,7 +7,7 @@ from psycopg.rows import dict_row
log = logging.getLogger(__name__)
class PgDatabase():
class PgDatabase:
"""PostgreSQL Database context manager using psycopg"""
def __init__(self) -> None:
@@ -35,8 +35,8 @@ class PgDatabase():
def init_db():
log.info("Initializing database")
from database import tv, user
user.init_table()
from database import tv, users
users.init_table()
tv.init_table()
log.info("Tables initialized successfully")

View File

@@ -73,7 +73,7 @@ def get_user(email: str = None, uid: str = None) -> UserInternal | None:
user = UserInternal(id=result["id"].__str__(), name=result["name"], lastname=result["lastname"],
email=result["email"],
hashed_password=result["hashed_password"])
log.debug(f"Retrieved User succesfully: {user.model_dump()} ")
log.debug(f"Retrieved User successfully: {user.model_dump()} ")
return user

View File

View File

@@ -6,7 +6,7 @@ from starlette.responses import JSONResponse
import database
from auth import get_current_user
from auth.password import get_password_hash
from database.user import UserInternal, User
from database.users import UserInternal, User
from routers import log
router = APIRouter(
@@ -34,7 +34,7 @@ async def create_user(
):
internal_user = UserInternal(name=user.name, lastname=user.lastname, email=user.email,
hashed_password=get_password_hash(user.password))
if database.user.create_user(internal_user):
if database.users.create_user(internal_user):
log.info("Created new user", internal_user.model_dump())
return internal_user
else: