如何用python yaml解析内部变量？

pip install pyyaml

import yaml

# 假设这是你的YAML数据，其中包含了一些内部变量
yaml_data = """
database:
  host: localhost
  port: 5432
  name: my_database_${ENV_VAR}
"""

# 你可以定义一个函数来替换这些内部变量
def replace_env_vars(data, env_vars):
    if isinstance(data, dict):
        for key, value in data.items():
            data[key] = replace_env_vars(value, env_vars)
    elif isinstance(data, list):
        for i, item in enumerate(data):
            data[i] = replace_env_vars(item, env_vars)
    elif isinstance(data, str):
        for var in env_vars:
            data = data.replace('${' + var + '}', env_vars[var])
    return data

# 定义环境变量
env_vars = {
    'ENV_VAR': 'production'
}

# 解析YAML数据
parsed_data = yaml.safe_load(yaml_data)

# 替换内部变量
parsed_data_with_vars = replace_env_vars(parsed_data, env_vars)

print(parsed_data_with_vars)

try:
    parsed_data = yaml.safe_load(yaml_data)
except yaml.YAMLError as exc:
    print("Error in YAML data:", exc)