frontend/afe/readonly_connection.py

from django.db import connection as django_connection
from django.conf import settings
from django.dispatch import dispatcher
from django.core import signals

class ReadOnlyConnection(object):
    """
    This class constructs a new connection to the DB using the read-only
    credentials from settings.  It reaches into some internals of
    django.db.connection which are undocumented as far as I know, but I believe
    it works across many, if not all, of the backends.
    """
    def __init__(self):
        self._connection = None


    def _open_connection(self):
        if self._connection is not None:
            return
        self._save_django_state()
        self._connection = self._get_readonly_connection()
        self._restore_django_state()


    def _save_django_state(self):
        self._old_connection = django_connection.connection
        self._old_host = settings.DATABASE_HOST
        self._old_username = settings.DATABASE_USER
        self._old_password = settings.DATABASE_PASSWORD


    def _restore_django_state(self):
        django_connection.connection = self._old_connection
        settings.DATABASE_HOST = self._old_host
        settings.DATABASE_USER = self._old_username
        settings.DATABASE_PASSWORD = self._old_password


    def _get_readonly_connection(self):
        settings.DATABASE_HOST = settings.DATABASE_READONLY_HOST
        settings.DATABASE_USER = settings.DATABASE_READONLY_USER
        settings.DATABASE_PASSWORD = settings.DATABASE_READONLY_PASSWORD
        django_connection.connection = None
        # cursor() causes a new connection to be created
        cursor = django_connection.cursor()
        assert django_connection.connection is not None
        return django_connection.connection


    def set_django_connection(self):
        assert (django_connection.connection != self._connection or
                self._connection is None)
        self._open_connection()
        self._old_connection = django_connection.connection
        django_connection.connection = self._connection


    def unset_django_connection(self):
        assert self._connection is not None
        assert django_connection.connection == self._connection
        django_connection.connection = self._old_connection


    def cursor(self):
        self._open_connection()
        return self._connection.cursor()


    def close(self):
        if self._connection is not None:
            assert django_connection != self._connection
            self._connection.close()
            self._connection = None


connection = ReadOnlyConnection()
# close any open connection when request finishes
dispatcher.connect(connection.close, signal=signals.request_finished)