+
−"""
+
−PostgreSQL database backend for Django.
+
−
+
−Requires psycopg 2: http://initd.org/projects/psycopg2
+
−"""
+
−
+
−from django.db.backends import *
+
−from django.db.backends.postgresql.operations import DatabaseOperations as PostgresqlDatabaseOperations
+
−from django.db.backends.postgresql.client import DatabaseClient
+
−from django.db.backends.postgresql.creation import DatabaseCreation
+
−from django.db.backends.postgresql.version import get_version
+
−from django.db.backends.postgresql_psycopg2.introspection import DatabaseIntrospection
+
−from django.utils.safestring import SafeUnicode, SafeString
+
−
+
−try:
+
−    import psycopg2 as Database
+
−    import psycopg2.extensions
+
−except ImportError, e:
+
−    from django.core.exceptions import ImproperlyConfigured
+
−    raise ImproperlyConfigured("Error loading psycopg2 module: %s" % e)
+
−
+
−DatabaseError = Database.DatabaseError
+
−IntegrityError = Database.IntegrityError
+
−
+
−psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
+
−psycopg2.extensions.register_adapter(SafeString, psycopg2.extensions.QuotedString)
+
−psycopg2.extensions.register_adapter(SafeUnicode, psycopg2.extensions.QuotedString)
+
−
+
−class DatabaseFeatures(BaseDatabaseFeatures):
+
−    needs_datetime_string_cast = False
+
−    uses_savepoints = True
+
−
+
−class DatabaseOperations(PostgresqlDatabaseOperations):
+
−    def last_executed_query(self, cursor, sql, params):
+
−        # With psycopg2, cursor objects have a "query" attribute that is the
+
−        # exact query sent to the database. See docs here:
+
−        # http://www.initd.org/tracker/psycopg/wiki/psycopg2_documentation#postgresql-status-message-and-executed-query
+
−        return cursor.query
+
−
+
−class DatabaseWrapper(BaseDatabaseWrapper):
+
−    operators = {
+
−        'exact': '= %s',
+
−        'iexact': '= UPPER(%s)',
+
−        'contains': 'LIKE %s',
+
−        'icontains': 'LIKE UPPER(%s)',
+
−        'regex': '~ %s',
+
−        'iregex': '~* %s',
+
−        'gt': '> %s',
+
−        'gte': '>= %s',
+
−        'lt': '< %s',
+
−        'lte': '<= %s',
+
−        'startswith': 'LIKE %s',
+
−        'endswith': 'LIKE %s',
+
−        'istartswith': 'LIKE UPPER(%s)',
+
−        'iendswith': 'LIKE UPPER(%s)',
+
−    }
+
−
+
−    def __init__(self, *args, **kwargs):
+
−        super(DatabaseWrapper, self).__init__(*args, **kwargs)
+
−        
+
−        self.features = DatabaseFeatures()
+
−        self.ops = DatabaseOperations()
+
−        self.client = DatabaseClient()
+
−        self.creation = DatabaseCreation(self)
+
−        self.introspection = DatabaseIntrospection(self)
+
−        self.validation = BaseDatabaseValidation()
+
−
+
−    def _cursor(self, settings):
+
−        set_tz = False
+
−        if self.connection is None:
+
−            set_tz = True
+
−            if settings.DATABASE_NAME == '':
+
−                from django.core.exceptions import ImproperlyConfigured
+
−                raise ImproperlyConfigured("You need to specify DATABASE_NAME in your Django settings file.")
+
−            conn_string = "dbname=%s" % settings.DATABASE_NAME
+
−            if settings.DATABASE_USER:
+
−                conn_string = "user=%s %s" % (settings.DATABASE_USER, conn_string)
+
−            if settings.DATABASE_PASSWORD:
+
−                conn_string += " password='%s'" % settings.DATABASE_PASSWORD
+
−            if settings.DATABASE_HOST:
+
−                conn_string += " host=%s" % settings.DATABASE_HOST
+
−            if settings.DATABASE_PORT:
+
−                conn_string += " port=%s" % settings.DATABASE_PORT
+
−            self.connection = Database.connect(conn_string, **self.options)
+
−            self.connection.set_isolation_level(1) # make transactions transparent to all cursors
+
−            self.connection.set_client_encoding('UTF8')
+
−        cursor = self.connection.cursor()
+
−        cursor.tzinfo_factory = None
+
−        if set_tz:
+
−            cursor.execute("SET TIME ZONE %s", [settings.TIME_ZONE])
+
−            if not hasattr(self, '_version'):
+
−                self.__class__._version = get_version(cursor)
+
−            if self._version < (8, 0):
+
−                # No savepoint support for earlier version of PostgreSQL.
+
−                self.features.uses_savepoints = False
+
−        return cursor