added config file detection to migration env.py file

This commit is contained in:
Linus Vogel 2026-02-01 21:52:10 +01:00
parent 21b1c70989
commit 6c37589d1a

View File

@ -1,3 +1,4 @@
import os
import tomllib import tomllib
from inspect import stack from inspect import stack
from logging.config import fileConfig from logging.config import fileConfig
@ -30,6 +31,20 @@ target_metadata = Base.metadata
# ... etc. # ... etc.
def get_custom_url(): def get_custom_url():
# get the correct path
paths: list[str] = [
"./pillar_tool.toml",
"/etc/pillar_tool/config.toml",
]
selected_path = None
for path in paths:
if os.path.exists(path):
selected_path = path
break
if selected_path is None:
raise RuntimeError("No loadable config found!")
with open("./pillar_tool.toml", 'rb') as f: with open("./pillar_tool.toml", 'rb') as f:
data = tomllib.load(f) data = tomllib.load(f)
user = data['db']['user'] user = data['db']['user']