Fix not fetching cursor on insert/update

PlaidCloud · Feb 6, 2025 · 98737c4 · 98737c4
1 parent e5afdca
commit 98737c4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/databend_sqlalchemy/databend_dialect.py b/databend_sqlalchemy/databend_dialect.py
@@ -1102,6 +1102,11 @@ def create_server_side_cursor(self):
     def create_default_cursor(self):
         return self._dbapi_connection.cursor()
 
+    def post_exec(self):
+        if self.isinsert or self.isupdate or self.isdelete:
+            r = self.cursor.fetchall()
+            self._rowcount = r[0][0]
+
 
 class DatabendTypeCompiler(compiler.GenericTypeCompiler):
     def visit_ARRAY(self, type_, **kw):