def loadSpeedPrototypeFromSqlite(filename):
"""
This function loads the prototypes table in the database of name <filename>.
"""
prototypes = {}
nMatching={}
connection = sqlite3.connect(filename)
cursor = connection.cursor()
try:
cursor.execute('SELECT * from speedprototypes order by spdprototype_id,prototype_id, routeID_start, routeID_end, nMatching')
except sqlite3.OperationalError as error:
utils.printDBError(error)
return []
for row in cursor:
route=(row[2],row[3])
if route not in prototypes.keys():
prototypes[route]={}
if row[1] not in prototypes[route].keys():
prototypes[route][row[1]]=[]
prototypes[route][row[1]].append(row[0])
nMatching[row[0]]=row[4]
connection.close()
return prototypes,nMatching
def loadPrototypesFromSqlite(filename):
"""
This function loads the prototype file in the database
It returns a dictionary for prototypes for each route and nMatching
"""
prototypes = {}
nMatching={}
connection = sqlite3.connect(filename)
cursor = connection.cursor()
try:
cursor.execute('SELECT * from prototypes order by prototype_id, routeIDstart,routeIDend, nMatching')
except sqlite3.OperationalError as error:
utils.printDBError(error)
return []
for row in cursor:
route=(row[1],row[2])
if route not in prototypes.keys():
prototypes[route]=[]
prototypes[route].append(row[0])
nMatching[row[0]]=row[3]
connection.close()
return prototypes,nMatching
def loadRoutesFromSqlite(filename):
Routes = {}
connection = sqlite3.connect(filename)
cursor = connection.cursor()
try:
cursor.execute('SELECT * from routes order by object_id, routeIDstart,routeIDend')
except sqlite3.OperationalError as error:
utils.printDBError(error)
return []
for row in cursor:
route=(row[1],row[2])
if route not in Routes.keys():
Routes[route]=[]
Routes[route].append(row[0])
connection.close()
return Routes
def loadLabelsFromSqlite(filename):
labels = {}
connection = sqlite3.connect(filename)
cursor = connection.cursor()
try:
cursor.execute('SELECT * from labels order by object_id, routeIDstart,routeIDend, prototype_id')
except sqlite3.OperationalError as error:
utils.printDBError(error)
return []
for row in cursor:
route=(row[1],row[2])
p=row[3]
if route not in labels.keys():
labels[route]={}
if p not in labels[route].keys():
labels[route][p]=[]
labels[route][p].append(row[0])
connection.close()
return labels
