feat(data): use generate_default_toml() in load_config fallback

This commit is contained in:
sankalp committed 2026-07-30 22:52:01 +05:45
1 parent 3aa0d34fea
commit 514868fe51
1 file changed
+17 -1
+17 -1
View File
@@ -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