From 81fdb659f35a533c29fc163ac679d0c6068385d9 Mon Sep 17 00:00:00 2001 From: Anthon van der Neut Date: Thu, 16 Aug 2018 09:49:03 +0200 Subject: allow simple mappings as mapping keys --- representer.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'representer.py') diff --git a/representer.py b/representer.py index c663397..9b97bdd 100644 --- a/representer.py +++ b/representer.py @@ -637,6 +637,7 @@ from ruamel.yaml.comments import ( CommentedOrderedMap, CommentedSeq, CommentedKeySeq, + CommentedKeyMap, CommentedSet, comment_attrib, merge_attrib, @@ -886,6 +887,9 @@ class RoundTripRepresenter(SafeRepresenter): if isinstance(data, CommentedKeySeq): self.alias_key = None return self.represent_sequence(u'tag:yaml.org,2002:seq', data, flow_style=True) + if isinstance(data, CommentedKeyMap): + self.alias_key = None + return self.represent_mapping(u'tag:yaml.org,2002:map', data, flow_style=True) return SafeRepresenter.represent_key(self, data) def represent_mapping(self, tag, mapping, flow_style=None): -- cgit v1.2.1