在PyYAML中创建自定义标签

您的PyYAML类有一些问题：

1. yaml_tag

   是区分大小写的，

   !Env

   并且

   !ENV

   是不同的标记。
2. 因此，根据文档，

   yaml.YAMLObject

   使用元类来定义自身，并具有针对这些情况的默认值

   to_yaml

   和

   from_yaml

   功能。但是，默认情况下，这些函数要求您对自定义标记（在这种情况下

   !ENV

   ）的参数是 mapping 。因此，要使用默认功能，您的

   defaults.yaml

   文件必须看起来像这样（例如）：

example: !ENV {env_var: "PWD", test: "test"}

然后，您的代码将保持不变，在我的情况下，

print(settings)

{'example': /home/Fred}

load

safe_load

你仍然可以很容易地使用你的YAML文件格式，

example: !ENVfoo

to_yaml

from_yaml

EnvTag

所以：

import osimport yamlclass EnvTag(yaml.YAMLObject):    yaml_tag = u'!ENV'    def __init__(self, env_var):        self.env_var = env_var    def __repr__(self):        v = os.environ.get(self.env_var) or ''        return 'EnvTag({}, contains={})'.format(self.env_var, v)    @classmethod    def from_yaml(cls, loader, node):        return EnvTag(node.value)    @classmethod    def to_yaml(cls, dumper, data):        return dumper.represent_scalar(cls.yaml_tag, data.env_var)# Required for safe_loadyaml.SafeLoader.add_constructor('!ENV', EnvTag.from_yaml)# Required for safe_dumpyaml.SafeDumper.add_multi_representer(EnvTag, EnvTag.to_yaml)settings_file = open('defaults.yaml', 'r')settings = yaml.safe_load(settings_file)print(settings)s = yaml.safe_dump(settings)print(s)

运行该程序时，输出：

{'example': EnvTag(foo, contains=)}{example: !ENV 'foo'}

此代码的优点是（1）使用原始的pyyaml，因此无需额外安装，并且（2）添加了一个表示符。:)