starred/MediaManager-maxdorninger
1
0
Fork 0
mirror of https://github.com/maxdorninger/MediaManager.git synced 2026-04-20 12:55:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

add ability to create user, add password auth, add logging and work on dev containers

Browse Source
This commit is contained in:
maxid
2025-02-16 20:52:05 +01:00
parent ab6fc52e42
commit 5ac7e0a0d4
8 changed files with 63 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
MediaManager/src/database.py
View File

@@ -9,7 +9,6 @@ import psycopg
from pydantic import BaseModel
log = getLogger(__name__)
log.addHandler(logging.StreamHandler(sys.stdout))
log.level = logging.DEBUG
@@ -53,6 +52,7 @@ class PgDatabase(Database):
def connect_to_database(self):
return self.driver.connect(
autocommit=True,
host=os.getenv("DB_HOST"),
port=os.getenv("DB_PORT"),
user=os.getenv("DB_USERNAME"),
@@ -60,7 +60,6 @@ class PgDatabase(Database):
dbname=os.getenv("DB_NAME")
)
def init_db():
with PgDatabase() as db:
db.connection.execute("""
@@ -83,23 +82,24 @@ def drop_tables() -> None:
def create_user(user: UserInternal) -> bool:
with PgDatabase() as db:
try:
db.connection.execute(
"""
INSERT INTO users (id, name, lastname, email, hashed_password)
VALUES (%s, %s, %s, %s, %s)
""",
(user.id, user.name, user.lastname, user.email, user.hashed_password)
)
except psycopg.errors.UniqueViolation:
return False
try:
db.connection.execute(
"""
INSERT INTO users (id, name, lastname, email, hashed_password)
VALUES (%s, %s, %s, %s, %s)
""",
(user.id, user.name, user.lastname, user.email, user.hashed_password)
)
except psycopg.errors.UniqueViolation as e:
log.error(e)
return False
log.info("User inserted successfully")
log.debug(f"User {user.model_dump()} created successfully")
log.debug(f"Inserted following User:", user.model_dump())
return True
def get_user(email: str) -> UserInternal:
def get_user(email: str) -> UserInternal | None:
with PgDatabase() as db:
result = db.connection.execute(
"SELECT id, name, lastname, email, hashed_password FROM users WHERE email=%s",
@@ -110,5 +110,5 @@ def get_user(email: str) -> UserInternal:
return None
user = UserInternal.model_construct(**dict(zip(["id", "name", "lastname", "email", "hashed_password"], result)))
log.debug(f"User {user.model_dump()} retrieved successfully")
log.debug(f"Retrieved User succesfully: {user.model_dump()} ")
return user