Projects/dibbler
10
3
Fork 0
Code Issues 26 Pull Requests 1 Actions Activity

fixup! WIP
All checks were successful
Run tests / run-tests (push) Successful in 44s
Details

Browse Source
This commit is contained in:
Oystein Kristoffer Tveit
2025-12-09 17:02:28 +09:00
parent 7199cbf34a
commit 1515eb6dff
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dibbler/db.py
View File

@@ -5,12 +5,15 @@ from sqlalchemy.orm import sessionmaker
from dibbler.conf import config from dibbler.conf import config
database_url: str | None = None
if (url := config.get("database", "url")) is not None: 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: elif (url_file := config.get("database", "url_file")) is not None:
with Path(url_file).open() as file: with Path(url_file).open() as file:
database_url = file.read().strip() database_url = file.read().strip()
assert database_url is not None, "Database URL must be specified in config"
engine = create_engine(database_url) engine = create_engine(database_url)
Session = sessionmaker(bind=engine) Session = sessionmaker(bind=engine)