fe64667106
The "force" parameter in SQLAlchemy IdentifierPreparer.quote()
has been a no-op since 0.9 in
031ef08078,
which was six years ago. In SQLAlchemy 1.3 this parameter
will be removed entirely. Bump requirements to 0.9 series
here and remove usage of the "force" flag.
Change-Id: I4492df2e7d2075fefbf13d6782de11f7d402f6b8
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
"""
|
|
MySQL database specific implementations of changeset classes.
|
|
"""
|
|
|
|
import sqlalchemy
|
|
from sqlalchemy.databases import mysql as sa_base
|
|
from sqlalchemy import types as sqltypes
|
|
|
|
from migrate import exceptions
|
|
from migrate.changeset import ansisql
|
|
from migrate.changeset import util
|
|
|
|
|
|
|
|
MySQLSchemaGenerator = sa_base.MySQLDDLCompiler
|
|
|
|
class MySQLColumnGenerator(MySQLSchemaGenerator, ansisql.ANSIColumnGenerator):
|
|
pass
|
|
|
|
|
|
class MySQLColumnDropper(ansisql.ANSIColumnDropper):
|
|
pass
|
|
|
|
|
|
class MySQLSchemaChanger(MySQLSchemaGenerator, ansisql.ANSISchemaChanger):
|
|
|
|
def visit_column(self, delta):
|
|
table = delta.table
|
|
colspec = self.get_column_specification(delta.result_column)
|
|
if delta.result_column.autoincrement:
|
|
primary_keys = [c for c in table.primary_key.columns
|
|
if (c.autoincrement and
|
|
isinstance(c.type, sqltypes.Integer) and
|
|
not c.foreign_keys)]
|
|
|
|
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)
|
|
|
|
self.start_alter_table(table)
|
|
|
|
self.append("CHANGE COLUMN %s " % old_col_name)
|
|
self.append(colspec)
|
|
self.execute()
|
|
|
|
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(MySQLSchemaGenerator, ansisql.ANSIConstraintDropper):
|
|
def visit_migrate_check_constraint(self, *p, **k):
|
|
raise exceptions.NotSupportedError("MySQL does not support CHECK"
|
|
" constraints, use triggers instead.")
|
|
|
|
|
|
class MySQLDialect(ansisql.ANSIDialect):
|
|
columngenerator = MySQLColumnGenerator
|
|
columndropper = MySQLColumnDropper
|
|
schemachanger = MySQLSchemaChanger
|
|
constraintgenerator = MySQLConstraintGenerator
|
|
constraintdropper = MySQLConstraintDropper
|