summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xprettyml10
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))