13bc4520bf
Major changes: - Add dataclass-based configuration schema with validation (config/schema.py) - Create centralized config loader and default values (config/defaults.py, config/loader.py) - Remove duplicate database_config.py, merge into unified structure - Consolidate browser settings into ERP config - Add batch_size parameter support to extractor Bug fixes: - Fix settings save error by updating config paths (browser.* → erp.*) - Fix batch_size not being applied in data extraction Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
119 lines
2.7 KiB
Python
119 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
配置管理器
|
|
|
|
负责加载、保存和管理用户配置。
|
|
"""
|
|
import os
|
|
from typing import TYPE_CHECKING
|
|
from config.loader import ConfigLoader
|
|
from config.schema import AppConfig
|
|
from config.defaults import DEFAULT_SETTINGS_DICT
|
|
|
|
# 类型提示时导入，避免循环引用
|
|
if TYPE_CHECKING:
|
|
from config.schema import ERPConfig, DatabaseConfig, PathConfig, ExtractionConfig
|
|
|
|
|
|
class ConfigManager:
|
|
"""配置管理器"""
|
|
|
|
def __init__(self, config_file: str = "config/user_settings.json"):
|
|
"""
|
|
初始化配置管理器
|
|
|
|
Args:
|
|
config_file: 配置文件路径
|
|
"""
|
|
self.config_file = config_file
|
|
self.config: AppConfig = ConfigLoader.load(config_file)
|
|
|
|
# 验证配置
|
|
errors = self.config.validate()
|
|
if errors:
|
|
print("配置验证失败：")
|
|
for error in errors:
|
|
print(f" - {error}")
|
|
|
|
def save(self) -> bool:
|
|
"""
|
|
保存配置到文件
|
|
|
|
Returns:
|
|
保存是否成功
|
|
"""
|
|
return ConfigLoader.save(self.config, self.config_file)
|
|
|
|
def get(self, key: str, default=None):
|
|
"""
|
|
获取配置项
|
|
|
|
支持点号分隔的路径，如 "erp.url"
|
|
|
|
Args:
|
|
key: 配置键
|
|
default: 默认值
|
|
|
|
Returns:
|
|
配置值
|
|
"""
|
|
keys = key.split('.')
|
|
value = self.config
|
|
|
|
try:
|
|
for k in keys:
|
|
value = getattr(value, k)
|
|
return value
|
|
except (AttributeError, TypeError):
|
|
return default
|
|
|
|
def set(self, key: str, value) -> None:
|
|
"""
|
|
设置配置项
|
|
|
|
支持点号分隔的路径，如 "erp.url"
|
|
|
|
Args:
|
|
key: 配置键
|
|
value: 配置值
|
|
"""
|
|
keys = key.split('.')
|
|
obj = self.config
|
|
|
|
# 导航到父对象
|
|
for k in keys[:-1]:
|
|
obj = getattr(obj, k)
|
|
|
|
# 设置最终值
|
|
setattr(obj, keys[-1], value)
|
|
|
|
def reset_to_defaults(self) -> None:
|
|
"""重置为默认配置"""
|
|
self.config = ConfigLoader.load("default") # 重新加载默认配置
|
|
self.save()
|
|
|
|
@property
|
|
def erp(self):
|
|
"""获取 ERP 配置"""
|
|
return self.config.erp
|
|
|
|
@property
|
|
def database(self):
|
|
"""获取数据库配置"""
|
|
return self.config.database
|
|
|
|
@property
|
|
def paths(self):
|
|
"""获取路径配置"""
|
|
return self.config.paths
|
|
|
|
@property
|
|
def extraction(self):
|
|
"""获取提取配置"""
|
|
return self.config.extraction
|
|
|
|
|
|
# 为了向后兼容，保留旧版本的导入
|
|
DEFAULT_SETTINGS = DEFAULT_SETTINGS_DICT
|