sqlite3: Reliably detect data query vs modification SQL statements.

pfalcon · pfalcon · commit b2aa75938442 · 2014-06-17T00:50:08.000+03:00
diff --git a/sqlite3/sqlite3.py b/sqlite3/sqlite3.py
@@ -89,8 +89,10 @@ def execute(self, sql, params=None):
         self.num_cols = sqlite3_column_count(self.stmnt)
         print("num_cols", self.num_cols)
         # If it's not select, actually execute it here
-        if not sql.startswith("SELECT"):
-            self.fetchone()
+        # num_cols == 0 for statements which don't return data (=> modify it)
+        if not self.num_cols:
+            v = self.fetchone()
+            assert v is None
 
     def close(self):
         s = sqlite3_finalize(self.stmnt)
