Add scramble to solves table

This is the scramble that was generated from CsTimer

1 files changed, 9 insertions, 4 deletions

diff --git a/jacsr b/jacsr
index 151d77d..2848c21 100755
--- a/jacsr
+++ b/jacsr
@@ -8,8 +8,13 @@ DATABASE_FILE = 'solves.db'
 
 def create_solve_table(connection):
     cursor = connection.cursor()
+    # TODO: Not sure what to do about DNF (Did Not Finish) solves
     statement = """
-CREATE TABLE solves (number integer primary key autoincrement, time datetime)
+    CREATE TABLE solves (
+        number integer primary key autoincrement,
+        time datetime,
+        scramble text
+    )
     """
     cursor.execute(statement)
     cursor.close()
@@ -22,9 +27,9 @@ def insert_times_into_db(connection, result_file):
         csv_file = csv.DictReader(f, delimiter=';')
 
         for line in csv_file:
-            # TODO: this is dangerous for sql injections.
-            cursor.execute(
-                'INSERT into solves (time) VALUES (' + line['Time'] + ')')
+            # TODO: Add date, comment, etc. fields from CsTimer csv
+            cursor.execute("INSERT into solves (time, scramble) VALUES (?, ?)",
+                           (line['Time'], line['Scramble']))
 
     cursor.close()