基于环境变量值转换字符串中的envvar风格子串

我需要处理包含${环境变量名}或$(环境变量名)格式子串的字符串，规则如下：

• 若占位符中的环境变量实际存在，就将占位符替换为该环境变量的对应值
• 若占位符中的环境变量不存在，则保留原占位符字符串不变

示例1 #

已存在的环境变量 #

ENV_VAR1 = "XYZ"
ENV_VAR2 = "123"

输入字符串 #

"something/${ENV_VAR1}/else|$(ENV_VAR2)"

输出结果 #

"something/XYZ/else|123"

示例2 #

已存在的环境变量 #

ENV_VAR1 = "XYZ"

输入字符串 #

"something/${ENV_VAR1}/else|$(ENV_VAR2)"

输出结果 #

"something/XYZ/else|$(ENV_VAR2)"

核心实现逻辑 #

1. 用正则表达式匹配所有${VAR}或$(VAR)格式的占位符，提取其中的环境变量名
2. 遍历每个匹配结果：
   • 检查当前环境中是否存在该变量
   • 存在则替换为变量对应的值，不存在则保留原占位符

Python 示例代码 #

import os
import re

def replace_env_vars(input_str):
    # 匹配 ${VAR} 或 $(VAR) 格式的占位符
    pattern = re.compile(r'\$\{([A-Za-z0-9_]+)\}|\$\(([A-Za-z0-9_]+)\)')
    
    def replace_match(match):
        # 提取环境变量名，两个分组分别对应两种格式
        var_name = match.group(1) or match.group(2)
        # 存在则返回变量值，否则返回原占位符
        return os.getenv(var_name, match.group(0))
    
    return pattern.sub(replace_match, input_str)

# 测试示例1
os.environ['ENV_VAR1'] = 'XYZ'
os.environ['ENV_VAR2'] = '123'
test_str1 = '"something/${ENV_VAR1}/else|$(ENV_VAR2)"'
print(replace_env_vars(test_str1))  # 输出: "something/XYZ/else|123"

# 测试示例2
del os.environ['ENV_VAR2']
test_str2 = '"something/${ENV_VAR1}/else|$(ENV_VAR2)"'
print(replace_env_vars(test_str2))  # 输出: "something/XYZ/else|$(ENV_VAR2)"

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