capitals_quiz/scripts/generate_countries_list.py

#!/usr/bin/env python3

import sys
import json


def get_args():
    """ Get command line arguments """

    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('file', help="The countries.json file to generate the output from", type=str)
    return parser.parse_args()


def main(args):
    """ Entry point for script """
    with open(args.file) as fp:
        countries = json.load(fp)

    r = {}

    for country in countries:
        if len(country['capital']) < 1 or country['capital'][0] == "" or not country['independent']:
            continue
        r[country['name']['common']] = {
                'capital': country['capital'][0],
                'regions': [country['region'],
                'subregion': country['subregion'],
                'languages': country['languages']
                }

    print('countries = ', end='')
    print(json.dumps(r))

    return 0


if __name__ == '__main__':
    try:
        sys.exit(main(get_args()))
    except KeyboardInterrupt:
        sys.exit(0)