Connection utilities for aiopg

utils.py

from aiopg import Connection as BaseConnection


class Connection(BaseConnection):
    async def query(self, query, args=[]):
        cur = await self.cursor()
        try:
            await cur.execute(query, args)
            return await cur.fetchall()
        finally:
            cur.close()

    async def operation(self, query, args=[]):
        cur = await self.cursor()
        try:
            await cur.execute(query, args)
            assert cur.rowcount == -1, \
                'Operation returned results.'
        finally:
            cur.close()