From 7d2ae55e057b8b6fe8844928950b17fe0a1e91e2 Mon Sep 17 00:00:00 2001 From: h7x4 Date: Wed, 10 Dec 2025 15:53:45 +0900 Subject: [PATCH] assert `database_url` is present --- dibbler/db.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/dibbler/db.py b/dibbler/db.py index 3d81dac..299a091 100644 --- a/dibbler/db.py +++ b/dibbler/db.py @@ -5,12 +5,15 @@ from sqlalchemy.orm import sessionmaker from dibbler.conf import config -if (url := config.get("database", "url")) is not None: - database_url = url +database_url: str | None = None +if (url := config.get("database", "url")) is not None: + 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() + +assert database_url is not None, "Database URL must be specified in config" engine = create_engine(database_url) Session = sessionmaker(bind=engine)