82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
# config_loader.py - 读取config.toml的示例代码
|
||
|
||
import toml
|
||
import os
|
||
from pathlib import Path
|
||
from typing import Any, Dict, Optional
|
||
|
||
class ConfigLoader:
|
||
"""TOML配置文件加载器"""
|
||
|
||
def __init__(self, config_path: Optional[str] = None):
|
||
"""
|
||
初始化配置加载器
|
||
|
||
Args:
|
||
config_path: 配置文件路径，如果为None则自动查找
|
||
"""
|
||
if config_path is None:
|
||
# 自动查找配置文件
|
||
config_path = self._find_config_file()
|
||
|
||
self.config_path = Path(config_path)
|
||
self._config: Dict[str, Any] = {}
|
||
self.load()
|
||
|
||
def _find_config_file(self) -> str:
|
||
"""自动查找配置文件"""
|
||
possible_paths = [
|
||
"config.toml",
|
||
"config/config.toml",
|
||
"~/.config/myapp/config.toml",
|
||
"/etc/myapp/config.toml"
|
||
]
|
||
|
||
for path in possible_paths:
|
||
expanded_path = os.path.expanduser(path)
|
||
if os.path.exists(expanded_path):
|
||
return expanded_path
|
||
|
||
raise FileNotFoundError("未找到配置文件")
|
||
|
||
def load(self) -> None:
|
||
"""加载配置文件"""
|
||
if not self.config_path.exists():
|
||
raise FileNotFoundError(f"配置文件不存在: {self.config_path}")
|
||
|
||
with open(self.config_path, 'r', encoding='utf-8') as f:
|
||
self._config = toml.load(f)
|
||
|
||
def get(self, key: str, default: Any = None) -> Any:
|
||
"""获取配置值，支持点号分隔的路径"""
|
||
keys = key.split('.')
|
||
value = self._config
|
||
|
||
try:
|
||
for k in keys:
|
||
value = value[k]
|
||
return value
|
||
except (KeyError, TypeError):
|
||
return default
|
||
|
||
def get_section(self, section: str) -> Dict[str, Any]:
|
||
"""获取整个配置节"""
|
||
return self._config.get(section, {})
|
||
|
||
def reload(self) -> None:
|
||
"""重新加载配置文件"""
|
||
self.load()
|
||
|
||
def __getitem__(self, key: str) -> Any:
|
||
"""支持字典式访问"""
|
||
return self.get(key)
|
||
|
||
def __contains__(self, key: str) -> bool:
|
||
"""检查配置是否存在"""
|
||
return self.get(key) is not None
|
||
|
||
@property
|
||
def all(self) -> Dict[str, Any]:
|
||
"""获取所有配置"""
|
||
return self._config
|
||
|