view bin/airport-lookup @ 9572:98c473a71c18

<fizzie> ` mv airport bin/airport-lookup; mkx \'bin/airport//airport-lookup any "$*"\'; mkx \'bin/icao//airport-lookup icao "$*"\'; mkx \'bin/iata//airport-lookup iata "$*"\'
author HackBot
date Sun, 30 Oct 2016 23:43:47 +0000
parents airport@a4b3b4ec7681
children c09e4dd3afd7
line wrap: on
line source

#! /usr/bin/env python

import csv
import sys

if len(sys.argv) < 3: sys.stderr.write('usage: airport any|name|iata|icao key\n'); sys.exit(1)
kind, q = sys.argv[1], ' '.join(sys.argv[2:])

fieldnames = dict(name=1, iata=4, icao=5)
if kind == 'any': fields = [1, 4, 5]
elif kind in fieldnames: fields = [fieldnames[kind]]
else: sys.stderr.write('unknown search type: %s\n' % kind); sys.exit(1)

def f(s): return s if s and s != '\N' else '?'

with open('share/airports.dat', 'rb') as datafile:
  for row in csv.reader(datafile):
    for fi in fields:
      if q.lower() in row[fi].lower():
        print '%s (%s, %s)' % (row[1], f(row[4]), f(row[5]))