yaml文件里引用yaml文件

原创已于 2022-06-27 21:51:33 修改 · 5k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#python #yaml

于 2021-04-16 00:22:02 首次发布

接口自动化专栏收录该内容

31 篇文章

订阅专栏

该博客介绍了如何使用Python的yaml库解析包含`!include`指令的yaml文件，通过自定义构造函数实现文件的嵌套加载。示例中展示了如何创建一个基于类的解决方案，利用元类注册自定义的`!include`构造函数，从而实现对yaml文件的递归加载。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

a.yaml文件内容：

a: 1
b:
- 2
- 3
c: !include b.yaml

b.yaml文件内容：

- 10
- [100, 200, 300]

python代码

# 基于类的解决方案，使用元类注册自定义构造函数。

import yaml
import os
import os.path

# 项目路径
project_path =os.path.split(os.path.realpath(__file__))[0].split('tools')[0]

class Loader(yaml.Loader): # 继承
    def __init__(self, stream):
        self._root = os.path.split(stream.name)[0]
        super(Loader,self).__init__(stream)

    def include(self, node):
        filename = os.path.join(self._root,self.construct_scalar(node))
        with open(filename,'r') as f:
            return yaml.load(f, Loader)

Loader.add_constructor('!include',Loader.include)

def get_yaml_data(fileDir):
    """
    读取 test.yaml 文件内容
    :param fileDir:
    :return:
    """
    # 1、在内存里加载这个文件
    # f = open(fileDir, 'r', encoding='utf-8')
    with open('a.yaml','r') as f:
        data = yaml.load(f, yaml.Loader)
    print(data)

    # 2、调用yaml读取文件
    # Loader=yaml.FullLoader 更加安全
    res = yaml.load(f, Loader=yaml.FullLoader)
    return res

if __name__ == '__main__':
    info = get_yaml_data(project_path + r'\test.yaml')
    print(info)

# 输出：
# {'a': 1,'b': [2, 3],'c': [10, [100,200, 300]]}