#!/usr/bin/env python3

# Copyright (C) 2021 ArjanOnwezen
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#

# -------------------------------------------------------------------------------------
# Create airline.db, used for ADS-B receiver application, using
# https://raw.githubusercontent.com/kx1t/planefence-airlinecodes/main/airlinecodes.txt
# as a source.
# -------------------------------------------------------------------------------------
import csv
import unicodedata
icao_codes=bytearray()
airlines_countries=bytearray()
row_count=0

database=open("airlines.db", "wb")

with open('airlinecodes.txt', 'rt') as csv_file:
        sorted_lines=sorted(csv_file.readlines())

        for row in csv.reader(sorted_lines, quotechar='"', delimiter=',', quoting=csv.QUOTE_ALL, skipinitialspace=True):
                icao_code=row[0]
                # Normalize some unicode characters
                airline=unicodedata.normalize('NFKD', row[1][:32]).encode('ascii', 'ignore')
                country=unicodedata.normalize('NFKD', row[3][:32]).encode('ascii', 'ignore')
                if len(icao_code) == 3 :
                        airline_padding=bytearray()
                        country_padding=bytearray()
                        print(icao_code,' - ', airline,' - ', country)
                        icao_codes=icao_codes+bytearray(icao_code+'\0', encoding='ascii')
                        airline_padding=bytearray('\0' * (32 - len(airline)), encoding='ascii')
                        country_padding=bytearray('\0' * (32 - len(country)), encoding='ascii')
                        airlines_countries=airlines_countries+bytearray(airline+airline_padding+country+country_padding)
                        row_count+=1

database.write(icao_codes+airlines_countries)
print("Total of", row_count, "ICAO codes stored in database")