From f1b15357f9ce15b64b177b8993ef7741474f5b6f Mon Sep 17 00:00:00 2001 From: h7x4 Date: Mon, 8 Dec 2025 21:43:05 +0900 Subject: [PATCH] fixup! WIP --- dibbler/db.py | 6 +++--- dibbler/models/ProductCache.py | 1 + dibbler/models/UserCache.py | 1 + dibbler/queries/adjust_penalty.py | 1 + 4 files changed, 6 insertions(+), 3 deletions(-) diff --git a/dibbler/db.py b/dibbler/db.py index 3d81dac..24c5d08 100644 --- a/dibbler/db.py +++ b/dibbler/db.py @@ -6,11 +6,11 @@ from sqlalchemy.orm import sessionmaker from dibbler.conf import config if (url := config.get("database", "url")) is not None: - database_url = url + database_url = url elif (url_file := config.get("database", "url_file")) is not None: - with Path(url_file).open() as file: - database_url = file.read().strip() + with Path(url_file).open() as file: + database_url = file.read().strip() engine = create_engine(database_url) Session = sessionmaker(bind=engine) diff --git a/dibbler/models/ProductCache.py b/dibbler/models/ProductCache.py index b2ac8a4..f6bcb66 100644 --- a/dibbler/models/ProductCache.py +++ b/dibbler/models/ProductCache.py @@ -5,6 +5,7 @@ from sqlalchemy.orm import Mapped, mapped_column from dibbler.models import Base + class ProductCache(Base): product_id: Mapped[int] = mapped_column(Integer, primary_key=True) diff --git a/dibbler/models/UserCache.py b/dibbler/models/UserCache.py index 31ea604..7b35e52 100644 --- a/dibbler/models/UserCache.py +++ b/dibbler/models/UserCache.py @@ -5,6 +5,7 @@ from sqlalchemy.orm import Mapped, mapped_column from dibbler.models import Base + # More like user balance cash money flow, amirite? class UserBalanceCache(Base): user_id: Mapped[int] = mapped_column(Integer, primary_key=True) diff --git a/dibbler/queries/adjust_penalty.py b/dibbler/queries/adjust_penalty.py index bd1dadb..718c2c5 100644 --- a/dibbler/queries/adjust_penalty.py +++ b/dibbler/queries/adjust_penalty.py @@ -6,6 +6,7 @@ from dibbler.queries.current_penalty import current_penalty # TODO: this type of transaction should be password protected. # the password can be set as a string literal in the config file. + def adjust_penalty( sql_session: Session, user_id: int,