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)