From e3a838220c7394e0985e627a4d7c090ba09e6bb2 Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@gmail.com>
Date: Wed, 30 Apr 2014 14:44:45 -0400
Subject: [PATCH] Flesh out most of Database.insert().

---
 bitshift/database/__init__.py | 44 +++++++++++++++++--------------------------
 1 file changed, 17 insertions(+), 27 deletions(-)

diff --git a/bitshift/database/__init__.py b/bitshift/database/__init__.py
index 14f7575..03a5c2c 100644
--- a/bitshift/database/__init__.py
+++ b/bitshift/database/__init__.py
@@ -87,35 +87,25 @@ class Database(object):
         """
         query1 = """INSERT INTO code VALUES (?, ?)
                     ON DUPLICATE KEY UPDATE code_id=code_id"""
-        query2 = "INSERT INTO codelets VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
-        query3 = "INSERT INTO authors VALUES", " (?, ?, ?)"
-        query4 = "INSERT INTO symbols VALUES", " (?, ?, ?, ?, ?)"
-
-        # LAST_INSERT_ID()
-
-        # codelet_id -- auto_increment used here
-        codelet_name
-        codelet_code_id
-        codelet_lang
-        codelet_origin
-        codelet_url
-        codelet_rank
-        codelet_date_created
-        codelet_date_modified
-
-        # codelet fields
-        codelet.name
-        codelet.code
-        codelet.filename
-        codelet.language
-        codelet.authors
-        codelet.code_url
-        codelet.date_created
-        codelet.date_modified
-
-        #######################################################################
+        query2 = """INSERT INTO codelets VALUES
+                    (?, ?, ?, ?, ?, ?, ?, ?)"""
+        query3 = "SELECT LAST_INSERT_ID()"
+        query4 = "INSERT INTO authors VALUES (?, ?, ?)"
+        query5 = "INSERT INTO symbols VALUES (?, ?, ?, ?, ?, ?, ?)"
 
         code_id = mmh3.hash64(codelet.code.encode("utf8"))[0]
+        origin, url = decompose(codelet.url)  ## TODO: create decompose() function
 
         with self._conn.cursor() as cursor:
             cursor.execute(query1, (code_id, codelet.code))
+            cursor.execute(query2, (codelet.name, code_id, codelet.language,
+                                    origin, url, codelet.rank,
+                                    codelet.date_created,
+                                    codelet.date_modified))
+            cursor.execute(query3)
+            codelet_id = cursor.fetchone()[0]
+            authors = [(codelet_id, a.name, a.url) for a in codelet.authors]  ## TODO: check author fields (is it a tuple?)
+            cursor.executemany(query4, authors)
+            if code_id is new:  ## TODO: check for this properly
+                symbols = [(code_id, sym.type, sym.name, sym.row, sym.col, sym.end_row, sym.end_col) for sym in codelet.symbols]  # TODO: check symbol fields (dict?)
+                cursor.executemany(query5, symbols)