如何在要使用python脚本解析的yaml文件中替换环境变量值

npatel 发表于 Linux

258

npatel：

我需要在yaml文件中使用环境变量“ PATH”，该环境变量需要使用脚本进行解析。

这是我在终端上设置的环境变量：

$ echo $PATH
/Users/abc/Downloads/tbwork

这是我的sample.yml：

---
Top: ${PATH}/my.txt
Vars:
- a
- b

当我用脚本解析该Yaml文件时，看不到PATH变量的实际值。

这是我的脚本：

import yaml
import os
import sys

stream = open("sample.yml", "r")
docs = yaml.load_all(stream)
for doc in docs:
    for k,v in doc.items():
        print k, "->", v
    print "\n",

输出：

Top -> ${PATH}/my.txt
Vars -> ['a', 'b']

预期输出为：

Top -> /Users/abc/Downloads/tbwork/my.txt
Vars -> ['a', 'b']

如果我做错了事，有人可以帮我弄清楚做事的正确方法吗？

dmchk：

PY-yaml库默认情况下不解析环境变量。您需要定义一个隐式解析器，该解析器将找到定义环境变量的正则表达式并执行函数来对其进行解析。

您可以通过yaml.add_implicit_resolver和完成此操作yaml.add_constructor。在下面的代码中，您将定义一个解析器，该解析器将与YAML值中的$ {env variable}匹配，并调用函数path_constructor来查找环境变量。

import yaml
import re
import os

path_matcher = re.compile(r'\$\{([^}^{]+)\}')
def path_constructor(loader, node):
  ''' Extract the matched value, expand env variable, and replace the match '''
  value = node.value
  match = path_matcher.match(value)
  env_var = match.group()[2:-1]
  return os.environ.get(env_var) + value[match.end():]

yaml.add_implicit_resolver('!path', path_matcher)
yaml.add_constructor('!path', path_constructor)

data = """
env: ${VAR}/file.txt
other: file.txt
"""

if __name__ == '__main__':
  p = yaml.load(data, Loader=yaml.FullLoader)
  print(os.environ.get('VAR')) ## /home/abc
  print(p['env']) ## /home/abc/file.txt

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。