Merge pull request #150 from gnat/execute-return-rows-affected-mysql-sqlite

Fix: execute() should return rows affected by operation if lastrowid is not set.

Author: gvbgduh

databases/backends/mysql.py

@@ -131,6 +131,8 @@ async def execute(self, query: ClauseElement) -> typing.Any:
         cursor = await self._connection.cursor()
         try:
             await cursor.execute(query, args)
+            if cursor.lastrowid == 0:
+                return cursor.rowcount
             return cursor.lastrowid
         finally:
             await cursor.close()

databases/backends/sqlite.py

@@ -111,9 +111,14 @@ async def fetch_one(self, query: ClauseElement) -> typing.Optional[typing.Mappin
     async def execute(self, query: ClauseElement) -> typing.Any:
         assert self._connection is not None, "Connection is not acquired"
         query, args, context = self._compile(query)
-        cursor = await self._connection.execute(query, args)
-        await cursor.close()
-        return cursor.lastrowid
+        cursor = await self._connection.cursor()
+        try:
+            await cursor.execute(query, args)
+            if cursor.lastrowid == 0:
+                return cursor.rowcount
+            return cursor.lastrowid
+        finally:
+            await cursor.close()
 
     async def execute_many(self, queries: typing.List[ClauseElement]) -> None:
         assert self._connection is not None, "Connection is not acquired"