From ece9ffed35e800fc10a6c80e79d300d7b52f2e92 Mon Sep 17 00:00:00 2001
From: iElectric <unknown>
Date: Sun, 2 May 2010 20:31:50 +0200
Subject: fix MySQL failing tests with autoincrement

---
 migrate/changeset/databases/mysql.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'migrate')

diff --git a/migrate/changeset/databases/mysql.py b/migrate/changeset/databases/mysql.py
index f32ea5f..191b77a 100644
--- a/migrate/changeset/databases/mysql.py
+++ b/migrate/changeset/databases/mysql.py
@@ -25,13 +25,15 @@ class MySQLSchemaChanger(MySQLSchemaGenerator, ansisql.ANSISchemaChanger):
         table = delta.table
         colspec = self.get_column_specification(delta.result_column)
         if delta.result_column.autoincrement:
-            first = [c for c in table.primary_key.columns
+            primary_keys = [c for c in table.primary_key.columns
                        if (c.autoincrement and
                             isinstance(c.type, sqltypes.Integer) and
-                            not c.foreign_keys)].pop(0)
+                            not c.foreign_keys)]
 
-            if first.name == delta.current_name:
-                colspec += " AUTO_INCREMENT"
+            if primary_keys:
+                first = primary_keys.pop(0)
+                if first.name == delta.current_name:
+                    colspec += " AUTO_INCREMENT"
         old_col_name = self.preparer.quote(delta.current_name, table.quote)
 
         self.start_alter_table(table)
-- 
cgit v1.2.1