70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
import json
|
|
from typing import Any, Dict, Optional
|
|
|
|
from .path_manager import path_manager, file_manager
|
|
|
|
class ConfigManager:
|
|
"""Simplified config manager that uses centralized file management."""
|
|
|
|
def __init__(self, config_name: str, overwrite: bool = False):
|
|
self.config_name = config_name
|
|
self.data: Optional[Dict[str, Any]] = None
|
|
|
|
# Ensure the user config file exists
|
|
file_manager.ensure_user_config_file(config_name, overwrite=overwrite)
|
|
|
|
def load(self) -> Dict[str, Any]:
|
|
"""Load configuration data with user overrides."""
|
|
self.data = file_manager.load_config_with_overrides(self.config_name)
|
|
return self.data
|
|
|
|
def save(self, data: Optional[Dict[str, Any]] = None, user_override: bool = False) -> bool:
|
|
"""Save the configuration data to the appropriate user file."""
|
|
to_save = data if data is not None else self.data
|
|
if to_save is None:
|
|
raise ValueError("No data to save.")
|
|
|
|
if user_override:
|
|
target_file = path_manager.get_user_override_file_path(f"{self.config_name}.json")
|
|
else:
|
|
target_file = path_manager.get_user_file_path(f"{self.config_name}.json")
|
|
|
|
success = file_manager.save_json_file(target_file, to_save, f"{self.config_name} config")
|
|
if success:
|
|
print(f"Saved {self.config_name} to {target_file}.")
|
|
return success
|
|
|
|
# Traditional config managers for styles and prompts
|
|
styles_manager = ConfigManager("sage_styles")
|
|
sage_styles = styles_manager.load()
|
|
|
|
prompts_manager = ConfigManager("llm_prompts")
|
|
llm_prompts = prompts_manager.load()
|
|
|
|
metadata_template_manager = ConfigManager("metadata_templates")
|
|
metadata_templates = metadata_template_manager.load()
|
|
|
|
# Tag library management for prompt builder
|
|
tag_library_manager = ConfigManager("default_tag_library")
|
|
default_tag_library = tag_library_manager.load()
|
|
|
|
# Enhanced settings management using the new settings system
|
|
# This maintains backwards compatibility while providing enhanced functionality
|
|
try:
|
|
from .settings import get_settings, get_sage_config
|
|
# Use the new settings system
|
|
_enhanced_settings = get_settings()
|
|
settings_manager = _enhanced_settings._config_manager # Access the underlying ConfigManager
|
|
sage_config = get_sage_config() # Get settings in the old format for backwards compatibility
|
|
|
|
# Update the settings_manager.data to point to the enhanced settings
|
|
settings_manager.data = sage_config
|
|
|
|
print("SageUtils: Settings system loaded successfully.")
|
|
except ImportError as e:
|
|
print(f"SageUtils: Warning - Could not load settings system: {e}. Falling back to basic settings management.")
|
|
# Fallback to basic config manager
|
|
settings_manager = ConfigManager("config")
|
|
sage_config = settings_manager.load()
|
|
|