Filter the keys from first level of a json. Used like "curl yad.da/api/something | ./filter_json.py wanted_key_1 wanted_key_2"

json_filter.py

#!/usr/bin/env python
from __future__ import print_function, division

from operator import itemgetter
import sys

import json


def json_filter(filter_keys, data):
    def dictionary_filter(filter_keys, dictionary):
        filter_values = itemgetter(*filter_keys)
        filtered_values = filter_values(dictionary)

        return dict(
            zip(
                filter_keys,
                filtered_values,
            )
        )

    return json.dumps(
        dictionary_filter(
            filter_keys=filter_keys,
            dictionary=json.loads(data),
        )
    )


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('filter_keys', type=str, nargs="*")
    arguments = parser.parse_args()

    print(
        json_filter(
            filter_keys=arguments.filter_keys,
            data=sys.stdin.read()
        )
    )