add index creation/deletion support to SchemaEditor.alter_field()

Author: timgraham

django_mongodb/features.py

@@ -72,16 +72,12 @@ class DatabaseFeatures(BaseDatabaseFeatures):
         # AddField
         "schema.tests.SchemaTests.test_add_unique_charfield",
         # AlterField
-        "schema.tests.SchemaTests.test_alter_field_add_index_to_integerfield",
         "schema.tests.SchemaTests.test_alter_field_fk_to_o2o",
         "schema.tests.SchemaTests.test_alter_field_o2o_keeps_unique",
         "schema.tests.SchemaTests.test_alter_field_o2o_to_fk",
         "schema.tests.SchemaTests.test_alter_int_pk_to_int_unique",
-        # AlterField (db_index)
-        "schema.tests.SchemaTests.test_alter_renames_index",
-        "schema.tests.SchemaTests.test_indexes",
-        "schema.tests.SchemaTests.test_remove_db_index_doesnt_remove_custom_indexes",
         # AlterField (unique)
+        "schema.tests.SchemaTests.test_indexes",
         "schema.tests.SchemaTests.test_unique",
         "schema.tests.SchemaTests.test_unique_and_reverse_m2m",
         # alter_unique_together

django_mongodb/schema.py

@@ -66,15 +66,27 @@ def _alter_field(
         strict=False,
     ):
         collection = self.connection.database[model._meta.db_table]
+        # Removed an index?
+        old_field_indexed = self._field_should_be_indexed(model, old_field)
+        new_field_indexed = self._field_should_be_indexed(model, new_field)
+        if old_field_indexed and not new_field_indexed:
+            self._remove_field_index(model, old_field)
         # Have they renamed the column?
         if old_field.column != new_field.column:
             collection.update_many({}, {"$rename": {old_field.column: new_field.column}})
+            # Move index to the new field, if needed.
+            if old_field_indexed and new_field_indexed:
+                self._remove_field_index(model, old_field)
+                self._add_field_index(model, new_field)
         # Replace NULL with the field default if the field and was changed from
         # NULL to NOT NULL.
         if new_field.has_default() and old_field.null and not new_field.null:
             column = new_field.column
             default = self.effective_default(new_field)
             collection.update_many({column: {"$eq": None}}, [{"$set": {column: default}}])
+        # Added an index?
+        if not old_field_indexed and new_field_indexed:
+            self._add_field_index(model, new_field)
 
     def remove_field(self, model, field):
         # Remove implicit M2M tables.
@@ -155,6 +167,27 @@ def _remove_composed_index(self, model, field_names, constraint_kwargs):
         collection = self.connection.database[model._meta.db_table]
         collection.drop_index(constraint_names[0])
 
+    def _remove_field_index(self, model, field):
+        """Remove a field's db_index=True index."""
+        collection = self.connection.database[model._meta.db_table]
+        meta_index_names = {index.name for index in model._meta.indexes}
+        index_names = self._constraint_names(
+            model,
+            [field.column],
+            index=True,
+            # Retrieve only BTREE indexes since this is what's created with
+            # db_index=True.
+            type_=Index.suffix,
+            exclude=meta_index_names,
+        )
+        if len(index_names) != 1:
+            num_found = len(index_names)
+            raise ValueError(
+                f"Found wrong number ({num_found}) of constraints for "
+                f"{model._meta.db_table}.{field.column}."
+            )
+        collection.drop_index(index_names[0])
+
     def add_constraint(self, model, constraint):
         pass