Custom db backend for statement_timeout support (postgresql_psycopg2)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
# base.py
from django.db.backends.postgresql_psycopg2.base import *
from django.conf import settings


class DatabaseWrapper(DatabaseWrapper):
    def _cursor(self):
        cursor = super(DatabaseWrapper, self)._cursor()
        statement_timeout = getattr(settings, 'DATABASE_STATEMENT_TIMEOUT', None)
        if statement_timeout is not None:
            cursor.execute("SET STATEMENT_TIMEOUT=%s" % statement_timeout)
        return cursor

More like this

  1. model instance to sql insert statement by gsoltis 1 year, 5 months ago
  2. TwitterBackend by hameedullah 3 years ago
  3. Automatically create urls for templates in a directory by blackrobot 6 months, 3 weeks ago
  4. PowerDNS models by diverman 2 years, 12 months ago
  5. Server Side Cursors for Django's psycopg2 Backend by ryanbutterfield 1 year, 2 months ago

Comments

(Forgotten your password?)