Renamed everything in python/ for consistent namespace

python/sql.py

@@ -0,0 +1,132 @@
+#
+# Copyright 2010 Nick Foster
+# 
+# This file is part of gr-air-modes
+# 
+# gr-air-modes 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 3, or (at your option)
+# any later version.
+# 
+# gr-air-modes 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 gr-air-modes; see the file COPYING.  If not, write to
+# the Free Software Foundation, Inc., 51 Franklin Street,
+# Boston, MA 02110-1301, USA.
+# 
+
+import time, os, sys
+from string import split, join
+import air_modes
+import sqlite3
+from air_modes.exceptions import *
+
+class output_sql(air_modes.parse):
+  def __init__(self, mypos, filename):
+    air_modes.parse.__init__(self, mypos)
+    #create the database
+    self.filename = filename
+    self.db = sqlite3.connect(filename)
+    #now execute a schema to create the tables you need
+    c = self.db.cursor()
+    query = """CREATE TABLE IF NOT EXISTS "positions" (
+                "icao" INTEGER KEY NOT NULL,
+                "seen" TEXT NOT NULL,
+                "alt"  INTEGER,
+                "lat"  REAL,
+                "lon"  REAL
+            );"""
+    c.execute(query)
+    query = """CREATE TABLE IF NOT EXISTS "vectors" (
+                "icao"     INTEGER KEY NOT NULL,
+                "seen"     TEXT NOT NULL,
+                "speed"    REAL,
+                "heading"  REAL,
+                "vertical" REAL
+            );"""
+    c.execute(query)
+    query = """CREATE TABLE IF NOT EXISTS "ident" (
+                "icao"     INTEGER PRIMARY KEY NOT NULL,
+                "ident"    TEXT NOT NULL
+            );"""
+    c.execute(query)
+    c.close()
+    #we close the db conn now to reopen it in the output() thread context.
+    self.db.close()
+    self.db = None
+
+  def __del__(self):
+    self.db = None
+
+  def output(self, message):
+    try:
+      #we're checking to see if the db is empty, and creating the db object
+      #if it is. the reason for this is so that the db writing is done within
+      #the thread context of output(), rather than the thread context of the
+      #constructor. that way you can spawn a thread to do output().
+      if self.db is None:
+        self.db = sqlite3.connect(self.filename)
+          
+      query = self.make_insert_query(message)
+      if query is not None:
+        c = self.db.cursor()
+        c.execute(query)
+        c.close()
+        self.db.commit() #don't know if this is necessary
+    except ADSBError:
+      pass
+
+  def make_insert_query(self, message):
+    #assembles a SQL query tailored to our database
+    #this version ignores anything that isn't Type 17 for now, because we just don't care
+    [data, ecc, reference, timestamp] = message.split()
+
+    data = air_modes.modes_reply(long(data, 16))
+    ecc = long(ecc, 16)
+#   reference = float(reference)
+
+
+    query = None
+    msgtype = data["df"]
+    if msgtype == 17:
+      query = self.sql17(data)
+
+    return query
+
+  def sql17(self, data):
+    icao24 = data["aa"]
+    bdsreg = data["me"].get_type()
+
+    retstr = None
+
+    if bdsreg == 0x08:
+      (msg, typename) = self.parseBDS08(data)
+      retstr = "INSERT OR REPLACE INTO ident (icao, ident) VALUES (" + "%i" % icao24 + ", '" + msg + "')"
+
+    elif bdsreg == 0x06:
+      [ground_track, decoded_lat, decoded_lon, rnge, bearing] = self.parseBDS06(data)
+      altitude = 0
+      if decoded_lat is None: #no unambiguously valid position available
+        retstr = None
+      else:
+        retstr = "INSERT INTO positions (icao, seen, alt, lat, lon) VALUES (" + "%i" % icao24 + ", datetime('now'), " + str(altitude) + ", " + "%.6f" % decoded_lat + ", " + "%.6f" % decoded_lon + ")"
+
+    elif bdsreg == 0x05:
+      [altitude, decoded_lat, decoded_lon, rnge, bearing] = self.parseBDS05(data)
+      if decoded_lat is None: #no unambiguously valid position available
+        retstr = None
+      else:
+        retstr = "INSERT INTO positions (icao, seen, alt, lat, lon) VALUES (" + "%i" % icao24 + ", datetime('now'), " + str(altitude) + ", " + "%.6f" % decoded_lat + ", " + "%.6f" % decoded_lon + ")"
+
+    elif bdsreg == 0x09:
+      subtype = data["bds09"].get_type()
+      if subtype == 0 or subtype == 1:
+        parser = self.parseBDS09_0 if subtype == 0 else self.parseBDS09_1
+        [velocity, heading, vert_spd] = parser(data)
+        retstr = "INSERT INTO vectors (icao, seen, speed, heading, vertical) VALUES (" + "%i" % icao24 + ", datetime('now'), " + "%.0f" % velocity + ", " + "%.0f" % heading + ", " + "%.0f" % vert_spd + ")"
+
+    return retstr