Python读取YAML文件时如何自动替换$符号引用的变量值？

结论 #

PyYAML 没有内置支持类Shell格式$变量名的自动替换功能，该语法不属于YAML官方标准，所以你要么自行实现简单的替换逻辑，要么使用支持该特性的第三方配置库。

方案1：自行实现轻量替换逻辑（无需额外依赖） #

针对简单的顶层变量引用场景，用正则匹配替换即可，示例代码如下：

import yaml
import re

# 匹配 $VAR 和 ${VAR} 两种常见占位符格式
var_pattern = re.compile(r"\$([A-Za-z0-9_]+)|\$\{([A-Za-z0-9_]+)\}")

def resolve_config_vars(config: dict) -> dict:
    # 第一步：收集所有无占位符的常量变量
    const_vars = {k: v for k, v in config.items() if isinstance(v, str) and "$" not in v}
    # 第二步：遍历替换所有占位符
    for key, value in config.items():
        if not isinstance(value, str):
            continue
        # 匹配替换回调
        def replace_match(match: re.Match) -> str:
            var_name = match.group(1) or match.group(2)
            # 变量不存在则保留原占位符，避免报错
            return const_vars.get(var_name, match.group(0))
        config[key] = var_pattern.sub(replace_match, value)
    return config

# 原有读取逻辑
with open("../etc/test.yml", "r") as f:
    raw_config = yaml.safe_load(f)
resolved_config = resolve_config_vars(raw_config)
print(resolved_config)

运行后将得到你期望的输出结果。如果存在嵌套配置、多级变量引用的场景，在此基础上补充递归遍历、循环引用校验逻辑即可。

方案2：使用支持变量引用的第三方库（适合复杂配置场景） #

如果你的配置有嵌套结构、多配置文件合并、环境变量注入等复杂需求，可以直接用成熟的配置库，例如OmegaConf，原生支持配置内部变量引用，使用示例如下：

1. 先修改YAML配置为兼容的占位符格式：

PATH1: /this/is/the/first/path
PATH2: ${PATH1}/plus/second/path

2. 代码示例：

from omegaconf import OmegaConf

config = OmegaConf.load("../etc/test.yml")
# resolve=True 表示自动解析所有变量引用，转成普通Python字典
print(OmegaConf.to_container(config, resolve=True))

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