mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-07 19:27:24 -04:00
Bill
7e95ce356b
Add merge_configs function that performs root-level merging of custom config into default config. Custom config sections completely replace default sections. Implementation does not mutate input dictionaries. Includes comprehensive tests for: - Empty custom config - Section override behavior - Adding new sections - Non-mutating behavior All 7 tests pass.
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
"""Configuration merging and validation for AI-Trader."""
|
|
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Dict, Any, Optional
|
|
|
|
|
|
class ConfigValidationError(Exception):
|
|
"""Raised when config validation fails."""
|
|
pass
|
|
|
|
|
|
def load_config(path: str) -> Dict[str, Any]:
|
|
"""
|
|
Load and parse JSON config file.
|
|
|
|
Args:
|
|
path: Path to JSON config file
|
|
|
|
Returns:
|
|
Parsed config dictionary
|
|
|
|
Raises:
|
|
ConfigValidationError: If file not found or invalid JSON
|
|
"""
|
|
config_path = Path(path)
|
|
|
|
if not config_path.exists():
|
|
raise ConfigValidationError(f"Config file not found: {path}")
|
|
|
|
try:
|
|
with open(config_path, 'r') as f:
|
|
return json.load(f)
|
|
except json.JSONDecodeError as e:
|
|
raise ConfigValidationError(f"Invalid JSON in {path}: {e}")
|
|
|
|
|
|
def merge_configs(default: Dict[str, Any], custom: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""
|
|
Merge custom config into default config (root-level override).
|
|
|
|
Custom config sections completely replace default sections.
|
|
Does not mutate input dictionaries.
|
|
|
|
Args:
|
|
default: Default configuration dict
|
|
custom: Custom configuration dict (overrides)
|
|
|
|
Returns:
|
|
Merged configuration dict
|
|
"""
|
|
merged = dict(default) # Shallow copy
|
|
|
|
for key, value in custom.items():
|
|
merged[key] = value
|
|
|
|
return merged
|