codekingpro/portable-devtools

1#!D:\code\apps\devtools\python\python.exe
2 3 4import argparse5import json6import sys7 8import jsonpointer9 10parser = argparse.ArgumentParser(11    description='Resolve a JSON pointer on JSON files')12 13# Accept pointer as argument or as file14ptr_group = parser.add_mutually_exclusive_group(required=True)15 16ptr_group.add_argument('-f', '--pointer-file', type=argparse.FileType('r'),17                       nargs='?',18                       help='File containing a JSON pointer expression')19 20ptr_group.add_argument('POINTER', type=str, nargs='?',21                       help='A JSON pointer expression')22 23parser.add_argument('FILE', type=argparse.FileType('r'), nargs='+',24                    help='Files for which the pointer should be resolved')25parser.add_argument('--indent', type=int, default=None,26                    help='Indent output by n spaces')27parser.add_argument('-v', '--version', action='version',28                    version='%(prog)s ' + jsonpointer.__version__)29 30 31def main():32    try:33        resolve_files()34    except KeyboardInterrupt:35        sys.exit(1)36 37 38def parse_pointer(args):39    if args.POINTER:40        ptr = args.POINTER41    elif args.pointer_file:42        ptr = args.pointer_file.read().strip()43    else:44        parser.print_usage()45        sys.exit(1)46 47    return ptr48 49 50def resolve_files():51    """ Resolve a JSON pointer on JSON files """52    args = parser.parse_args()53 54    ptr = parse_pointer(args)55 56    for f in args.FILE:57        doc = json.load(f)58        try:59            result = jsonpointer.resolve_pointer(doc, ptr)60            print(json.dumps(result, indent=args.indent))61        except jsonpointer.JsonPointerException as e:62            print('Could not resolve pointer: %s' % str(e), file=sys.stderr)63 64 65if __name__ == "__main__":66    main()67