82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
import yaml
|
|
from collections import OrderedDict
|
|
def construct_odict(load, node):
|
|
"""This is the same as SafeConstructor.construct_yaml_omap(),
|
|
except the data type is changed to OrderedDict() and setitem is
|
|
used instead of append in the loop.
|
|
|
|
>>> yaml.load('''
|
|
... !!omap
|
|
... - foo: bar
|
|
... - mumble: quux
|
|
... - baz: gorp
|
|
... ''')
|
|
OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')])
|
|
|
|
>>> yaml.load('''!!omap [ foo: bar, mumble: quux, baz : gorp ]''')
|
|
OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')])
|
|
"""
|
|
|
|
omap = OrderedDict()
|
|
yield omap
|
|
if not isinstance(node, yaml.SequenceNode):
|
|
raise yaml.constructor.ConstructorError(
|
|
"while constructing an ordered map",
|
|
node.start_mark,
|
|
"expected a sequence, but found %s" % node.id, node.start_mark
|
|
)
|
|
for subnode in node.value:
|
|
if not isinstance(subnode, yaml.MappingNode):
|
|
raise yaml.constructor.ConstructorError(
|
|
"while constructing an ordered map", node.start_mark,
|
|
"expected a mapping of length 1, but found %s" % subnode.id,
|
|
subnode.start_mark
|
|
)
|
|
if len(subnode.value) != 1:
|
|
raise yaml.constructor.ConstructorError(
|
|
"while constructing an ordered map", node.start_mark,
|
|
"expected a single mapping item, but found %d items" % len(subnode.value),
|
|
subnode.start_mark
|
|
)
|
|
key_node, value_node = subnode.value[0]
|
|
key = load.construct_object(key_node)
|
|
value = load.construct_object(value_node)
|
|
omap[key] = value
|
|
|
|
yaml.add_constructor(u'tag:yaml.org,2002:omap', construct_odict)
|
|
|
|
def repr_pairs(dump, tag, sequence, flow_style=None):
|
|
"""This is the same code as BaseRepresenter.represent_sequence(),
|
|
but the value passed to dump.represent_data() in the loop is a
|
|
dictionary instead of a tuple."""
|
|
|
|
value = []
|
|
node = yaml.SequenceNode(tag, value, flow_style=flow_style)
|
|
if dump.alias_key is not None:
|
|
dump.represented_objects[dump.alias_key] = node
|
|
best_style = True
|
|
for (key, val) in sequence:
|
|
item = dump.represent_data({key: val})
|
|
if not (isinstance(item, yaml.ScalarNode) and not item.style):
|
|
best_style = False
|
|
value.append(item)
|
|
if flow_style is None:
|
|
if dump.default_flow_style is not None:
|
|
node.flow_style = dump.default_flow_style
|
|
else:
|
|
node.flow_style = best_style
|
|
return node
|
|
|
|
def repr_odict(dumper, data):
|
|
"""
|
|
>>> data = OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')])
|
|
>>> yaml.dump(data, default_flow_style=False)
|
|
'!!omap\\n- foo: bar\\n- mumble: quux\\n- baz: gorp\\n'
|
|
>>> yaml.dump(data, default_flow_style=True)
|
|
'!!omap [foo: bar, mumble: quux, baz: gorp]\\n'
|
|
"""
|
|
return repr_pairs(dumper, u'tag:yaml.org,2002:omap', data.iteritems())
|
|
|
|
yaml.add_representer(OrderedDict, repr_odict)
|
|
|