如何在Python中读取YAML文件并实现变量替换？

针对YAML配置里的${xxx}格式变量替换需求，这里提供两种可行方案：

方案一：自定义递归替换函数（无额外依赖） #

如果不想引入第三方库，可以自己实现递归遍历配置的逻辑，识别并替换变量路径：

import yaml
from typing import Dict, Any

def resolve_variables(data: Dict[str, Any], root: Dict[str, Any]) -> Dict[str, Any]:
    resolved_data = {}
    for key, value in data.items():
        if isinstance(value, dict):
            # 递归处理嵌套的字典结构
            resolved_data[key] = resolve_variables(value, root)
        elif isinstance(value, str) and value.startswith("${") and value.endswith("}"):
            # 提取变量路径，比如 "${queues.mail-queue-name}" 转为 "queues.mail-queue-name"
            var_path = value[2:-1].strip()
            current_node = root
            try:
                # 按点分割路径，逐层查找对应值
                for path_part in var_path.split("."):
                    current_node = current_node[path_part]
                resolved_data[key] = current_node
            except KeyError:
                # 变量不存在时，可选择保留原字符串或抛出异常，这里默认保留
                resolved_data[key] = value
        else:
            resolved_data[key] = value
    return resolved_data

# 使用示例
with open('application.yaml', 'r') as file:
    raw_config = yaml.safe_load(file)

resolved_config = resolve_variables(raw_config, raw_config)

print(resolved_config['queues']['mail-queue-name'])  # 输出 mail-app-test
print(resolved_config['queues']['mail-dlq-name'])    # 输出 mail-app-test-dlq

特点 #

• 无需额外安装依赖，完全自定义替换逻辑
• 支持多层嵌套的变量路径（如parent.child.key）
• 可灵活处理变量不存在的场景（比如抛出KeyError或保留原字符串）

方案二：使用支持变量插值的第三方库 #

如果需要更完善的配置管理能力，推荐使用confuse库，它内置了配置变量引用的功能：

步骤1：安装依赖 #

pip install confuse

步骤2：使用示例 #

import confuse

# 初始化配置对象，第一个参数是应用名称，第二个参数是模块名
config = confuse.Configuration('MyApp', __name__)
config.set_file('application.yaml')

# 获取解析后的配置值
print(config['queues']['mail-queue-name'].get())  # 输出 mail-app-test
print(config['queues']['mail-dlq-name'].get())    # 输出 mail-app-test-dlq

特点 #

• 内置变量插值，无需手动写替换逻辑
• 支持环境变量引用（比如${ENV_VAR}）
• 提供配置验证、默认值设置等额外功能，适合复杂配置场景

内容的提问来源于stack exchange，提问作者Peppe