Fix: represent long string with the | style

1 files changed, 10 insertions, 0 deletions

diff --git a/prettyml b/prettyml
index 12f1a8e..e15b3b2 100755
--- a/prettyml
+++ b/prettyml
@@ -13,6 +13,16 @@ def inputs(args):
         yield '-', sys.stdin
 
 
+def string_representer(dumper, data):
+    style = None
+    if '\n' in data:
+        style = '|'
+    return dumper.represent_scalar('tag:yaml.org,2002:str', data, style=style)
+
+
+yaml.add_representer(str, string_representer, Dumper=yaml.SafeDumper)
+
+
 objs = []
 for filename, f in inputs(sys.argv[1:]):
     objs.append(yaml.load(f, Loader=yaml.CSafeLoader))