[1.7.x] Fixed #24015 -- Factorized create_index_sql expression

Backport of 6072f17d0 from master, with one test reinforced.
Thanks Tim Graham for the review.
This commit is contained in:
Claude Paroz 2014-06-20 14:32:11 +02:00
parent 2f13a48f33
commit f46a16614d
3 changed files with 20 additions and 26 deletions

View File

@ -322,14 +322,9 @@ class BaseDatabaseSchemaEditor(object):
)) ))
self.execute(self._delete_constraint_sql(self.sql_delete_index, model, constraint_names[0])) self.execute(self._delete_constraint_sql(self.sql_delete_index, model, constraint_names[0]))
# Created indexes # Created indexes
for fields in news.difference(olds): for field_names in news.difference(olds):
columns = [model._meta.get_field_by_name(field)[0].column for field in fields] fields = [model._meta.get_field_by_name(field)[0] for field in field_names]
self.execute(self.sql_create_index % { self.execute(self._create_index_sql(model, fields, suffix="_idx"))
"table": self.quote_name(model._meta.db_table),
"name": self._create_index_name(model, columns, suffix="_idx"),
"columns": ", ".join(self.quote_name(column) for column in columns),
"extra": "",
})
def alter_db_table(self, model, old_db_table, new_db_table): def alter_db_table(self, model, old_db_table, new_db_table):
""" """
@ -389,14 +384,7 @@ class BaseDatabaseSchemaEditor(object):
self.execute(sql) self.execute(sql)
# Add an index, if required # Add an index, if required
if field.db_index and not field.unique: if field.db_index and not field.unique:
self.deferred_sql.append( self.deferred_sql.append(self._create_index_sql(model, [field]))
self.sql_create_index % {
"name": self._create_index_name(model, [field.column], suffix=""),
"table": self.quote_name(model._meta.db_table),
"columns": self.quote_name(field.column),
"extra": "",
}
)
# Add any FK constraints later # Add any FK constraints later
if field.rel and self.connection.features.supports_foreign_keys and field.db_constraint: if field.rel and self.connection.features.supports_foreign_keys and field.db_constraint:
self.deferred_sql.append(self._create_fk_sql(model, field, "_fk_%(to_table)s_%(to_column)s")) self.deferred_sql.append(self._create_fk_sql(model, field, "_fk_%(to_table)s_%(to_column)s"))
@ -637,15 +625,10 @@ class BaseDatabaseSchemaEditor(object):
if not old_field.unique and new_field.unique: if not old_field.unique and new_field.unique:
self.execute(self._create_unique_sql(model, [new_field.column])) self.execute(self._create_unique_sql(model, [new_field.column]))
# Added an index? # Added an index?
if not old_field.db_index and new_field.db_index and not new_field.unique and not (not old_field.unique and new_field.unique): if (not old_field.db_index and new_field.db_index and
self.execute( not new_field.unique and not
self.sql_create_index % { (not old_field.unique and new_field.unique)):
"table": self.quote_name(model._meta.db_table), self.execute(self._create_index_sql(model, [new_field], suffix="_uniq"))
"name": self._create_index_name(model, [new_field.column], suffix="_uniq"),
"columns": self.quote_name(new_field.column),
"extra": "",
}
)
# Type alteration on primary key? Then we need to alter the column # Type alteration on primary key? Then we need to alter the column
# referring to us. # referring to us.
rels_to_update = [] rels_to_update = []

View File

@ -155,3 +155,6 @@ Bugfixes
* Ensured the app registry is fully populated when unpickling models. When an * Ensured the app registry is fully populated when unpickling models. When an
external script (like a queueing infrastructure) reloads pickled models, it external script (like a queueing infrastructure) reloads pickled models, it
could crash with an ``AppRegistryNotReady`` exception (:ticket:`24007`). could crash with an ``AppRegistryNotReady`` exception (:ticket:`24007`).
* Added quoting to field indexes in the SQL generated by migrations to prevent
a crash when the index name requires it (:ticket:`#24015`).

View File

@ -46,8 +46,16 @@ class SchemaIndexesTests(TestCase):
Test index handling by the db.backends.schema infrastructure. Test index handling by the db.backends.schema infrastructure.
""" """
def test_index_together(self): def test_index_together(self):
index_sql = connection.schema_editor()._model_indexes_sql(Article) editor = connection.schema_editor()
index_sql = editor._model_indexes_sql(Article)
self.assertEqual(len(index_sql), 1) self.assertEqual(len(index_sql), 1)
# Ensure the index name is properly quoted
self.assertIn(
connection.ops.quote_name(
editor._create_index_name(Article, ['headline', 'pub_date'], suffix='_idx')
),
index_sql[0]
)
def test_index_together_single_list(self): def test_index_together_single_list(self):
# Test for using index_together with a single list (#22172) # Test for using index_together with a single list (#22172)