Fix serializing multi-line yaml strings. Fixes #1

bkabrda · Mar 21, 2017 · 200336d · 200336d
1 parent efc4b3f
commit 200336d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/anymarkup_core/__init__.py b/anymarkup_core/__init__.py
@@ -483,6 +483,15 @@ def represent_ordereddict(dumper, data):
         values.append(node_item)
     return node
 
+def represent_str(dumper, data):
+    # borrowed from http://stackoverflow.com/a/33300001
+    if len(data.splitlines()) > 1:
+        return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='|')
+    return dumper.represent_scalar('tag:yaml.org,2002:str', data)
+
 if yaml is not None:
     yaml.SafeLoader.add_constructor(u'tag:yaml.org,2002:omap', construct_ordereddict)
     yaml.SafeDumper.add_representer(collections.OrderedDict, represent_ordereddict)
+    yaml.SafeDumper.add_representer(str, represent_str)
+    if six.PY2:
+        yaml.SafeDumper.add_representer(unicode, represent_str)
diff --git a/test/test_serialize.py b/test/test_serialize.py
@@ -75,3 +75,8 @@ def test_serialize_file_format_overrides_extension(self, tmpdir):
         f = os.path.join(str(tmpdir), 'foo.ini')
         serialize_file(example_as_dict, f, 'json')
         assert parse(self._read_decode(f)) == example_as_dict
+
+    def test_parse_and_serialize_yaml_multiline_string(self):
+        # https://github.com/bkabrda/anymarkup-core/issues/1
+        inp = b'foo: |-\n  line1\n  line2\n  line3\n'
+        assert serialize(parse(inp), 'yaml') == inp