在对象上执行 YAML 转储时,我试图隐藏某些属性。我试过 this回答并允许您转储,但加载不起作用。
import yaml
from copy import deepcopy
class SecretYamlObject(yaml.YAMLObject):
hidden_fields = []
@classmethod
def to_yaml(cls, dumper, data):
new_data = deepcopy(data)
for item in cls.hidden_fields:
del new_data.__dict__[item]
return dumper.represent_yaml_object(cls.yaml_tag, new_data, cls,
flow_style=cls.yaml_flow_style)
class Trivial(SecretYamlObject):
hidden_fields = ["_ignore"]
yaml_tag = u'!!Trivial'
def __init__(self):
self.a = 1
self.b = 2
self._ignore = 3
运行这段代码
import yaml
yaml.load('!Trivial {a: 1, b: 2}')
我收到以下错误:
ConstructorError: could not determine a constructor for the tag '!Trivial' in "", line 1, column 1: !Trivial {a: 1, b: 2}
我试图“破解”它,以便 yaml 可以使用 classproperty 找到构造函数使用这样的东西:
class SecretYamlObject(yaml.YAMLObject):
# ... same as before... remove for brevity
class Trivial(SecretYamlObject):
hidden_fields = ["_ignore"]
@classproperty # decorator definition not shown here for brevity
def yaml_tag(cls):
return ('!!')+'python/object:'+'{}.{}'.format(cls.__module__, cls.__name__)
def __init__(self):
self.a = 1
self.b = 2
self._ignore = 3
这会产生错误的 yaml 字符串:
import yaml
from secret_yaml2 import Trivial
print yaml.dump(Trivial())
!%21python/object:secret_yaml2.Trivial {a: 1, b: 2}
出于某种原因,它将第二个 !
转换为 %21
... 这再次导致构造函数错误。
顺便说一句,下面的工作,但我需要在加载 yaml 之前知道对象的类,我可能不知道。
import yaml
from secret_yaml import Trivial
yaml.load(yaml.dump(Trivial()))
我正在尝试创建一个知道如何将自身正确转储到 yaml 的类,但我仍然可以通过正常的 yaml.load 调用加载它。
最佳答案
您可以定义 __getstate__
,它被 Python 原生 pickle 和 PyYaml 使用:
class A(object):
def __init__(self):
self.hidden = 42
self.visible = 5
def __getstate__(self):
state = self.__dict__.copy()
del state['hidden']
return state
a = A()
d = yaml.dump(a)
print(d)
这打印:
!!python/object:__main__.A
visible: 5
这是从 PyYaml 或 Pickle 等序列化程序中忽略或隐藏属性的最佳方式。
关于python - 使用 yaml.dump 时如何忽略属性?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49905287/