32 lines
1 KiB
Python
32 lines
1 KiB
Python
|
#!/usr/bin/env python2
|
||
|
|
|
||
|
|
#
|
||
|
|
# Converts a Virtual Radar Server BasicAircraftLookup.sqb database
|
||
|
|
# to a CSV file suitable for feeding to csv-to-json.py
|
||
|
|
#
|
||
|
|
|
||
|
|
import sqlite3, csv, sys
|
||
|
|
from contextlib import closing
|
||
|
|
|
||
|
|
def extract(dbfile):
|
||
|
|
writer = csv.DictWriter(sys.stdout,
|
||
|
|
fieldnames=['icao24', 'r', 't'])
|
||
|
|
writer.writeheader()
|
||
|
|
with closing(sqlite3.connect(dbfile)) as db:
|
||
|
|
with closing(db.execute('SELECT a.Icao, a.Registration, m.Icao FROM Aircraft a, Model m WHERE a.ModelID = m.ModelID')) as c:
|
||
|
|
for icao24, reg, icaotype in c:
|
||
|
|
writer.writerow({
|
||
|
|
'icao24': icao24,
|
||
|
|
'r': reg,
|
||
|
|
't': icaotype
|
||
|
|
})
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
print >>sys.stderr, 'Reads a VRS sqlite database and writes a CSV to stdout'
|
||
|
|
print >>sys.stderr, 'Syntax: %s <path to BasicAircraftLookup.sqb>' % sys.argv[0]
|
||
|
|
sys.exit(1)
|
||
|
|
else:
|
||
|
|
extract(sys.argv[1])
|
||
|
|
sys.exit(0)
|