Improve insert performance using a transaction.

wireservice · Jan 30, 2012 · 9ab787b · 9ab787b
1 parent c663e36
commit 9ab787b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/csvkit/utilities/csvsql.py b/csvkit/utilities/csvsql.py
@@ -56,8 +56,12 @@ def main(self):
                 insert = sql_table.insert()
                 headers = csv_table.headers()
 
+                conn = engine.connect()
+                trans = conn.begin()
                 for row in csv_table.to_rows():
-                    engine.execute(insert, [dict(zip(headers, row)), ])
+                    conn.execute(insert, [dict(zip(headers, row)), ])
+                trans.commit()
+                conn.close()
 
         # Writing to file
         else: