| |
|
|
| import sys |
| import json |
| import argparse |
| from pprint import pformat |
|
|
| import jmespath |
| from jmespath import exceptions |
|
|
|
|
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('expression') |
| parser.add_argument('-f', '--filename', |
| help=('The filename containing the input data. ' |
| 'If a filename is not given then data is ' |
| 'read from stdin.')) |
| parser.add_argument('--ast', action='store_true', |
| help=('Pretty print the AST, do not search the data.')) |
| args = parser.parse_args() |
| expression = args.expression |
| if args.ast: |
| |
| expression = jmespath.compile(args.expression) |
| sys.stdout.write(pformat(expression.parsed)) |
| sys.stdout.write('\n') |
| return 0 |
| if args.filename: |
| with open(args.filename, 'r') as f: |
| data = json.load(f) |
| else: |
| data = sys.stdin.read() |
| data = json.loads(data) |
| try: |
| sys.stdout.write(json.dumps( |
| jmespath.search(expression, data), indent=4, ensure_ascii=False)) |
| sys.stdout.write('\n') |
| except exceptions.ArityError as e: |
| sys.stderr.write("invalid-arity: %s\n" % e) |
| return 1 |
| except exceptions.JMESPathTypeError as e: |
| sys.stderr.write("invalid-type: %s\n" % e) |
| return 1 |
| except exceptions.UnknownFunctionError as e: |
| sys.stderr.write("unknown-function: %s\n" % e) |
| return 1 |
| except exceptions.ParseError as e: |
| sys.stderr.write("syntax-error: %s\n" % e) |
| return 1 |
|
|
|
|
| if __name__ == '__main__': |
| sys.exit(main()) |
|
|
