Split out the data model because I'm using a flat SQL table for the dashboard now. Most everything hooked up.
This commit is contained in:
71
python/modes_datamodel.py
Normal file
71
python/modes_datamodel.py
Normal file
@@ -0,0 +1,71 @@
|
||||
|
||||
#so, not to be a dick or anything, but this thing could be entirely
|
||||
#replaced with a clever SQL join or just a separate SQL table flattened
|
||||
#in modes_sql and designed specifically for this. you can keep the older
|
||||
#ones, too. this would let you access indices by column name, too. bonus.
|
||||
#you wouldn't get the nice hierarchical tree view, though. what does that lose you?
|
||||
#you'd need a different datamodel for the tree view, that's all.
|
||||
class modes_datamodel(QtCore.QAbstractItemModel):
|
||||
def __init__(self, db):
|
||||
QtCore.QAbstractItemModel.__init__(self)
|
||||
self.db = db
|
||||
|
||||
def rowCount(self, parent=QtCore.QModelIndex()):
|
||||
icaoquery = "select count(distinct icao) from positions"
|
||||
query = QtSql.QSqlQuery()
|
||||
query.exec_(icaoquery)
|
||||
query.next()
|
||||
return query.value(0).toInt()[0]
|
||||
|
||||
def columnCount(self, parent=QtCore.QModelIndex()):
|
||||
return 5
|
||||
|
||||
#for future use
|
||||
def flags(self, index):
|
||||
return QtCore.QAbstractItemModel.flags(self, index)
|
||||
|
||||
def data(self, index, role=QtCore.Qt.DisplayRole):
|
||||
if not index.isValid():
|
||||
return QtCore.QVariant()
|
||||
if index.row() >= self.rowCount():
|
||||
return QtCore.QVariant()
|
||||
if role == QtCore.Qt.DisplayRole or role == QtCore.Qt.EditRole:
|
||||
#get ICAO of current index
|
||||
query = QtSql.QSqlQuery()
|
||||
#TODO: limit this to aircraft seen in last 600 seconds? or make a spinbox for this limit.
|
||||
icaoquery = "select distinct icao from positions order by icao limit %i,1" % index.row()
|
||||
query.exec_(icaoquery)
|
||||
query.next()
|
||||
icao = query.value(0).toInt()[0]
|
||||
#TODO figure out how to grab multiple records in one query and return them all?
|
||||
if index.column() == 0: #ICAO
|
||||
return "%06x" % icao
|
||||
elif index.column() == 1: #last seen
|
||||
seenquery = "select seen from positions where icao = %i order by seen desc limit 1" % icao
|
||||
query.exec_(seenquery)
|
||||
return "" if query.next() is False else query.value(0).toString()
|
||||
elif index.column() == 2: #ident
|
||||
identquery = "select ident from ident where icao = %i" % icao
|
||||
query.exec_(identquery)
|
||||
return "" if query.next() is False else query.value(0).toString()
|
||||
elif index.column() == 3: #altitude
|
||||
querystr = "select alt from positions where icao = %i order by seen desc limit 1" % icao
|
||||
query.exec_(querystr)
|
||||
return "" if query.next() is False else "%i" % query.value(0).toInt()[0]
|
||||
elif index.column() == 4: #latitude
|
||||
querystr = "select lat from positions where icao = %i order by seen desc limit 1" % icao
|
||||
query.exec_(querystr)
|
||||
return "" if query.next() is False else "%.6f" % query.value(0).toFloat()[0]
|
||||
elif index.column() == 5: #longitude
|
||||
querystr = "select lon from positions where icao = %i order by seen desc limit 1" % icao
|
||||
query.exec_(querystr)
|
||||
return "" if query.next() is False else "%.6f" % query.value(0).toFloat()[0]
|
||||
|
||||
else:
|
||||
return QtCore.QVariant()
|
||||
|
||||
def index(self, row, column, parent=QtCore.QModelIndex()):
|
||||
return self.createIndex(row, column)
|
||||
|
||||
def parent(self, child):
|
||||
return QtCore.QModelIndex()
|
||||
Reference in New Issue
Block a user