From fd910f859e0adbcbc7b0f9fa5ec314f5ddd305ae Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Mon, 11 Jul 2016 12:18:26 +0300 Subject: Make pretty* read from stdin by default --- prettyml | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'prettyml') 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) -- cgit v1.2.1