Add aiopg runner variant returning tuples

Author: elprans

_python/pgbench_python.py

@@ -53,13 +53,21 @@ async def aiopg_connect(args):
                                port=args.pgport)
     return conn
 
-
 async def aiopg_execute(conn, query, args):
     cur = await conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
     await cur.execute(query, args)
     return len(await cur.fetchall())
 
 
+aiopg_tuples_connect = aiopg_connect
+
+
+async def aiopg_tuples_execute(conn, query, args):
+    cur = await conn.cursor()
+    await cur.execute(query, args)
+    return len(await cur.fetchall())
+
+
 async def asyncpg_connect(args):
     conn = await asyncpg.connect(user=args.pguser, host=args.pghost,
                                  port=args.pgport)
@@ -252,7 +260,7 @@ def die(msg):
         help='PostgreSQL server user')
     parser.add_argument(
         'driver', help='driver implementation to use',
-        choices=['aiopg', 'asyncpg', 'psycopg', 'postgresql'])
+        choices=['aiopg', 'aiopg-tuples', 'asyncpg', 'psycopg', 'postgresql'])
     parser.add_argument(
         'queryfile', help='file to read benchmark query information from')
 
@@ -283,6 +291,10 @@ def die(msg):
         connector, executor = aiopg_connect, aiopg_execute
         is_async = True
         arg_format = 'python'
+    if args.driver == 'aiopg-tuples':
+        connector, executor = aiopg_tuples_connect, aiopg_tuples_execute
+        is_async = True
+        arg_format = 'python'
     elif args.driver == 'asyncpg':
         connector, executor = asyncpg_connect, asyncpg_execute
         is_async = True