虽然“正确的 YAML”一词并不准确,但最好将其表述为
“YAML 输出看起来像你想要的,除了标签信息”,这幸运地给出了一些
有关您希望 YAML 外观的信息,因为转储对象的方法有无数种。
如果您使用ruamel.yaml 转储对象:
import sys
import ruamel.yaml
class MyObject:
def __init__(self, a, b):
self.a = a
self.b = b
self.c = [a, b]
data = dict(x=MyObject(42, -1))
yaml = ruamel.yaml.YAML(typ='unsafe')
yaml.dump(data, sys.stdout)
这给出了:
x: !!python/object:__main__.MyObject
a: 42
b: -1
c: [42, -1]
您有一个标签 !!python/object:__main__.MyObject(您的标签可能会有所不同,具体取决于
类被定义等)并且类的每个属性都被转储为映射的键。
有多种方法可以去除转储中的标签:
注册课程
添加一个名为 to_yaml() 的 classmethod 到您的每个类和
注册这些课程。你必须为你的每个班级都这样做,
但这样做可以让您使用安全翻斗车。一个例子说明如何
可以在
documentation
后处理
对输出进行后处理并删除标签是相当容易的,因为对象总是出现在行上
在映射之前,您可以从!!python 删除直到行尾
def strip_python_tags(s):
result = []
for line in s.splitlines():
idx = line.find("!!python/")
if idx > -1:
line = line[:idx]
result.append(line)
return '\n'.join(result)
yaml.encoding = None
yaml.dump(data, sys.stdout, transform=strip_python_tags)
这给出了:
x:
a: 42
b: -1
c: [42, -1]
由于 achors 在标签之前被转储,这个“从 !!python 剥离
直到行尾”,当您转储具有
多个参考。
更换自卸车
您还可以将映射的不安全转储程序更改为
识别用于对象的标签并将标签更改为“正常”
一个用于字典/映射(通常不输出标签)
yaml.representer.org_represent_mapping = yaml.representer.represent_mapping
def my_represent_mapping(tag, mapping, flow_style=None):
if tag.startswith("tag:yaml.org,2002:python/object"):
tag = u'tag:yaml.org,2002:map'
return yaml.representer.org_represent_mapping(tag, mapping, flow_style=flow_style)
yaml.representer.represent_mapping = my_represent_mapping
yaml.dump(data, sys.stdout)
这又给了一次:
x:
a: 42
b: -1
c: [42, -1]
这最后两种方法适用于您定义的所有 Python 类的所有实例,无需额外工作。