Some more work on db stuff.

bitshift/database/__init__.py

@@ -20,7 +20,8 @@ class Database(object):
 
     def _connect(self):
         """Establish a connection to the database."""
-        default_file = os.path.join(os.path.dirname(__file__), ".my.cnf")
+        root = os.path.dirname(os.path.abspath(__file__))
+        default_file = os.path.join(root, ".my.cnf")
         self._conn = oursql.connect(read_default_file=default_file,
                                     autoping=True, autoreconnect=True)
 
@@ -54,9 +55,18 @@ class Database(object):
         :param codelet: The codelet to insert.
         :type codelet: :py:class:`.Codelet`
         """
-        query = "INSERT INTO codelets VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
+        frag_size = 16384  # 16 kB
+        query_slt1 = """SELECT code_id, LEFT(code_code, {0})
+                        FROM code WHERE code_hash = ?""".format(frag_size)
+        query_ins1 = "INSERT INTO code VALUES (?, ?)"
+        query_ins2 = "INSERT INTO codelets VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
+        query_ins3 = "INSERT INTO authors VALUES", " (?, ?, ?)"
+        query_ins4 = "INSERT INTO symbols VALUES", " (?, ?, ?, ?, ?)"
 
-        cursor.execute(query, ())
+        # LAST_INSERT_ID()
+
+        code_id = None
+        code_hash = mmh3.hash64(codelet.code.encode("utf8"))[0]
 
         # codelet_id -- auto_increment used here
         codelet_name
@@ -78,4 +88,14 @@ class Database(object):
         codelet.date_created
         codelet.date_modified
 
-        code_hash = mmh3.hash64(codelet.code.encode("utf8"))[0]
+        with self._conn.cursor() as cursor:
+            # Retrieve the ID of the source code if it's already in the DB:
+            cursor.execute(query_slt1, (code_hash,))
+            for c_id, c_code_frag in cursor.fetchall():
+                if c_code_frag == codelet.code[:frag_size]:
+                    code_id = c_id
+                    break
+
+            # If the source code isn't already in the DB, add it:
+            if not code_id:
+                cursor.execute()