Connect to db and insert times from csv

Need to come back and handler for SQL injections.
Currently will fail if run with an existing db file

1 files changed, 29 insertions, 4 deletions

diff --git a/jacsr b/jacsr
index 10ef5c3..82ed1ac 100755
--- a/jacsr
+++ b/jacsr
@@ -1,14 +1,39 @@
 #!/usr/bin/python3
 import csv
+import sqlite3
 
 RESULT_FILE = './cstimer.csv'
+DATABASE_FILE = 'solves.db'
 
-def main():
-    with open(RESULT_FILE) as f:
+
+def create_solve_table(connection):
+    cursor = connection.cursor()
+    statement = """
+CREATE TABLE solves (number integer primary key autoincrement, time datetime)
+    """
+    cursor.execute(statement)
+    cursor.close()
+
+
+def insert_times_into_db(connection, result_file):
+    cursor = connection.cursor()
+
+    with open(result_file) as f:
         csv_file = csv.DictReader(f, delimiter=';')
 
         for line in csv_file:
-            print(line)
+            # TODO: this is dangerous for sql injections.
+            cursor.execute(
+                'INSERT into solves (time) VALUES (' + line['Time'] + ')')
+
+    cursor.close()
 
 if __name__ == "__main__":
-    main()
+    connection = sqlite3.connect(DATABASE_FILE)
+
+    create_solve_table(connection)
+
+    insert_times_into_db(connection, RESULT_FILE)
+
+    connection.commit()
+    connection.close()