mirror of
https://github.com/django/django.git
synced 2024-11-29 22:56:46 +01:00
81f5408c7a
Thanks to bak1an for the patch.
97 lines
5.4 KiB
Diff
97 lines
5.4 KiB
Diff
diff --git a/django/db/migrations/autodetector.py b/django/db/migrations/autodetector.py
|
|
index cbc4604..152fde9 100644
|
|
--- a/django/db/migrations/autodetector.py
|
|
+++ b/django/db/migrations/autodetector.py
|
|
@@ -136,6 +136,7 @@ def _rel_agnostic_fields_def(fields):
|
|
# Phase 2 is progressively adding pending models, splitting up into two
|
|
# migrations if required.
|
|
pending_new_fks = []
|
|
+ pending_unique_together = []
|
|
added_phase_2 = set()
|
|
while pending_add:
|
|
# Is there one we can add that has all dependencies satisfied?
|
|
@@ -172,6 +173,11 @@ def _rel_agnostic_fields_def(fields):
|
|
else:
|
|
(app_label, model_name), related_fields = sorted(pending_add.items())[0]
|
|
model_state = self.to_state.models[app_label, model_name]
|
|
+ # Defer unique together constraints creation, see ticket #22275
|
|
+ unique_together_constraints = model_state.options.pop('unique_together', None)
|
|
+ if unique_together_constraints:
|
|
+ pending_unique_together.append((app_label, model_name,
|
|
+ unique_together_constraints))
|
|
# Work out the fields that need splitting out
|
|
bad_fields = dict((f, (al, mn)) for f, al, mn in related_fields if (al, mn) in pending_add)
|
|
# Create the model, without those
|
|
@@ -206,6 +212,15 @@ def _rel_agnostic_fields_def(fields):
|
|
self.add_swappable_dependency(app_label, swappable_setting)
|
|
elif app_label != other_app_label:
|
|
self.add_dependency(app_label, other_app_label)
|
|
+ # Phase 3.1 - unique together constraints
|
|
+ for app_label, model_name, unique_together in pending_unique_together:
|
|
+ self.add_to_migration(
|
|
+ app_label,
|
|
+ operations.AlterUniqueTogether(
|
|
+ name=model_name,
|
|
+ unique_together=unique_together
|
|
+ )
|
|
+ )
|
|
# Removing models
|
|
removed_models = set(old_model_keys) - set(new_model_keys)
|
|
for app_label, model_name in removed_models:
|
|
diff --git a/tests/migrations/test_autodetector.py b/tests/migrations/test_autodetector.py
|
|
index c3ada83..40e540d 100644
|
|
--- a/tests/migrations/test_autodetector.py
|
|
+++ b/tests/migrations/test_autodetector.py
|
|
@@ -37,6 +37,8 @@ class AutodetectorTests(TestCase):
|
|
book_unique_3 = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("newfield", models.IntegerField()), ("author", models.ForeignKey("testapp.Author")), ("title", models.CharField(max_length=200))], {"unique_together": [("title", "newfield")]})
|
|
edition = ModelState("thirdapp", "Edition", [("id", models.AutoField(primary_key=True)), ("book", models.ForeignKey("otherapp.Book"))])
|
|
custom_user = ModelState("thirdapp", "CustomUser", [("id", models.AutoField(primary_key=True)), ("username", models.CharField(max_length=255))])
|
|
+ knight = ModelState("eggs", "Knight", [("id", models.AutoField(primary_key=True))])
|
|
+ rabbit = ModelState("eggs", "Rabbit", [("id", models.AutoField(primary_key=True)), ("knight", models.ForeignKey("eggs.Knight")), ("parent", models.ForeignKey("eggs.Rabbit"))], {"unique_together": [("parent", "knight")]})
|
|
|
|
def make_project_state(self, model_states):
|
|
"Shortcut to make ProjectStates from lists of predefined models"
|
|
@@ -370,6 +372,42 @@ def test_same_app_circular_fk_dependency(self):
|
|
self.assertEqual(migration1.dependencies, [])
|
|
self.assertEqual(migration2.dependencies, [("testapp", "auto_1")])
|
|
|
|
+ def test_same_app_circular_fk_dependency_and_unique_together(self):
|
|
+ """
|
|
+ Tests that a migration with circular FK dependency does not try to
|
|
+ create unique together constraint before creating all required fields first.
|
|
+ See ticket #22275.
|
|
+ """
|
|
+ # Make state
|
|
+ before = self.make_project_state([])
|
|
+ after = self.make_project_state([self.knight, self.rabbit])
|
|
+ autodetector = MigrationAutodetector(before, after)
|
|
+ changes = autodetector._detect_changes()
|
|
+ # Right number of migrations?
|
|
+ self.assertEqual(len(changes['eggs']), 2)
|
|
+ # Right number of actions?
|
|
+ migration1 = changes['eggs'][0]
|
|
+ self.assertEqual(len(migration1.operations), 2)
|
|
+ migration2 = changes['eggs'][1]
|
|
+ self.assertEqual(len(migration2.operations), 2)
|
|
+ # Right actions?
|
|
+ action = migration1.operations[0]
|
|
+ self.assertEqual(action.__class__.__name__, "CreateModel")
|
|
+ action = migration1.operations[1]
|
|
+ self.assertEqual(action.__class__.__name__, "CreateModel")
|
|
+ # CreateModel action for Rabbit should not have unique_together now
|
|
+ self.assertEqual(action.name, "Rabbit")
|
|
+ self.assertFalse("unique_together" in action.options)
|
|
+ action = migration2.operations[0]
|
|
+ self.assertEqual(action.__class__.__name__, "AddField")
|
|
+ self.assertEqual(action.name, "parent")
|
|
+ action = migration2.operations[1]
|
|
+ self.assertEqual(action.__class__.__name__, "AlterUniqueTogether")
|
|
+ self.assertEqual(action.name, "rabbit")
|
|
+ # Right dependencies?
|
|
+ self.assertEqual(migration1.dependencies, [])
|
|
+ self.assertEqual(migration2.dependencies, [("eggs", "auto_1")])
|
|
+
|
|
def test_unique_together(self):
|
|
"Tests unique_together detection"
|
|
# Make state
|