changeset 212:ce44605f888a

minor modifications
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 06 Jun 2012 23:39:03 -0400
parents ada6e8fbe4c6
children 5cde6da74605
files python/storage.py
diffstat 1 files changed, 5 insertions(+), 8 deletions(-) [+]
line wrap: on
line diff
--- a/python/storage.py	Wed Jun 06 13:24:04 2012 -0400
+++ b/python/storage.py	Wed Jun 06 23:39:03 2012 -0400
@@ -25,9 +25,9 @@
     connection = sqlite3.connect(outFile)
     cursor = connection.cursor()
 
-    #creation de la table
-    schema = "CREATE TABLE \"trajectories\"(trajectory_id INTEGER,frame_number INTEGER, x_coordinate REAL, y_coordinate REAL, PRIMARY KEY(trajectory_id, frame_number))"
+    schema = "CREATE TABLE \"positions\"(trajectory_id INTEGER,frame_number INTEGER, x_coordinate REAL, y_coordinate REAL, PRIMARY KEY(trajectory_id, frame_number))"
     cursor.execute(schema)
+
     trajectory_id = 0
     frame_number = 0
     if trajectoryType == 'feature':
@@ -37,16 +37,12 @@
                 frame_number = 0
                 for position in trajectory.getPositions():
                     frame_number += 1
-                    requete = "insert into trajectories(trajectory_id, frame_number, x_coordinate, y_coordinate) values (?,?,?,?)"
-                    cursor.execute(requete,(trajectory_id,frame_number,position.x,position.y))
+                    query = "insert into positions (trajectory_id, frame_number, x_coordinate, y_coordinate) values (?,?,?,?)"
+                    cursor.execute(query,(trajectory_id,frame_number,position.x,position.y))
                     
     connection.commit()            
     connection.close()
 
-def testWrite():
-    features = loadTrajectoriesFromSqlite("/home/francois/Unison/École/12Été/CRSNG/Data/prototypes.sqlite",'feature',-1)
-    writeTrajectoriesToSqlite(features, "/home/francois/Unison/École/12Été/CRSNG/TAAM-Experiments/resultats/testWrite.sqlite", 'feature')
-
 def loadPrototypeMatchIndexesFromSqlite(filename):
     """
     This function loads the prototypes table in the database of name <filename>.
@@ -70,6 +66,7 @@
     return matched_indexes
 
 def testloadPrototypeMatchIndexesFromSqlite():
+    'TODO: write as doctest'
     empty_list = loadPrototypeMatchIndexesFromSqlite("bidon")
     if empty_list == []:
         print "Empty list test Ok"