add every country in the world, apart from those which don't have capitals

This commit is contained in:
2021-04-10 15:21:38 +01:00
parent 2ffcd0d377
commit 6d721910d2
5 changed files with 49 additions and 9 deletions

View File

@@ -0,0 +1,38 @@
#!/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] is None:
continue
r[country['name']['common']] = country['capital'][0]
print('capitals = ', end='')
print(json.dumps(r))
return 0
if __name__ == '__main__':
try:
sys.exit(main(get_args()))
except KeyboardInterrupt:
sys.exit(0)