Mercurial > repos > guerler > springsuite
diff planemo/lib/python3.7/site-packages/pyaml/__main__.py @ 1:56ad4e20f292 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author | guerler |
---|---|
date | Fri, 31 Jul 2020 00:32:28 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/planemo/lib/python3.7/site-packages/pyaml/__main__.py Fri Jul 31 00:32:28 2020 -0400 @@ -0,0 +1,48 @@ +# -*- coding: utf-8 -*- + +import os, sys, stat, tempfile, contextlib, yaml, pyaml + + +@contextlib.contextmanager +def safe_replacement(path, *open_args, **open_kws): + path = str(path) + try: mode = stat.S_IMODE(os.stat(path).st_mode) + except (OSError, IOError): mode = None + open_kws.update( delete=False, + dir=os.path.dirname(path), prefix=os.path.basename(path)+'.' ) + with tempfile.NamedTemporaryFile(*open_args, **open_kws) as tmp: + try: + if mode is not None: os.fchmod(tmp.fileno(), mode) + yield tmp + if not tmp.closed: tmp.flush() + os.rename(tmp.name, path) + finally: + try: os.unlink(tmp.name) + except (OSError, IOError): pass + + +def main(argv=None): + import argparse + parser = argparse.ArgumentParser( + description='Process and dump prettified YAML to stdout.') + parser.add_argument('path', nargs='?', metavar='path', + help='Path to YAML to read (default: use stdin).') + parser.add_argument('-r', '--replace', action='store_true', + help='Replace specified path with prettified version in-place.') + parser.add_argument('-w', '--width', type=int, metavar='chars', + help='Max line width hint to pass to pyyaml for the dump.' + ' Only used to format scalars and collections (e.g. lists).') + opts = parser.parse_args(argv or sys.argv[1:]) + + src = open(opts.path) if opts.path else sys.stdin + try: data = yaml.safe_load(src) + finally: src.close() + + pyaml_kwargs = dict() + if opts.width: pyaml_kwargs['width'] = opts.width + if opts.replace and opts.path: + with safe_replacement(opts.path) as tmp: + pyaml.pprint(data, file=tmp, **pyaml_kwargs) + else: pyaml.pprint(data, **pyaml_kwargs) + +if __name__ == '__main__': sys.exit(main())