Skip to content

Instantly share code, notes, and snippets.

@miracle2k

miracle2k/odict.py

Created July 26, 2012 20:56
Download ZIP

Revisions

  1. miracle2k revised this gist Jul 27, 2012. 2 changed files with 21 additions and 0 deletions.
    8 changes: 8 additions & 0 deletions odict.py
    View file
    Original file line number Diff line number Diff line change
    @@ -1,3 +1,11 @@
    """Make PyYAML output an OrderedDict.
    It will do so fine if you use yaml.dump(), but that generates ugly,
    non-standard YAML code.
    To use yaml.safe_dump(), you need the following.
    """

    def represent_odict(dump, tag, mapping, flow_style=None):
    """Like BaseRepresenter.represent_mapping, but does not issue the sort().
    """
    13 changes: 13 additions & 0 deletions tuple.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,13 @@
    """Make PyYAML load a file that uses lists as mapping keys.
    You would otherwise get an error like:
    yaml.constructor.ConstructorError: while constructing a mapping
    in "eiscp-commands.yaml", line 95, column 7
    found unacceptable key (unhashable type: 'list')
    """

    def construct_tuple(loader, node):
    return tuple(yaml.SafeLoader.construct_sequence(loader, node))
    yaml.SafeLoader.add_constructor(u'tag:yaml.org,2002:seq', construct_tuple)
    yaml.safe_load(file)
  2. miracle2k renamed this gist Jul 26, 2012. 1 changed file with 0 additions and 0 deletions.
    0 gistfile1.txt → odict.py
    View file
    File renamed without changes.
  3. miracle2k created this gist Jul 26, 2012.
    28 changes: 28 additions & 0 deletions gistfile1.txt
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,28 @@
    def represent_odict(dump, tag, mapping, flow_style=None):
    """Like BaseRepresenter.represent_mapping, but does not issue the sort().
    """
    value = []
    node = yaml.MappingNode(tag, value, flow_style=flow_style)
    if dump.alias_key is not None:
    dump.represented_objects[dump.alias_key] = node
    best_style = True
    if hasattr(mapping, 'items'):
    mapping = mapping.items()
    for item_key, item_value in mapping:
    node_key = dump.represent_data(item_key)
    node_value = dump.represent_data(item_value)
    if not (isinstance(node_key, yaml.ScalarNode) and not node_key.style):
    best_style = False
    if not (isinstance(node_value, yaml.ScalarNode) and not node_value.style):
    best_style = False
    value.append((node_key, node_value))
    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

    yaml.SafeDumper.add_representer(OrderedDict,
    lambda dumper, value: represent_odict(dumper, u'tag:yaml.org,2002:map', value))
    yaml.safe_dump(data, default_flow_style=False)