From 24154b216682dfef7ff647a5c2f698b89ba429a6 Mon Sep 17 00:00:00 2001 From: Adrian Holovaty Date: Sat, 15 Oct 2005 01:37:16 +0000 Subject: [PATCH] Fixed #225 -- Added first stab at MS SQL Server support (via ADO). Thanks to gmilas@gmail.com for the patch git-svn-id: http://code.djangoproject.com/svn/django/trunk@879 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- django/conf/global_settings.py | 12 +- django/conf/project_template/settings/main.py | 2 +- django/core/db/backends/ado_mssql.py | 153 ++++++++++++++++++ 3 files changed, 160 insertions(+), 7 deletions(-) create mode 100644 django/core/db/backends/ado_mssql.py diff --git a/django/conf/global_settings.py b/django/conf/global_settings.py index dde5612038..41fe147374 100644 --- a/django/conf/global_settings.py +++ b/django/conf/global_settings.py @@ -45,12 +45,12 @@ SERVER_EMAIL = 'root@localhost' SEND_BROKEN_LINK_EMAILS = False # Database connection info. -DATABASE_ENGINE = 'postgresql' # 'postgresql', 'mysql', or 'sqlite3'. -DATABASE_NAME = '' -DATABASE_USER = '' -DATABASE_PASSWORD = '' -DATABASE_HOST = '' # Set to empty string for localhost. -DATABASE_PORT = '' # Set to empty string for default. +DATABASE_ENGINE = 'postgresql' # 'postgresql', 'mysql', 'sqlite3' or 'ado_mssql'. +DATABASE_NAME = '' # Or path to database file if using sqlite3. +DATABASE_USER = '' # Not used with sqlite3. +DATABASE_PASSWORD = '' # Not used with sqlite3. +DATABASE_HOST = '' # Set to empty string for localhost. Not used with sqlite3. +DATABASE_PORT = '' # Set to empty string for default. Not used with sqlite3. # Host for sending e-mail. EMAIL_HOST = 'localhost' diff --git a/django/conf/project_template/settings/main.py b/django/conf/project_template/settings/main.py index 38df2ad01d..cbb32b7920 100644 --- a/django/conf/project_template/settings/main.py +++ b/django/conf/project_template/settings/main.py @@ -10,7 +10,7 @@ MANAGERS = ADMINS LANGUAGE_CODE = 'en-us' -DATABASE_ENGINE = 'postgresql' # 'postgresql', 'mysql', or 'sqlite3'. +DATABASE_ENGINE = 'postgresql' # 'postgresql', 'mysql', 'sqlite3' or 'ado_mssql'. DATABASE_NAME = '' # Or path to database file if using sqlite3. DATABASE_USER = '' # Not used with sqlite3. DATABASE_PASSWORD = '' # Not used with sqlite3. diff --git a/django/core/db/backends/ado_mssql.py b/django/core/db/backends/ado_mssql.py new file mode 100644 index 0000000000..46116ccdad --- /dev/null +++ b/django/core/db/backends/ado_mssql.py @@ -0,0 +1,153 @@ +""" +ADO MSSQL database backend for Django. + +Requires adodbapi 2.0.1: http://adodbapi.sourceforge.net/ +""" + +from django.core.db import base +from django.core.db.dicthelpers import * +import adodbapi as Database +import datetime +try: + import mx +except ImportError: + mx = None + +DatabaseError = Database.DatabaseError + +# We need to use a special Cursor class because adodbapi expects question-mark +# param style, but Django expects "%s". This cursor converts question marks to +# format-string style. +class Cursor(Database.Cursor): + def executeHelper(self, operation, isStoredProcedureCall, parameters=None): + if parameters is not None and "%s" in operation: + operation = operation.replace("%s", "?") + Database.Cursor.executeHelper(self, operation, isStoredProcedureCall, parameters) + +class Connection(Database.Connection): + def cursor(self): + return Cursor(self) +Database.Connection = Connection + +origCVtoP = Database.convertVariantToPython +def variantToPython(variant, adType): + if type(variant) == bool and adType == 11: + return variant # bool not 1/0 + res = origCVtoP(variant, adType) + if mx is not None and type(res) == mx.DateTime.mxDateTime.DateTimeType: + # Convert ms.DateTime objects to Python datetime.datetime objects. + tv = list(res.tuple()[:7]) + tv[-2] = int(tv[-2]) + return datetime.datetime(*tuple(tv)) + if type(res) == float and str(res)[-2:] == ".0": + return int(res) # If float but int, then int. + return res +Database.convertVariantToPython = variantToPython + +class DatabaseWrapper: + def __init__(self): + self.connection = None + self.queries = [] + + def cursor(self): + from django.conf.settings import DATABASE_USER, DATABASE_NAME, DATABASE_HOST, DATABASE_PORT, DATABASE_PASSWORD, DEBUG + if self.connection is None: + if DATABASE_NAME == '' or DATABASE_USER == '': + from django.core.exceptions import ImproperlyConfigured + raise ImproperlyConfigured, "You need to specify both DATABASE_NAME and DATABASE_USER in your Django settings file." + if not DATABASE_HOST: + DATABASE_HOST = "127.0.0.1" + # TODO: Handle DATABASE_PORT. + conn_string = "PROVIDER=SQLOLEDB;DATA SOURCE=%s;UID=%s;PWD=%s;DATABASE=%s" % (DATABASE_HOST, DATABASE_USER, DATABASE_PASSWORD, DATABASE_NAME) + self.connection = Database.connect(conn_string) + cursor = self.connection.cursor() + if DEBUG: + return base.CursorDebugWrapper(cursor, self) + return cursor + + def commit(self): + return self.connection.commit() + + def rollback(self): + if self.connection: + return self.connection.rollback() + + def close(self): + if self.connection is not None: + self.connection.close() + self.connection = None + +def get_last_insert_id(cursor, table_name, pk_name): + cursor.execute("SELECT %s FROM %s WHERE %s = @@IDENTITY" % (pk_name, table_name, pk_name)) + return cursor.fetchone()[0] + +def get_date_extract_sql(lookup_type, table_name): + # lookup_type is 'year', 'month', 'day' + return "DATEPART(%s, %s)" % (lookup_type, table_name) + +def get_date_trunc_sql(lookup_type, field_name): + # lookup_type is 'year', 'month', 'day' + if lookup_type=='year': + return "Convert(datetime, Convert(varchar, DATEPART(year, %s)) + '/01/01')" % field_name + if lookup_type=='month': + return "Convert(datetime, Convert(varchar, DATEPART(year, %s)) + '/' + Convert(varchar, DATEPART(month, %s)) + '/01')" % (field_name, field_name) + if lookup_type=='day': + return "Convert(datetime, Convert(varchar(12), %s))" % field_name + +def get_limit_offset_sql(limit, offset=None): + # TODO: This is a guess. Make sure this is correct. + sql = "LIMIT %s" % limit + if offset and offset != 0: + sql += " OFFSET %s" % offset + return sql + +def get_random_function_sql(): + # TODO: This is a guess. Make sure this is correct. + return "RANDOM()" + +def get_relations(cursor, table_name): + raise NotImplementedError + +OPERATOR_MAPPING = { + 'exact': '=', + 'iexact': 'LIKE', + 'contains': 'LIKE', + 'icontains': 'LIKE', + 'ne': '!=', + 'gt': '>', + 'gte': '>=', + 'lt': '<', + 'lte': '<=', + 'startswith': 'LIKE', + 'endswith': 'LIKE', + 'istartswith': 'LIKE', + 'iendswith': 'LIKE', +} + +DATA_TYPES = { + 'AutoField': 'int IDENTITY (1, 1)', + 'BooleanField': 'bit', + 'CharField': 'varchar(%(maxlength)s)', + 'CommaSeparatedIntegerField': 'varchar(%(maxlength)s)', + 'DateField': 'smalldatetime', + 'DateTimeField': 'smalldatetime', + 'EmailField': 'varchar(75)', + 'FileField': 'varchar(100)', + 'FilePathField': 'varchar(100)', + 'FloatField': 'numeric(%(max_digits)s, %(decimal_places)s)', + 'ImageField': 'varchar(100)', + 'IntegerField': 'int', + 'IPAddressField': 'char(15)', + 'ManyToManyField': None, + 'NullBooleanField': 'bit', + 'OneToOneField': 'int', + 'PhoneNumberField': 'varchar(20)', + 'PositiveIntegerField': 'int CONSTRAINT [CK_int_pos_%(name)s] CHECK ([%(name)s] > 0)', + 'PositiveSmallIntegerField': 'smallint CONSTRAINT [CK_smallint_pos_%(name)s] CHECK ([%(name)s] > 0)', + 'SlugField': 'varchar(50)', + 'SmallIntegerField': 'smallint', + 'TextField': 'text', + 'TimeField': 'time', + 'URLField': 'varchar(200)', + 'USStateField': 'varchar(2)', +}