Get and Set Statement Timeout to PostgreSQL using django

postgre_statement_timeout.py

class StatementTimeout(object):

    @classmethod
    def _execute_query(cls, query, db_alias):
        from django.db import connections

        cur = connections[db_alias].cursor()
        try:
            cur.execute(query)
            return cls._as_dict(cur) if cur.rowcount > 0 else True
        finally:
            cur.close()

    @classmethod
    def set_timeout(cls, timeout, db_alias):
        """
        :param timeout: in miliseconds
        """
        query = 'set statement_timeout to {}'.format(timeout)
        return cls._execute_query(query, db_alias)

    @classmethod
    def get_timeout(cls, db_alias):
        query = 'show statement_timeout'
        return cls._execute_query(query, db_alias)

    @staticmethod
    def _as_dict(cursor):
        desc = cursor.description
        return [
            dict(zip([col[0] for col in desc], row))
            for row in cursor.fetchall()
        ]