lpc43xx: Move YAML OrderedDict code to separate module
This commit is contained in:
parent
21e4d4600f
commit
4d6a5d9606
|
@ -4,40 +4,7 @@ import sys
|
|||
import yaml
|
||||
import csv
|
||||
from collections import OrderedDict
|
||||
|
||||
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)
|
||||
|
||||
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
|
||||
import yaml_odict
|
||||
|
||||
def convert_file(fname):
|
||||
reader = csv.reader(open(fname, 'r'))
|
||||
|
|
|
@ -0,0 +1,81 @@
|
|||
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)
|
||||
|
Loading…
Reference in New Issue