INTPYTHON-348 add support for QuerySet.raw_mql()

- Subclass RawModelIterable

Author: aclark4life

django_mongodb/query.py

@@ -7,7 +7,7 @@
 from django.db.models.expressions import Case, Col, When
 from django.db.models.functions import Mod
 from django.db.models.lookups import Exact
-from django.db.models.query import RawQuerySet
+from django.db.models.query import RawQuerySet, RawModelIterable
 from django.db.models.sql.constants import INNER
 from django.db.models.sql.datastructures import Join
 from django.db.models.sql.query import RawQuery
@@ -308,6 +308,7 @@ def register_nodes():
 
 
 class MongoQuerySet(QuerySet):
+
     def raw_mql(self, raw_query, params=(), translations=None, using=None):
         return MongoRawQuerySet(
             raw_query,
@@ -360,6 +361,7 @@ def __init__(
         using=None,
         hints=None,
     ):
+        self._prefetch_related_lookups = None
         self.model = model
         self._db = using
         self._hints = hints or {}
@@ -382,3 +384,36 @@ def resolve_model_init_order(self):
         model_init_order = [self.columns.index(converter(f)) for f in model_init_fields]
         model_init_names = [f.attname for f in model_init_fields]
         return model_init_names, model_init_order, annotation_fields
+
+    def iterator(self):
+        yield from MongoRawModelIterable(self)
+
+
+class MongoRawModelIterable(RawModelIterable):
+
+    def __iter__(self):
+        db = self.queryset.db
+        query = self.queryset.query
+        connection = connections[db]
+        compiler = connection.ops.compiler("SQLCompiler")(query, connection, db)
+        query_iterator = iter(query)
+        try:
+            (
+                model_init_names,
+                model_init_pos,
+                annotation_fields,
+            ) = self.queryset.resolve_model_init_order()
+            model_cls = self.queryset.model
+            fields = [self.queryset.model_fields.get(c) for c in self.queryset.columns]
+            converters = compiler.get_converters(
+                [f.get_col(f.model._meta.db_table) if f else None for f in fields]
+            )
+            if converters:
+                query_iterator = compiler.apply_converters(query_iterator, converters)
+            for values in query_iterator:
+                model_init_values = values.values()
+                instance = model_cls.from_db(db, model_init_names, model_init_values)
+                yield instance
+        finally:
+            if hasattr(query, "cursor") and query.cursor:
+                query.cursor.close()