implement SchemaEditor.add/remove_field()

Author: timgraham

django_mongodb/features.py

@@ -86,24 +86,6 @@ class DatabaseFeatures(BaseDatabaseFeatures):
         "backends.tests.ThreadTests.test_pass_connection_between_threads",
         "backends.tests.ThreadTests.test_closing_non_shared_connections",
         "backends.tests.ThreadTests.test_default_connection_thread_local",
-        # AddField
-        "schema.tests.SchemaTests.test_add_datefield_and_datetimefield_use_effective_default",
-        "schema.tests.SchemaTests.test_add_field",
-        "schema.tests.SchemaTests.test_add_field_binary",
-        "schema.tests.SchemaTests.test_add_field_both_defaults_preserves_db_default",
-        "schema.tests.SchemaTests.test_add_field_default_dropped",
-        "schema.tests.SchemaTests.test_add_field_default_nullable",
-        "schema.tests.SchemaTests.test_add_field_default_transform",
-        "schema.tests.SchemaTests.test_add_field_durationfield_with_default",
-        "schema.tests.SchemaTests.test_add_field_o2o_nullable",
-        "schema.tests.SchemaTests.test_add_field_temp_default",
-        "schema.tests.SchemaTests.test_add_field_temp_default_boolean",
-        "schema.tests.SchemaTests.test_add_field_use_effective_default",
-        "schema.tests.SchemaTests.test_add_text_field_with_db_default",
-        "schema.tests.SchemaTests.test_add_textfield_default_nullable",
-        # RemoveField
-        "schema.tests.SchemaTests.test_remove_field",
-        "schema.tests.SchemaTests.test_remove_indexed_field",
         # Add/RemoveIndex
         "schema.tests.SchemaTests.test_add_remove_index",
         "schema.tests.SchemaTests.test_composed_desc_index_with_fk",
@@ -145,12 +127,9 @@ class DatabaseFeatures(BaseDatabaseFeatures):
         "schema.tests.SchemaTests.test_remove_unique_together_does_not_remove_meta_constraints",
         "schema.tests.SchemaTests.test_unique_together",
         # ManyToManyField
-        "schema.tests.SchemaTests.test_m2m",
         "schema.tests.SchemaTests.test_m2m_create",
         "schema.tests.SchemaTests.test_m2m_create_custom",
         "schema.tests.SchemaTests.test_m2m_create_inherited",
-        "schema.tests.SchemaTests.test_m2m_custom",
-        "schema.tests.SchemaTests.test_m2m_inherited",
         "schema.tests.SchemaTests.test_m2m_rename_field_in_target_model",
         "schema.tests.SchemaTests.test_m2m_repoint",
         "schema.tests.SchemaTests.test_m2m_repoint_custom",

django_mongodb/schema.py

@@ -9,13 +9,27 @@ def delete_model(self, model):
         self.connection.database[model._meta.db_table].drop()
 
     def add_field(self, model, field):
-        pass
+        # Create implicit M2M tables.
+        if field.many_to_many and field.remote_field.through._meta.auto_created:
+            self.create_model(field.remote_field.through)
+            return
+        # Set default value on existing documents.
+        if column := field.column:
+            self.connection.database[model._meta.db_table].update_many(
+                {}, [{"$set": {column: self.effective_default(field)}}]
+            )
 
     def alter_field(self, model, old_field, new_field, strict=False):
         pass
 
     def remove_field(self, model, field):
-        pass
+        # Remove implicit M2M tables.
+        if field.many_to_many and field.remote_field.through._meta.auto_created:
+            self.delete_model(field.remote_field.through)
+            return
+        # Unset field on existing documents.
+        if column := field.column:
+            self.connection.database[model._meta.db_table].update_many({}, {"$unset": {column: ""}})
 
     def alter_index_together(self, model, old_index_together, new_index_together):
         pass