Psycopg2 Connection extension

db.py

import psycopg2.extensions


class Connection(psycopg2.extensions.connection):
    """
    An extension of Psycopg's connection class, to give us some additional
    utilties.

    """

    def query(self, query, args=tuple()):
        """
        Run a query and return the full result set.

        """
        with self.cursor() as c:
            c.execute(query, args)
            return c.fetchall()

    def operation(self, query, args=tuple()):
        """
        Run a query, return nothing.

        """
        with self.cursor() as c:
            c.execute(query, args)