summaryrefslogtreecommitdiff
path: root/django/db
diff options
context:
space:
mode:
authorSimon Charette <charette.s@gmail.com>2016-08-20 18:05:04 -0400
committerSimon Charette <charette.s@gmail.com>2016-08-24 01:38:30 -0400
commit0b454804db5729eedbb117c90de4a3ad5fb397ee (patch)
treee6667315c3b5777ed69c4eaa86e909554ff5ee50 /django/db
parent1c60765d6307cf5bb659d9157a990d7410108424 (diff)
[1.10.x] Fixed #27100 -- Included already applied migration changes in the pre-migrate state.
Refs #24100. Thanks Tim for the review. Backport of d5c4ea524679a787fe11c927448e44e95646096b from master
Diffstat (limited to 'django/db')
-rw-r--r--django/db/migrations/executor.py53
1 files changed, 27 insertions, 26 deletions
diff --git a/django/db/migrations/executor.py b/django/db/migrations/executor.py
index 11a96ec81b..1a0b6f6322 100644
--- a/django/db/migrations/executor.py
+++ b/django/db/migrations/executor.py
@@ -63,10 +63,25 @@ class MigrationExecutor(object):
applied.add(migration)
return plan
- def _create_project_state(self):
- return ProjectState(real_apps=list(self.loader.unmigrated_apps))
+ def _create_project_state(self, with_applied_migrations=False):
+ """
+ Create a project state including all the applications without
+ migrations and applied migrations if with_applied_migrations=True.
+ """
+ state = ProjectState(real_apps=list(self.loader.unmigrated_apps))
+ if with_applied_migrations:
+ # Create the forwards plan Django would follow on an empty database
+ full_plan = self.migration_plan(self.loader.graph.leaf_nodes(), clean_start=True)
+ applied_migrations = {
+ self.loader.graph.nodes[key] for key in self.loader.applied_migrations
+ if key in self.loader.graph.nodes
+ }
+ for migration, _ in full_plan:
+ if migration in applied_migrations:
+ migration.mutate_state(state, preserve=False)
+ return state
- def migrate(self, targets, plan=None, fake=False, fake_initial=False):
+ def migrate(self, targets, plan=None, state=None, fake=False, fake_initial=False):
"""
Migrates the database up to the given targets.
@@ -82,15 +97,9 @@ class MigrationExecutor(object):
all_backwards = all(backwards for mig, backwards in plan)
if not plan:
- # The resulting state should include applied migrations.
- state = self._create_project_state()
- applied_migrations = {
- self.loader.graph.nodes[key] for key in self.loader.applied_migrations
- if key in self.loader.graph.nodes
- }
- for migration, _ in full_plan:
- if migration in applied_migrations:
- migration.mutate_state(state, preserve=False)
+ if state is None:
+ # The resulting state should include applied migrations.
+ state = self._create_project_state(with_applied_migrations=True)
elif all_forwards == all_backwards:
# This should only happen if there's a mixed plan
raise InvalidMigrationPlan(
@@ -100,7 +109,10 @@ class MigrationExecutor(object):
plan
)
elif all_forwards:
- state = self._migrate_all_forwards(plan, full_plan, fake=fake, fake_initial=fake_initial)
+ if state is None:
+ # The resulting state should still include applied migrations.
+ state = self._create_project_state(with_applied_migrations=True)
+ state = self._migrate_all_forwards(state, plan, full_plan, fake=fake, fake_initial=fake_initial)
else:
# No need to check for `elif all_backwards` here, as that condition
# would always evaluate to true.
@@ -110,19 +122,14 @@ class MigrationExecutor(object):
return state
- def _migrate_all_forwards(self, plan, full_plan, fake, fake_initial):
+ def _migrate_all_forwards(self, state, plan, full_plan, fake, fake_initial):
"""
Take a list of 2-tuples of the form (migration instance, False) and
apply them in the order they occur in the full_plan.
"""
migrations_to_run = {m[0] for m in plan}
- state = self._create_project_state()
- applied_migrations = {
- self.loader.graph.nodes[key] for key in self.loader.applied_migrations
- if key in self.loader.graph.nodes
- }
for migration, _ in full_plan:
- if not migrations_to_run and not applied_migrations:
+ if not migrations_to_run:
# We remove every migration that we applied from these sets so
# that we can bail out once the last migration has been applied
# and don't always run until the very end of the migration
@@ -137,12 +144,6 @@ class MigrationExecutor(object):
self.progress_callback("render_success")
state = self.apply_migration(state, migration, fake=fake, fake_initial=fake_initial)
migrations_to_run.remove(migration)
- elif migration in applied_migrations:
- # Only mutate the state if the migration is actually applied
- # to make sure the resulting state doesn't include changes
- # from unrelated migrations.
- migration.mutate_state(state, preserve=False)
- applied_migrations.remove(migration)
return state