python - 如何读取/解析目录中的所有 yaml 文件？

Question

嗯，我正在尝试使用 python 2.7 和 pyyaml 读取目录中的 utf-8 yaml 文件集合。我尝试使用 os.listdir(os.getcwd()) ，但它实际上返回一个文件列表，我需要读取文件并显示内容(键，值)。该目录只有yaml文件，并且已经安装了pyyaml库。

yaml文件的所有内容都与此类似:

fruits:
 - bananas
 - apples
 - grapes

有人可以帮助我吗？

Answer 1

你可以试试这个:

demo_file1.yaml的内容:

fruits:
  - bananas
  - apples
  - grapes

demo_file2.yaml的内容:

veg:
  - cauli
  - lady finger
  - tomato

代码:

import yaml
import glob

files = glob.glob("/path/to/directory/*.yaml") # list of all .yaml files in a directory 

def read_yaml_file(filename):
    with open(filename, 'r') as stream:
        try:
            print(yaml.safe_load(stream))
        except yaml.YAMLError as exc:
            print(exc)

for file in files:
    read_yaml_file(file)

输出:

{'fruits': ['bananas', 'apples', 'grapes']}
{'veg': ['cauli', 'lady finger', 'tomato']}