examples/psycopg2_pool.py: make execute() return 'rowcount'

1d931f1d · Denis Bilenko · 297af289 · 1d931f1d
Commit 1d931f1d authored Feb 09, 2012 by Denis Bilenko
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/psycopg2_pool.py examples/psycopg2_pool.py +1 -0

No files found.
--- a/examples/psycopg2_pool.py
+++ b/examples/psycopg2_pool.py
@@ -109,6 +109,7 @@ class DatabaseConnectionPool(object):
    def execute(self, *args, **kwargs):
        with self.cursor() as cursor:
            cursor.execute(*args, **kwargs)
+            return cursor.rowcount

    def fetchone(self, *args, **kwargs):
        with self.cursor() as cursor: