Refactor: remove singledispatchmethod

Author: WaVEV

django_mongodb_backend/schema.py

@@ -1,5 +1,3 @@
-from functools import singledispatchmethod
-
 from django.db.backends.base.schema import BaseDatabaseSchemaEditor
 from django.db.models import Index, UniqueConstraint
 from pymongo.operations import IndexModel, SearchIndexModel
@@ -261,18 +259,13 @@ def alter_unique_together(
                 model, constraint, parent_model=parent_model, column_prefix=column_prefix
             )
 
-    @singledispatchmethod
     def _add_index(self, index, model):
+        if isinstance(index, SearchIndexModel):
+            return self.get_collection(model._meta.db_table).create_search_index(index)
+        if isinstance(index, IndexModel):
+            return self.get_collection(model._meta.db_table).create_indexes([index])
         raise ValueError(f"{type(index)} isn't a supported index type")
 
-    @_add_index.register
-    def _(self, index: IndexModel, model):
-        return self.get_collection(model._meta.db_table).create_indexes([index])
-
-    @_add_index.register
-    def _(self, index: SearchIndexModel, model):
-        return self.get_collection(model._meta.db_table).create_search_index(index)
-
     @ignore_embedded_models
     def add_index(
         self, model, index, *, field=None, unique=False, column_prefix="", parent_model=None
@@ -296,18 +289,13 @@ def _add_field_index(self, model, field, *, column_prefix=""):
         index.name = self._create_index_name(model._meta.db_table, [column_prefix + field.column])
         self.add_index(model, index, field=field, column_prefix=column_prefix)
 
-    @singledispatchmethod
     def _remove_index(self, index, model):
+        if isinstance(index, AtlasSearchIndex | AtlasVectorSearchIndex):
+            return self.get_collection(model._meta.db_table).drop_search_index(index.name)
+        if isinstance(index, Index):
+            return self.get_collection(model._meta.db_table).drop_index(index.name)
         raise ValueError(f"{type(index)} isn't a supported index type")
 
-    @_remove_index.register
-    def _(self, index: Index, model):
-        return self.get_collection(model._meta.db_table).drop_index(index.name)
-
-    @_remove_index.register
-    def _(self, index: AtlasSearchIndex | AtlasVectorSearchIndex, model):
-        return self.get_collection(model._meta.db_table).drop_search_index(index.name)
-
     @ignore_embedded_models
     def remove_index(self, model, index):
         if index.contains_expressions: