nodes/arcum42-ComfyUI_SageUtils
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cleanup config manager.

Browse Source
This commit is contained in:
arcum42
2025-06-12 16:44:03 -07:00
parent 71847a9786
commit 186829bb98
1 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/config_manager.py
+20 -12
View File
@@ -4,11 +4,11 @@ import pathlib
import folder_paths
base_path = pathlib.Path(os.path.dirname(os.path.realpath(__file__))).parent
base_path = pathlib.Path(__file__).resolve().parent.parent
users_path = pathlib.Path(folder_paths.get_user_directory())
sage_users_path = users_path / "default" / "SageUtils"
assets_path = base_path / "assets"
os.makedirs(str(sage_users_path), exist_ok=True)
sage_users_path.mkdir(parents=True, exist_ok=True)
def deep_merge_dicts(a, b):
"""Recursively merge dict b into dict a and return the result."""
@@ -28,7 +28,7 @@ class ConfigManager:
def __init__(self, config_name, overwrite=False):
self.config_name = config_name
self.data = None
self.base_file = pathlib.Path(assets_path) / f"{config_name}.json"
self.base_file = assets_path / f"{config_name}.json"
self.user_file = sage_users_path / f"{config_name}.json"
self.user_override_file = sage_users_path / f"{config_name}_user.json"
self.ensure_user_file(overwrite=overwrite)
@@ -37,20 +37,19 @@ class ConfigManager:
if not self.user_file.is_file() or overwrite:
if not overwrite:
print(f"No {self.config_name}.json file found in user directory.")
if self.base_file.is_file():
with open(self.base_file, "r") as read_file:
with self.base_file.open("r", encoding="utf-8") as read_file:
temp = json.load(read_file)
with open(self.user_file, "w") as write_file:
with self.user_file.open("w", encoding="utf-8") as write_file:
json.dump(temp, write_file, separators=(",", ":"), sort_keys=True, indent=4)
print(f"Copied default {self.config_name}.json to {str(self.user_file)}.")
print(f"Copied default {self.config_name}.json to {self.user_file}.")
def load(self):
configs = []
for path in [self.user_file, self.user_override_file]:
for path in (self.user_file, self.user_override_file):
if path.is_file():
try:
with path.open(mode="r", errors='ignore', encoding='utf-8') as read_file:
with path.open("r", encoding="utf-8", errors="ignore") as read_file:
configs.append(json.load(read_file))
print(f"Loading {self.config_name} from {path}")
except Exception as e:
@@ -66,11 +65,20 @@ class ConfigManager:
self.data = merged
return self.data
# Usage examples:
styles_manager = ConfigManager("sage_styles", overwrite=True)
def save(self, data=None, user_override=False):
"""Save the configuration data to the appropriate user file."""
target_file = self.user_override_file if user_override else self.user_file
to_save = data if data is not None else self.data
if to_save is None:
raise ValueError("No data to save.")
with target_file.open("w", encoding="utf-8") as write_file:
json.dump(to_save, write_file, separators=(",", ":"), sort_keys=True, indent=4)
print(f"Saved {self.config_name} to {target_file}.")
styles_manager = ConfigManager("sage_styles")
sage_styles = styles_manager.load()
prompts_manager = ConfigManager("llm_prompts", overwrite=True)
prompts_manager = ConfigManager("llm_prompts")
llm_prompts = prompts_manager.load()
settings_manager = ConfigManager("config")