130 lines
4.9 KiB
Python
130 lines
4.9 KiB
Python
"""
|
|
Configuration Manager for the application.
|
|
Handles loading and saving of application settings to a JSON file.
|
|
"""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Dict, Any, Optional, List
|
|
|
|
from . import logger
|
|
from ..core.export_profiles import ExportProfile, ExportField
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
class ConfigManager:
|
|
"""Manages loading and saving configuration from/to a JSON file."""
|
|
|
|
def __init__(self, config_path: Path):
|
|
self.config_path: Path = config_path
|
|
self.config: Dict[str, Any] = self._get_default_config()
|
|
self.export_profiles: List[ExportProfile] = []
|
|
|
|
@staticmethod
|
|
def _get_default_config() -> Dict[str, Any]:
|
|
"""Provides the default configuration structure."""
|
|
default_profile = ExportProfile(
|
|
name="Default",
|
|
fields=[
|
|
ExportField(column_name="batch_id", data_path="batch_id"),
|
|
ExportField(
|
|
column_name="timetag",
|
|
data_path="main_header.ge_header.signal_descr.ttag",
|
|
),
|
|
ExportField(
|
|
column_name="file_batch_counter",
|
|
data_path="main_header.ge_header.signal_descr.batch_counter",
|
|
),
|
|
],
|
|
)
|
|
return {
|
|
"last_opened_out_file": "",
|
|
"last_opened_rec_file": "",
|
|
"last_out_output_dir": "",
|
|
"last_rec_output_dir": "",
|
|
"active_out_export_profile_name": "Default",
|
|
"active_rec_csv_profile_name": "Default",
|
|
"active_rec_json_profile_name": "Default",
|
|
"export_profiles": [default_profile.to_dict()],
|
|
}
|
|
|
|
def load_config(self) -> None:
|
|
"""Loads the main application configuration from the JSON file."""
|
|
log.info(f"Attempting to load configuration from: {self.config_path}")
|
|
if self.config_path.is_file():
|
|
try:
|
|
with open(self.config_path, "r", encoding="utf-8") as f:
|
|
loaded_data = json.load(f)
|
|
|
|
default_conf = self._get_default_config()
|
|
for key in default_conf:
|
|
self.config[key] = loaded_data.get(key, default_conf[key])
|
|
|
|
profiles_data = loaded_data.get("export_profiles", [])
|
|
self.export_profiles = [
|
|
ExportProfile.from_dict(p_data) for p_data in profiles_data
|
|
]
|
|
|
|
if not self.export_profiles:
|
|
log.warning("No export profiles found in config, loading defaults.")
|
|
self._load_default_profiles()
|
|
|
|
log.info(
|
|
f"Configuration loaded successfully. Found {len(self.export_profiles)} export profiles."
|
|
)
|
|
|
|
except (json.JSONDecodeError, IOError, KeyError) as e:
|
|
log.error(f"Failed to load or parse config file: {e}. Using defaults.")
|
|
self._load_defaults()
|
|
else:
|
|
log.warning("Configuration file not found. Using default settings.")
|
|
self._load_defaults()
|
|
|
|
def _load_defaults(self):
|
|
"""Loads all default settings into the manager."""
|
|
self.config = self._get_default_config()
|
|
self._load_default_profiles()
|
|
|
|
def _load_default_profiles(self):
|
|
"""Loads just the default profiles from the default config."""
|
|
default_conf = self._get_default_config()
|
|
profiles_data = default_conf.get("export_profiles", [])
|
|
self.export_profiles = [
|
|
ExportProfile.from_dict(p_data) for p_data in profiles_data
|
|
]
|
|
|
|
def save_config(self) -> None:
|
|
"""Saves the current configuration to the JSON file."""
|
|
log.info(f"Saving configuration to: {self.config_path}")
|
|
|
|
data_to_save = self.config.copy()
|
|
data_to_save["export_profiles"] = [
|
|
profile.to_dict() for profile in self.export_profiles
|
|
]
|
|
|
|
try:
|
|
self.config_path.parent.mkdir(parents=True, exist_ok=True)
|
|
with open(self.config_path, "w", encoding="utf-8") as f:
|
|
json.dump(data_to_save, f, indent=4)
|
|
log.info("Configuration saved successfully.")
|
|
except IOError as e:
|
|
log.error(f"Failed to save configuration file: {e}")
|
|
|
|
def get(self, key: str, default: Optional[Any] = None) -> Any:
|
|
"""Gets a value from the configuration."""
|
|
return self.config.get(key, default)
|
|
|
|
def set(self, key: str, value: Any) -> None:
|
|
"""Sets a value in the configuration."""
|
|
self.config[key] = value
|
|
|
|
def get_export_profiles(self) -> List[ExportProfile]:
|
|
"""Returns the list of loaded export profiles."""
|
|
return self.export_profiles
|
|
|
|
def save_export_profiles(self, profiles: List[ExportProfile]):
|
|
"""Updates the list of export profiles and saves the config."""
|
|
self.export_profiles = profiles
|
|
self.save_config()
|