"""
配置文件读写与合并逻辑
"""
import yaml
from pathlib import Path
from typing import Dict, Optional
class ConfigIO:
def __init__(self, config_path: str = None):
self.config_path = Path(config_path) if config_path else Path.home() / '.remote-terminal' / 'config.yaml'
def load_config(self) -> Dict:
if not self.config_path.exists():
return {}
with self.config_path.open('r', encoding='utf-8') as f:
content = f.read()
try:
return yaml.safe_load(content) if content and content.strip() else {}
except Exception as e:
print(f"⚠️ 配置文件解析失败: {e}")
return {}
def save_config(self, config: dict, merge: bool = True):
final_cfg = config
if merge and self.config_path.exists():
existing = self.load_config().get('servers', {})
existing.update(config.get('servers', {}))
final_cfg = {'servers': existing}
# 确保父目录存在
self.config_path.parent.mkdir(parents=True, exist_ok=True)
with self.config_path.open('w', encoding='utf-8') as f:
yaml.dump(final_cfg, f, allow_unicode=True)
print(f"\n✅ 配置已保存至 {self.config_path}")
