capitals_quiz/scripts/generate_countries_list.py
2021-04-11 15:35:16 +01:00

44 lines
1.0 KiB
Python
Executable File

#!/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],
'region': 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)