PyYAML 中 yaml.load 和 yaml.SafeLoader 的区别

Question

这里是Python新手。

我安装了 PyYAML 并尝试使用我在网上找到的修改过的这段代码来解析给我的 YAML 文件。

import yaml

if __name__ == '__main__':
    try:
        foo = open("foo.txt","a+")
    except:
        print("Error in opening file.")

stream = open("s.yaml", 'r')
dictionary = yaml.load(stream)
#dictionary = yaml.SafeLoader(stream)
for key, value in dictionary.items():

    foo.write(key + " : " + str(value)+"\n")

然后我在输出中看到 yaml.load 由于安全问题而被弃用。所以我尝试使用 SafeLoader 来运行它。但这给了我错误

Traceback (most recent call last):
  File ".\parseYAML.py", line 11, in <module>
    for key, value in dictionary.items():
AttributeError: 'SafeLoader' object has no attribute 'items'

由于商业原因，我无法在此处发布实际的数据文件，但是有人对如何让 SafeLoader 工作有任何提示吗？

Answer 1

我们使用以下代码片段展示了使用 SafeLoader 的一些方法：（最后一行可能是您最感兴趣的，对于最简单的情况）

import yaml

env_variable_matcher = re.compile(r'<your custom pattern here>')

def env_variable_parser(loader, node):
    '''
    Parse a yaml value containing ${A-Z0-9_} as an environment variable
    '''
    ...
    return output

# Add support for yaml tag named env_var which matches a string containing
# an environment variable.  The environment variable will be replaced with its value
yaml.add_implicit_resolver('!env_var', env_variable_matcher, Loader=yaml.SafeLoader)
yaml.add_constructor('!env_var', env_variable_parser, Loader=yaml.SafeLoader)

with open(config_file, 'r') as c_file:
    config = yaml.safe_load(c_file)