reformat.py

#!/usr/bin/env python3

# This script reformat the airport-codes data from
# - https://ourairports.com/data/
# - https://github.com/davidmegginson/ourairports-data

import csv
import json

countries = {}
data = {}

with open('airports.json') as jsonfile:
    # noinspection PyRedeclaration
    data = json.load(jsonfile)

# Download from: https://davidmegginson.github.io/ourairports-data/countries.csv
with open('countries.csv', newline='') as csvfile:
    csvreader = csv.DictReader(csvfile)
    for row in csvreader:
        countries[row['code']] = row

# Download from: https://davidmegginson.github.io/ourairports-data/airports.csv
with open('airports.csv', newline='') as csvfile:
    csvreader = csv.DictReader(csvfile)
    for row in csvreader:
        old_type = ''
        if row['iata_code'] in data:
            airport = data[row['iata_code']]
            if 'type' in airport:
                old_type = airport['type']
        else:
            airport = {}
            if not 'airport' in row['type']:
                continue
        if old_type == '' or old_type == row['type']:
            airport['name'] = row['name']
            airport['city'] = row['municipality']
            if row['iso_country'] in countries:
                airport['country'] = countries[row['iso_country']]['name']
                airport['country_code'] = row['iso_country']
            airport['iata'] = row['iata_code']
            airport['latitude'] = "{:.3f}".format(float(row['latitude_deg']))
            airport['longitude'] = "{:.3f}".format(float(row['longitude_deg']))
            if row['elevation_ft'] != '':
                airport['altitude'] = row['elevation_ft']
            airport['type'] = row['type']
            # airport['extra'] = row
            data[row['iata_code']] = airport

#print(json.dumps(data, indent=4))
print(json.dumps(data))