python - 将 YAML 反序列化回 Python 对象

Question

我正在尝试向我的类编写正确的 from_yaml 方法，以便在使用 ruamel.yaml 库加载 YAML 文件时能够反序列化回该方法。

假设在我的 to_yaml 类方法中，我返回如下内容:

@classmethod
def to_yaml(cls, dumper, data):
    dict_representation = {
        'foo': data.foo, 
        'bar': data.bar
    }

    return dumper.represent_mapping(cls.yaml_tag, dict_representation)

现在进入反序列化方法

@classmethod
def from_yaml(cls, constructor, node):
    dict_representation = constructor.construct_mapping(node, deep=True)

这样我得到了TypeError:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-6-782b00e19bec> in <module>()
----> 1 dict_representation = yaml.constructor.construct_mapping(node, deep=True)

/home/**/.envs/myenv/local/lib/python2.7/site-packages/ruamel/yaml/constructor.pyc in construct_mapping(self, node, maptyp, deep)
   1186                         "found unhashable key", key_node.start_mark)
   1187             value = self.construct_object(value_node, deep=deep)
-> 1188             self.check_mapping_key(node, key_node, maptyp, key, value)
   1189
   1190             if key_node.comment and len(key_node.comment) > 4 and \

/home/**/.envs/myenv/local/lib/python2.7/site-packages/ruamel/yaml/constructor.pyc in check_mapping_key(self, node, key_node, mapping, key, value)
    241     def check_mapping_key(self, node, key_node, mapping, key, value):
    242         # type: (Any, Any, Any, Any, Any) -> None
--> 243         if key in mapping:
    244             if not self.allow_duplicate_keys:
    245                 args = [

TypeError: argument of type 'NoneType' is not iterable

事实上，尝试在交互式 shell 中更凭经验地做到这一点:

import ruamel.yaml
yaml = ruamel.yaml.YAML()
dd = {'foo': 'foo'}
node = yaml.representer.represent_mapping('!dd', dd)
dict_representation = yaml.constructor.construct_mapping(node)

引发相同的异常。我在这里缺少什么？

Answer 1

为了使往返工作，RoundTripConstructor() 的 construct_mapping() 需要获取传入的实际映射类型实例，所以事情类似的注释可以从节点获取并附加到该实例(通常是 CommentedMap())。进行非往返加载时不需要该额外参数(因为它们不需要传递评论信息)。

该方法本来可以设计得更智能，目前如果未提供，则默认为 None 作为映射类型，这就是您获得 NoneType is not iterable 异常的地方。

要从问题末尾的代码开始，您可以通过执行以下操作来调用更简单的映射构造函数:

dict_representation = ruamel.yaml.constructor.SafeConstructor.construct_mapping(
    yaml.constructor, node)

您的 from_yaml() 类方法应该以相同的方式工作:

@classmethod
def from_yaml(cls, constructor, node):
    dict_representation = ruamel.yaml.constructor.SafeConstructor.construct_mapping(
        constructor, node, deep=True)

尽管如果您正在构造复杂类型(其中某些间接可访问的值可能引用此节点)，则应考虑使用两步创建过程:

@classmethod
def from_yaml(cls, constructor, node):
    dict_representation = dict()
    yield dict_representation
    d = ruamel.yaml.constructor.SafeConstructor.construct_mapping(
         constructor, node, deep=True)
    dict_representation.update(d)