implement SchemaEditor.alter_unique_together()

Author: timgraham

django_mongodb/features.py

@@ -87,9 +87,6 @@ class DatabaseFeatures(BaseDatabaseFeatures):
         # AlterField (unique)
         "schema.tests.SchemaTests.test_indexes",
         "schema.tests.SchemaTests.test_unique",
-        # alter_unique_together
-        "migrations.test_operations.OperationTests.test_alter_unique_together",
-        "schema.tests.SchemaTests.test_unique_together",
     }
     # $bitAnd, #bitOr, and $bitXor are new in MongoDB 6.3.
     _django_test_expected_failures_bitwise = {

django_mongodb/schema.py

@@ -35,6 +35,9 @@ def _create_model_indexes(self, model):
             index = Index(fields=field_names)
             index.set_name_with_model(model)
             self.add_index(model, index)
+        # Meta.unique_together
+        if model._meta.unique_together:
+            self.alter_unique_together(model, [], model._meta.unique_together)
         # Meta.constraints
         for constraint in model._meta.constraints:
             self.add_constraint(model, constraint)
@@ -177,7 +180,25 @@ def alter_index_together(self, model, old_index_together, new_index_together):
             self.add_index(model, idx)
 
     def alter_unique_together(self, model, old_unique_together, new_unique_together):
-        pass
+        def _get_name(field_names):
+            return "{}_uniq{}".format("_".join(field_names), len(field_names))
+
+        olds = {tuple(fields) for fields in old_unique_together}
+        news = {tuple(fields) for fields in new_unique_together}
+        # Deleted uniques
+        for field_names in olds.difference(news):
+            constraint = UniqueConstraint(
+                fields=field_names,
+                name=_get_name(field_names),
+            )
+            self.remove_constraint(model, constraint)
+        # Created uniques
+        for field_names in news.difference(olds):
+            constraint = UniqueConstraint(
+                fields=field_names,
+                name=_get_name(field_names),
+            )
+            self.add_constraint(model, constraint)
 
     def add_index(self, model, index, field=None, unique=False):
         if index.contains_expressions: