Edits

Author: WaVEV

django_mongodb_backend/indexes.py

@@ -40,28 +40,6 @@ def builtin_lookup_idx(self, compiler, connection):
     return {lhs_mql: {operator: value}}
 
 
-def where_node_idx(self, compiler, connection):
-    if self.connector == AND:
-        operator = "$and"
-    elif self.connector == XOR:
-        raise NotSupportedError("MongoDB does not support the '^' operator lookup in indexes.")
-    else:
-        operator = "$or"
-    if self.negated:
-        raise NotSupportedError("MongoDB does not support the '~' operator in indexes.")
-    children_mql = []
-    for child in self.children:
-        mql = child.as_mql_idx(compiler, connection)
-        children_mql.append(mql)
-    if len(children_mql) == 1:
-        mql = children_mql[0]
-    elif len(children_mql) > 1:
-        mql = {operator: children_mql}
-    else:
-        mql = {}
-    return mql
-
-
 def get_pymongo_index_model(self, model, schema_editor, field=None, unique=False, column_prefix=""):
     """Return a pymongo IndexModel for this Django Index."""
     if self.contains_expressions:
@@ -101,6 +79,28 @@ def get_pymongo_index_model(self, model, schema_editor, field=None, unique=False
     return IndexModel(index_orders, name=self.name, **kwargs)
 
 
+def where_node_idx(self, compiler, connection):
+    if self.connector == AND:
+        operator = "$and"
+    elif self.connector == XOR:
+        raise NotSupportedError("MongoDB does not support the '^' operator lookup in indexes.")
+    else:
+        operator = "$or"
+    if self.negated:
+        raise NotSupportedError("MongoDB does not support the '~' operator in indexes.")
+    children_mql = []
+    for child in self.children:
+        mql = child.as_mql_idx(compiler, connection)
+        children_mql.append(mql)
+    if len(children_mql) == 1:
+        mql = children_mql[0]
+    elif len(children_mql) > 1:
+        mql = {operator: children_mql}
+    else:
+        mql = {}
+    return mql
+
+
 def register_indexes():
     BuiltinLookup.as_mql_idx = builtin_lookup_idx
     Index._get_condition_mql = _get_condition_mql

django_mongodb_backend/schema.py

@@ -261,7 +261,7 @@ def add_index(
         self, model, index, *, field=None, unique=False, column_prefix="", parent_model=None
     ):
         idx = index.get_pymongo_index_model(
-            model, self, field=field, unique=unique, column_prefix=column_prefix
+            model, schema_editor=self, field=field, unique=unique, column_prefix=column_prefix
         )
         if idx:
             model = parent_model or model