add basic support for Windows Azure SQL Database

Author: michiya

django_pyodbc/base.py

@@ -44,6 +44,8 @@
 class DatabaseFeatures(BaseDatabaseFeatures):
     can_use_chunked_reads = False
     can_return_id_from_insert = True
+    supports_microsecond_precision = False
+    supports_transactions = True
     #uses_savepoints = True
 
 

django_pyodbc/creation.py

@@ -86,16 +86,25 @@ def _create_test_db(self, verbosity, autoclobber):
             self.connection.connection.commit()
             return test_name
 
+        if self.connection.ops.on_azure_sql_db:
+            self.connection.close()
+            settings_dict["NAME"] = 'master'
+
         return super(DatabaseCreation, self)._create_test_db(verbosity, autoclobber)
 
     def _destroy_test_db(self, test_database_name, verbosity):
         "Internal implementation - remove the test db tables."
         if self.connection.test_create:
+            if self.connection.ops.on_azure_sql_db:
+                self.connection.close()
+                self.connection.settings_dict["NAME"] = 'master'
+
             cursor = self.connection.cursor()
             self.connection.connection.autocommit = True
             #time.sleep(1) # To avoid "database is being accessed by other users" errors.
-            cursor.execute("ALTER DATABASE %s SET SINGLE_USER WITH ROLLBACK IMMEDIATE " % \
-                    self.connection.ops.quote_name(test_database_name))
+            if not self.connection.ops.on_azure_sql_db:
+                cursor.execute("ALTER DATABASE %s SET SINGLE_USER WITH ROLLBACK IMMEDIATE " % \
+                        self.connection.ops.quote_name(test_database_name))
             cursor.execute("DROP DATABASE %s" % \
                     self.connection.ops.quote_name(test_database_name))
         else:

django_pyodbc/operations.py

@@ -3,6 +3,8 @@
 import time
 import decimal
 
+EDITION_AZURE_SQL_DB = 5
+
 class DatabaseOperations(BaseDatabaseOperations):
     compiler_module = "django_pyodbc.compiler"
     def __init__(self, connection):
@@ -13,6 +15,7 @@ def __init__(self, connection):
 
         self.connection = connection
         self._ss_ver = None
+        self._ss_edition = None
 
     def _get_sql_server_ver(self):
         """
@@ -22,8 +25,11 @@ def _get_sql_server_ver(self):
             return self._ss_ver
         cur = self.connection.cursor()
         cur.execute("SELECT CAST(SERVERPROPERTY('ProductVersion') as varchar)")
-        ver_code = int(cur.fetchone()[0].split('.')[0])
-        if ver_code >= 10:
+        ver_code = cur.fetchone()[0]
+        ver_code = int(ver_code.split('.')[0])
+        if ver_code >= 11:
+            self._ss_ver = 2012
+        elif ver_code == 10:
             self._ss_ver = 2008
         elif ver_code == 9:
             self._ss_ver = 2005
@@ -32,6 +38,15 @@ def _get_sql_server_ver(self):
         return self._ss_ver
     sql_server_ver = property(_get_sql_server_ver)
 
+    def _on_azure_sql_db(self):
+        if self._ss_edition is not None:
+            return self._ss_edition == EDITION_AZURE_SQL_DB
+        cur = self.connection.cursor()
+        cur.execute("SELECT CAST(SERVERPROPERTY('EngineEdition') as integer)")
+        self._ss_edition = cur.fetchone()[0]
+        return self._ss_edition == EDITION_AZURE_SQL_DB
+    on_azure_sql_db = property(_on_azure_sql_db)
+
     def date_extract_sql(self, lookup_type, field_name):
         """
         Given a lookup_type of 'year', 'month', 'day' or 'week_day', returns
@@ -194,16 +209,24 @@ def sql_flush(self, style, tables, sequences):
                     (self.quote_name(fk[0]), self.quote_name(fk[1])) for fk in fks]
             sql_list.extend(['%s %s %s;' % (style.SQL_KEYWORD('DELETE'), style.SQL_KEYWORD('FROM'),
                              style.SQL_FIELD(self.quote_name(table)) ) for table in tables])
-            # Then reset the counters on each table.
-            sql_list.extend(['%s %s (%s, %s, %s) %s %s;' % (
-                style.SQL_KEYWORD('DBCC'),
-                style.SQL_KEYWORD('CHECKIDENT'),
-                style.SQL_FIELD(self.quote_name(seq["table"])),
-                style.SQL_KEYWORD('RESEED'),
-                style.SQL_FIELD('%d' % seq['start_id']),
-                style.SQL_KEYWORD('WITH'),
-                style.SQL_KEYWORD('NO_INFOMSGS'),
-                ) for seq in seqs])
+
+            if self.on_azure_sql_db:
+                import warnings
+                warnings.warn("The identity columns will never be reset " \
+                              "on Windows Azure SQL Database.",
+                              RuntimeWarning)
+            else:
+                # Then reset the counters on each table.
+                sql_list.extend(['%s %s (%s, %s, %s) %s %s;' % (
+                    style.SQL_KEYWORD('DBCC'),
+                    style.SQL_KEYWORD('CHECKIDENT'),
+                    style.SQL_FIELD(self.quote_name(seq["table"])),
+                    style.SQL_KEYWORD('RESEED'),
+                    style.SQL_FIELD('%d' % seq['start_id']),
+                    style.SQL_KEYWORD('WITH'),
+                    style.SQL_KEYWORD('NO_INFOMSGS'),
+                    ) for seq in seqs])
+
             sql_list.extend(['ALTER TABLE %s CHECK CONSTRAINT %s;' % \
                     (self.quote_name(fk[0]), self.quote_name(fk[1])) for fk in fks])
             return sql_list