moved trunk, branches and tags to project root

fixes Issue #5

1 files changed, 61 insertions, 0 deletions

diff --git a/migrate/changeset/databases/mysql.py b/migrate/changeset/databases/mysql.py
new file mode 100644
index 0000000..3611f66
--- /dev/null
+++ b/migrate/changeset/databases/mysql.py
@@ -0,0 +1,61 @@
+from migrate.changeset import ansisql,exceptions
+from sqlalchemy.databases import mysql as sa_base
+#import sqlalchemy as sa
+
+MySQLSchemaGenerator = sa_base.MySQLSchemaGenerator
+
+class MySQLColumnGenerator(MySQLSchemaGenerator,ansisql.ANSIColumnGenerator):
+    pass
+class MySQLColumnDropper(ansisql.ANSIColumnDropper):
+    pass
+class MySQLSchemaChanger(MySQLSchemaGenerator,ansisql.ANSISchemaChanger):
+    def visit_column(self,delta):
+        keys = delta.keys()
+        if 'type' in keys or 'nullable' in keys or 'name' in keys:
+            self._run_subvisit(delta,self._visit_column_change)
+        if 'default' in keys:
+            # Column name might have changed above
+            col_name = delta.get('name',delta.current_name)
+            self._run_subvisit(delta,self._visit_column_default,col_name=col_name)
+    def _visit_column_change(self,table_name,col_name,delta):
+        if not hasattr(delta,'result_column'):
+            # Mysql needs the whole column definition, not just a lone name/type
+            raise exceptions.NotSupportedError(
+                "A column object is required to do this")
+                
+        column = delta.result_column
+        colspec = self.get_column_specification(column)
+        self.start_alter_table(table_name)
+        self.append("CHANGE COLUMN ")
+        self.append(col_name)
+        self.append(' ')
+        self.append(colspec)
+    def visit_index(self,param):
+        # If MySQL can do this, I can't find how
+        raise exceptions.NotSupportedError("MySQL cannot rename indexes")
+class MySQLConstraintGenerator(ansisql.ANSIConstraintGenerator):
+    pass
+class MySQLConstraintDropper(ansisql.ANSIConstraintDropper):
+    #def visit_constraint(self,constraint):
+    #    if isinstance(constraint,sqlalchemy.schema.PrimaryKeyConstraint):
+    #        return self._visit_constraint_pk(constraint)
+    #    elif isinstance(constraint,sqlalchemy.schema.ForeignKeyConstraint):
+    #        return self._visit_constraint_fk(constraint)
+    #    return super(MySQLConstraintDropper,self).visit_constraint(constraint)
+    def visit_migrate_primary_key_constraint(self,constraint):
+        self.start_alter_table(constraint)
+        self.append("DROP PRIMARY KEY")
+        self.execute()
+
+    def visit_migrate_foreign_key_constraint(self,constraint):
+        self.start_alter_table(constraint)
+        self.append("DROP FOREIGN KEY ")
+        self.append(constraint.name)
+        self.execute()
+
+class MySQLDialect(ansisql.ANSIDialect):
+    columngenerator = MySQLColumnGenerator
+    columndropper = MySQLColumnDropper
+    schemachanger = MySQLSchemaChanger
+    constraintgenerator = MySQLConstraintGenerator
+    constraintdropper = MySQLConstraintDropper