Skip to content
Create a gist now

Instantly share code, notes, and snippets.

Load YAML mappings as ordered dictionaries
import yaml
import yaml.constructor
# included in standard lib from Python 2.7
from collections import OrderedDict
except ImportError:
# try importing the backported drop-in replacement
# it's available on PyPI
from ordereddict import OrderedDict
class OrderedDictYAMLLoader(yaml.Loader):
A YAML loader that loads mappings into ordered dictionaries.
def __init__(self, *args, **kwargs):
yaml.Loader.__init__(self, *args, **kwargs)
self.add_constructor(u',2002:map', type(self).construct_yaml_map)
self.add_constructor(u',2002:omap', type(self).construct_yaml_map)
def construct_yaml_map(self, node):
data = OrderedDict()
yield data
value = self.construct_mapping(node)
def construct_mapping(self, node, deep=False):
if isinstance(node, yaml.MappingNode):
raise yaml.constructor.ConstructorError(None, None,
'expected a mapping node, but found %s' %, node.start_mark)
mapping = OrderedDict()
for key_node, value_node in node.value:
key = self.construct_object(key_node, deep=deep)
except TypeError, exc:
raise yaml.constructor.ConstructorError('while constructing a mapping',
node.start_mark, 'found unacceptable key (%s)' % exc, key_node.start_mark)
value = self.construct_object(value_node, deep=deep)
mapping[key] = value
return mapping
if __name__ == '__main__':
import textwrap
sample = """
two: fish
red: fish
blue: fish
a: yes
b: no
c: null
data = yaml.load(textwrap.dedent(sample), OrderedDictYAMLLoader)
assert type(data) is OrderedDict
print data
alborzs commented Jun 3, 2013

I have a big code base and so far I used vanilla yaml load like this yaml.load(s, Loader=yaml.BaseLoader)
I tried your OrderedDictYAMLLoader but I pretty much get nothing from load now.
what is the exact diffrences, have you tested this in a real application maybe with nested dictionaries of diffrent length?


Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Something went wrong with that request. Please try again.