A command line jmespath filter

jmespath

#!/usr/bin/env python3

import sys
import json
import jmespath
import argparse

parser = argparse.ArgumentParser(description='Apply a jmespath filter to stdin')
parser.add_argument('jmespath', metavar='JMESPATH', type=str, nargs=1,
                    help='jmespath expression')
parser.add_argument('--output', dest='output_format', action='store',
                    default='json',
                    help='the output format: text or pretty or json')

args = parser.parse_args()

path = sys.argv[1]
data = json.load(sys.stdin)
ans  = jmespath.search(path, data)
if args.output_format in ['text','txt']:
    if isinstance(ans, list):
        for line in ans:
            print(line)
    else:
        print(ans)
elif args.output_format in ['pretty']:
    print(json.dumps(ans, indent=2, sort_keys=True))
else:
    print(json.dumps(ans))