From 2a9c4b4901d385f68a2c3f9cf98b7a776c2976f0 Mon Sep 17 00:00:00 2001
From: Claude Paroz <claude@2xlibre.net>
Date: Sun, 26 Oct 2014 18:30:58 +0100
Subject: [PATCH] Passed around the state between migrations

Refs #23745.
---
 django/db/migrations/executor.py  | 25 ++++++++++++++++---------
 tests/migrations/test_executor.py |  2 +-
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/django/db/migrations/executor.py b/django/db/migrations/executor.py
index 4a73546ba8..2e42e7c6d5 100644
--- a/django/db/migrations/executor.py
+++ b/django/db/migrations/executor.py
@@ -63,11 +63,14 @@ class MigrationExecutor(object):
         """
         if plan is None:
             plan = self.migration_plan(targets)
+        state = None
         for migration, backwards in plan:
+            if state is None:
+                state = self.loader.project_state((migration.app_label, migration.name), at_end=False)
             if not backwards:
-                self.apply_migration(migration, fake=fake)
+                state = self.apply_migration(state, migration, fake=fake)
             else:
-                self.unapply_migration(migration, fake=fake)
+                state = self.unapply_migration(state, migration, fake=fake)
 
     def collect_sql(self, plan):
         """
@@ -75,17 +78,19 @@ class MigrationExecutor(object):
         statements that represent the best-efforts version of that plan.
         """
         statements = []
+        state = None
         for migration, backwards in plan:
             with self.connection.schema_editor(collect_sql=True) as schema_editor:
-                project_state = self.loader.project_state((migration.app_label, migration.name), at_end=False)
+                if state is None:
+                    state = self.loader.project_state((migration.app_label, migration.name), at_end=False)
                 if not backwards:
-                    migration.apply(project_state, schema_editor, collect_sql=True)
+                    state = migration.apply(state, schema_editor, collect_sql=True)
                 else:
-                    migration.unapply(project_state, schema_editor, collect_sql=True)
+                    state = migration.unapply(state, schema_editor, collect_sql=True)
             statements.extend(schema_editor.collected_sql)
         return statements
 
-    def apply_migration(self, migration, fake=False):
+    def apply_migration(self, state, migration, fake=False):
         """
         Runs a migration forwards.
         """
@@ -93,7 +98,7 @@ class MigrationExecutor(object):
             self.progress_callback("apply_start", migration, fake)
         if not fake:
             # Test to see if this is an already-applied initial migration
-            if self.detect_soft_applied(migration):
+            if self.detect_soft_applied(state, migration):
                 fake = True
             else:
                 # Alright, do it normally
@@ -109,8 +114,9 @@ class MigrationExecutor(object):
         # Report progress
         if self.progress_callback:
             self.progress_callback("apply_success", migration, fake)
+        return state
 
-    def unapply_migration(self, migration, fake=False):
+    def unapply_migration(self, state, migration, fake=False):
         """
         Runs a migration backwards.
         """
@@ -129,8 +135,9 @@ class MigrationExecutor(object):
         # Report progress
         if self.progress_callback:
             self.progress_callback("unapply_success", migration, fake)
+        return state
 
-    def detect_soft_applied(self, migration):
+    def detect_soft_applied(self, project_state, migration):
         """
         Tests whether a migration has been implicitly applied - that the
         tables it would create exist. This is intended only for use
diff --git a/tests/migrations/test_executor.py b/tests/migrations/test_executor.py
index d21dc69472..3a7a779f8c 100644
--- a/tests/migrations/test_executor.py
+++ b/tests/migrations/test_executor.py
@@ -223,7 +223,7 @@ class ExecutorTests(MigrationTestBase):
         global_apps.get_app_config("migrations").models["author"] = migrations_apps.get_model("migrations", "author")
         try:
             migration = executor.loader.get_migration("auth", "0001_initial")
-            self.assertEqual(executor.detect_soft_applied(migration), True)
+            self.assertEqual(executor.detect_soft_applied(None, migration), True)
         finally:
             connection.introspection.table_names = old_table_names
             del global_apps.get_app_config("migrations").models["author"]