43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
"""
|
||
配置管理模块
|
||
使用OmegaConf读取config.yml和local_config.yml
|
||
"""
|
||
from pathlib import Path
|
||
from omegaconf import OmegaConf
|
||
|
||
def load_config():
|
||
"""
|
||
加载配置文件
|
||
|
||
Returns:
|
||
DictConfig: 合并后的配置对象
|
||
"""
|
||
static_path = Path("static")
|
||
|
||
# 配置文件路径
|
||
base_config_path = static_path / "config.yml"
|
||
local_config_path = static_path / "local_config.yml"
|
||
|
||
# 读取基础配置
|
||
base_config = OmegaConf.create({})
|
||
if base_config_path.exists():
|
||
base_config = OmegaConf.load(base_config_path)
|
||
|
||
# 读取本地配置
|
||
local_config = OmegaConf.create({})
|
||
if local_config_path.exists():
|
||
local_config = OmegaConf.load(local_config_path)
|
||
|
||
# 合并配置，local_config优先级更高
|
||
config = OmegaConf.merge(base_config, local_config)
|
||
|
||
# 把paths下的所有值pathlib化
|
||
if hasattr(config, "paths"):
|
||
for key, value in config.paths.items():
|
||
config.paths[key] = Path(value)
|
||
|
||
return config
|
||
|
||
# 导出配置对象
|
||
CONFIG = load_config()
|