Fix bug when inserting with a specific primary key.

Author: avidal

sql_server/pyodbc/compiler.py

@@ -357,6 +357,7 @@ def as_sql(self):
             # db_column is None if not explicitly specified by model field
             auto_field_column = meta.auto_field.db_column or meta.auto_field.column
 
+            out = []
             for item in items:
                 sql, params = item
                 if auto_field_column in columns:
@@ -367,7 +368,8 @@ def as_sql(self):
                     else:
                         sql = "SET IDENTITY_INSERT %s ON;\n%s;\nSET IDENTITY_INSERT %s OFF" % \
                             (quoted_table, sql, quoted_table)
-                item = [sql, params]
+                out.append([sql, params])
+            items = out
 
         return items
 

sql_server/pyodbc/query.py

@@ -73,7 +73,7 @@ def _insert_as_sql(self, *args, **kwargs):
             quoted_table = self.connection.ops.quote_name(meta.db_table)
             # Get (sql, params) from original InsertQuery.as_sql
             sql, params = self._orig_as_sql(*args, **kwargs)
-            if meta.pk.db_column in self.columns and meta.pk.__class__.__name__ == "AutoField":
+            if meta.has_auto_field and meta.auto_field.column in self.columns:
                 if len(self.columns) == 1 and not params:
                     sql = "INSERT INTO %s DEFAULT VALUES" % quoted_table
                 else: