INTPYTHON-348 add support for QuerySet.raw_mql()

Author: aclark4life

django_mongodb/manager.py

@@ -0,0 +1,7 @@
+from django.db.models.manager import BaseManager
+
+from .query import MongoQuerySet
+
+
+class MongoManager(BaseManager.from_queryset(MongoQuerySet)):
+    pass

django_mongodb/query.py

@@ -2,12 +2,14 @@
 from operator import add as add_operator
 
 from django.core.exceptions import EmptyResultSet, FullResultSet
-from django.db import DatabaseError, IntegrityError, NotSupportedError
+from django.db import DatabaseError, IntegrityError, NotSupportedError, connections
+from django.db.models import QuerySet
 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.sql.constants import INNER
 from django.db.models.sql.datastructures import Join
+from django.db.models.sql.query import RawQuery, RawQuerySet
 from django.db.models.sql.where import AND, OR, XOR, ExtraWhere, NothingNode, WhereNode
 from pymongo.errors import BulkWriteError, DuplicateKeyError, PyMongoError
 
@@ -302,3 +304,50 @@ def register_nodes():
     Join.as_mql = join
     NothingNode.as_mql = NothingNode.as_sql
     WhereNode.as_mql = where_node
+
+
+class MongoQuerySet(QuerySet):
+    def raw_mql(self, raw_query, params=(), translations=None, using=None):
+        return MongoRawQuerySet(
+            raw_query,
+            model=self.model,
+            params=params,
+            translations=translations,
+            using=using,
+        )
+
+
+class MongoRawQuery(RawQuery):
+    def _execute_query(self):
+        connection = connections[self.using]
+        params_type = self.params_type
+        adapter = connection.ops.adapt_unknown_value
+        if params_type is tuple:
+            params = tuple(adapter(val) for val in self.params)
+        elif params_type is dict:
+            params = {key: adapter(val) for key, val in self.params.items()}
+        elif params_type is None:
+            params = None
+        else:
+            raise RuntimeError("Unexpected params type: %s" % params_type)
+
+        self.cursor = connection.cursor()
+        self.cursor.execute(self.sql, params)
+
+
+class MongoRawQuerySet(RawQuerySet):
+    def __init__(
+        self,
+        raw_query,
+        model=None,
+        query=None,
+        params=(),
+        translations=None,
+        using=None,
+        hints=None,
+    ):
+        self.model = model
+        self._db = using
+        self._hints = hints or {}
+        self.query = query or MongoRawQuery(sql=raw_query, using=self.db, params=params)
+        self._result_cache = None