Patch option to support top label indexes.

Author: WaVEV

django_mongodb_backend/lookups.py

@@ -1,4 +1,6 @@
+import django.db.models.base as base
 from django.db import NotSupportedError
+from django.db.models.constants import LOOKUP_SEP
 from django.db.models.fields.related_lookups import In, RelatedIn
 from django.db.models.lookups import (
     BuiltinLookup,
@@ -8,6 +10,7 @@
     UUIDTextMixin,
 )
 
+from .fields import EmbeddedModelField
 from .query_utils import process_lhs, process_rhs
 
 
@@ -121,6 +124,26 @@ def uuid_text_mixin(self, compiler, connection):  # noqa: ARG001
     raise NotSupportedError("Pattern lookups on UUIDField are not supported.")
 
 
+class Options(base.Options):
+    def get_field(self, field_name):
+        if LOOKUP_SEP in field_name:
+            previous = self
+            keys = field_name.split(LOOKUP_SEP)
+            path = []
+            for field in keys:
+                field = base.Options.get_field(previous, field)
+                if isinstance(field, EmbeddedModelField):
+                    previous = field.embedded_model._meta
+                else:
+                    previous = field
+                path.append(field.column)
+            column = ".".join(path)
+            embedded_column = field.clone()
+            embedded_column.column = column
+            return embedded_column
+        return super().get_field(field_name)
+
+
 def register_lookups():
     BuiltinLookup.as_mql = builtin_lookup
     FieldGetDbPrepValueIterableMixin.resolve_expression_parameter = (
@@ -131,3 +154,4 @@ def register_lookups():
     IsNull.as_mql = is_null
     PatternLookup.prep_lookup_value_mongo = pattern_lookup_prep_lookup_value
     UUIDTextMixin.as_mql = uuid_text_mixin
+    base.Options = Options

django_mongodb_backend/schema.py

@@ -1,5 +1,6 @@
 from django.db.backends.base.schema import BaseDatabaseSchemaEditor
 from django.db.models import Index, UniqueConstraint
+from django.db.models.constants import LOOKUP_SEP
 from pymongo.operations import SearchIndexModel
 
 from django_mongodb_backend.indexes import SearchIndex
@@ -184,6 +185,20 @@ def remove_field(self, model, field):
             self._remove_model_indexes(
                 field.embedded_model, parent_model=model, column_prefix=new_path
             )
+            # Remove the top level indexes.
+            # TODO: Find a workaround
+            for index in model._meta.indexes:
+                if any(
+                    field_name.startswith(f"{field.column}{LOOKUP_SEP}")
+                    for field_name in index.fields
+                ):
+                    self.remove_index(model, index)
+            for constraint in model._meta.constraints:
+                if any(
+                    field_name.startswith(f"{field.column}{LOOKUP_SEP}")
+                    for field_name in constraint.fields
+                ):
+                    self.get_collection(model._meta.db_table).drop_index(constraint.name)
 
     def _remove_model_indexes(self, model, column_prefix="", parent_model=None):
         """