Create database.py

database.py

@@ -0,0 +1,46 @@
+# database.py
+import sqlite3
+import os
+from parser import parse_python_code
+
+def init_db():
+    conn = sqlite3.connect('python_programs.db')
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS programs
+                 (id INTEGER PRIMARY KEY, code TEXT, sequence TEXT, vectors TEXT)''')
+    conn.commit()
+    conn.close()
+
+def store_program(code, sequence, vectors):
+    conn = sqlite3.connect('python_programs.db')
+    c = conn.cursor()
+    c.execute("INSERT INTO programs (code, sequence, vectors) VALUES (?, ?, ?)",
+              (code, ','.join(sequence), str(vectors)))
+    conn.commit()
+    conn.close()
+
+def populate_sample_db():
+    # Sample programs for testing
+    samples = [
+        """
+        import os
+        def add_one(x):
+            y = x + 1
+            return y
+        """,
+        """
+        def multiply(a, b):
+            c = a * b
+            if c > 0:
+                return c
+        """
+    ]
+    
+    for code in samples:
+        parts, sequence = parse_python_code(code)
+        vectors = [part['vector'] for part in parts]
+        store_program(code, sequence, vectors)
+
+if __name__ == '__main__':
+    init_db()
+    populate_sample_db()