Make pretty* read from stdin by default

1 files changed, 10 insertions, 3 deletions

diff --git a/prettyml b/prettyml
index fe9f468..f4fbb5e 100755
--- a/prettyml
+++ b/prettyml
@@ -4,8 +4,15 @@ import sys
 
 import yaml
 
+def inputs(args):
+    if args:
+        for filename in args:
+            with open(filename) as f:
+                yield filename, f
+    else:
+        yield '-', sys.stdin
+
 objs = []
-for filename in sys.argv[1:]:
-    with open(filename) as f:
-        objs.append(yaml.safe_load(f))
+for filename, f in inputs(sys.argv[1:]):
+    objs.append(yaml.safe_load(f))
 yaml.safe_dump_all(objs, stream=sys.stdout, indent=4, default_flow_style=False)