diff --git a/dibbler/db.py b/dibbler/db.py index 24c5d08..299a091 100644 --- a/dibbler/db.py +++ b/dibbler/db.py @@ -5,12 +5,15 @@ from sqlalchemy.orm import sessionmaker from dibbler.conf import config +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() +assert database_url is not None, "Database URL must be specified in config" + engine = create_engine(database_url) Session = sessionmaker(bind=engine)