diff options
Diffstat (limited to 'prettyml')
-rwxr-xr-x | prettyml | 13 |
1 files changed, 10 insertions, 3 deletions
@@ -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) |