db: handle database.url_file
This commit is contained in:
@ -1,7 +1,16 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
|
||||||
from sqlalchemy import create_engine
|
from sqlalchemy import create_engine
|
||||||
from sqlalchemy.orm import sessionmaker
|
from sqlalchemy.orm import sessionmaker
|
||||||
|
|
||||||
from dibbler.conf import config
|
from dibbler.conf import config
|
||||||
|
|
||||||
engine = create_engine(config.get("database", "url"))
|
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()
|
||||||
|
|
||||||
|
engine = create_engine(database_url)
|
||||||
Session = sessionmaker(bind=engine)
|
Session = sessionmaker(bind=engine)
|
||||||
|
Reference in New Issue
Block a user