Make sqlmigrate ignore the RunPython operation

This commit is contained in:
Andrew Godwin 2013-09-25 14:37:44 +01:00
parent 3b810c5656
commit 8a3e543f26
3 changed files with 25 additions and 5 deletions

View File

@ -71,9 +71,9 @@ class MigrationExecutor(object):
with self.connection.schema_editor(collect_sql=True) as schema_editor: with self.connection.schema_editor(collect_sql=True) as schema_editor:
project_state = self.loader.graph.project_state((migration.app_label, migration.name), at_end=False) project_state = self.loader.graph.project_state((migration.app_label, migration.name), at_end=False)
if not backwards: if not backwards:
migration.apply(project_state, schema_editor) migration.apply(project_state, schema_editor, collect_sql=True)
else: else:
migration.unapply(project_state, schema_editor) migration.unapply(project_state, schema_editor, collect_sql=True)
statements.extend(schema_editor.collected_sql) statements.extend(schema_editor.collected_sql)
return statements return statements

View File

@ -67,7 +67,7 @@ class Migration(object):
operation.state_forwards(self.app_label, new_state) operation.state_forwards(self.app_label, new_state)
return new_state return new_state
def apply(self, project_state, schema_editor): def apply(self, project_state, schema_editor, collect_sql=False):
""" """
Takes a project_state representing all migrations prior to this one Takes a project_state representing all migrations prior to this one
and a schema_editor for a live database and applies the migration and a schema_editor for a live database and applies the migration
@ -77,6 +77,14 @@ class Migration(object):
Migrations. Migrations.
""" """
for operation in self.operations: for operation in self.operations:
# If this operation cannot be represented as SQL, place a comment
# there instead
if collect_sql and not operation.reduces_to_sql:
schema_editor.collected_sql.append("--")
schema_editor.collected_sql.append("-- MIGRATION NOW PERFORMS OPERATION THAT CANNOT BE WRITTEN AS SQL:")
schema_editor.collected_sql.append("-- %s" % operation.describe())
schema_editor.collected_sql.append("--")
continue
# Get the state after the operation has run # Get the state after the operation has run
new_state = project_state.clone() new_state = project_state.clone()
operation.state_forwards(self.app_label, new_state) operation.state_forwards(self.app_label, new_state)
@ -86,7 +94,7 @@ class Migration(object):
project_state = new_state project_state = new_state
return project_state return project_state
def unapply(self, project_state, schema_editor): def unapply(self, project_state, schema_editor, collect_sql=False):
""" """
Takes a project_state representing all migrations prior to this one Takes a project_state representing all migrations prior to this one
and a schema_editor for a live database and applies the migration and a schema_editor for a live database and applies the migration
@ -95,8 +103,17 @@ class Migration(object):
# We need to pre-calculate the stack of project states # We need to pre-calculate the stack of project states
to_run = [] to_run = []
for operation in self.operations: for operation in self.operations:
# If this operation cannot be represented as SQL, place a comment
# there instead
if collect_sql and not operation.reduces_to_sql:
schema_editor.collected_sql.append("--")
schema_editor.collected_sql.append("-- MIGRATION NOW PERFORMS OPERATION THAT CANNOT BE WRITTEN AS SQL:")
schema_editor.collected_sql.append("-- %s" % operation.describe())
schema_editor.collected_sql.append("--")
continue
# If it's irreversible, error out
if not operation.reversible: if not operation.reversible:
raise Migration.IrreversibleError("Operation %s in %s is not reversible" % (operation, sekf)) raise Migration.IrreversibleError("Operation %s in %s is not reversible" % (operation, self))
new_state = project_state.clone() new_state = project_state.clone()
operation.state_forwards(self.app_label, new_state) operation.state_forwards(self.app_label, new_state)
to_run.append((operation, project_state, new_state)) to_run.append((operation, project_state, new_state))

View File

@ -132,3 +132,6 @@ class RunPython(Operation):
def database_backwards(self, app_label, schema_editor, from_state, to_state): def database_backwards(self, app_label, schema_editor, from_state, to_state):
raise NotImplementedError("You cannot reverse this operation") raise NotImplementedError("You cannot reverse this operation")
def describe(self):
return "Raw Python operation"