feat(data): use generate_default_toml() in load_config fallback
This commit is contained in:
1 file changed
+17
-1
+17
-1
@@ -49,7 +49,13 @@ def load_config():
|
||||
service = start_config_service()
|
||||
return service.get_all()
|
||||
except ImportError:
|
||||
# Fallback to direct file loading
|
||||
pass
|
||||
|
||||
try:
|
||||
from utils.constants import generate_default_toml
|
||||
except ImportError:
|
||||
generate_default_toml = None
|
||||
|
||||
config = {}
|
||||
|
||||
if os.path.exists(CONFIG_FILE):
|
||||
@@ -60,6 +66,16 @@ def load_config():
|
||||
config = dict(tomlkit.load(f))
|
||||
except Exception as e:
|
||||
logger.error(f"Error loading config: {e}")
|
||||
elif generate_default_toml:
|
||||
try:
|
||||
raw = generate_default_toml()
|
||||
with open(CONFIG_FILE, "w") as f:
|
||||
f.write(raw)
|
||||
import tomlkit
|
||||
|
||||
config = dict(tomlkit.loads(raw))
|
||||
except Exception as e:
|
||||
logger.error(f"Error generating default config: {e}")
|
||||
|
||||
return config
|
||||
|
||||
|
||||
Reference in new issue
Block a user