INTPYTHON-348 add support for QuerySet.raw_mql()

Author: aclark4life

django_mongodb/query.py

@@ -319,8 +319,6 @@ def raw_mql(self, raw_query, params=(), translations=None, using=None):
 
 
 class MongoRawQuery(RawQuery):
-
-
     def __init__(self, sql, using, model, params=()):
         self.params = params
         self.sql = sql
@@ -351,30 +349,32 @@ def __iter__(self):
 
     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)
+        # 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)
 
         collection = connection.get_collection(self.model._meta.db_table)
-	
+
         return collection.aggregate(self.sql)
 
     def get_columns(self):
         if self.cursor is None:
             self._execute_query()
         converter = connections[self.using].introspection.identifier_converter
-        self.cursor.description = ""
-        return [converter(column_meta[0]) for column_meta in self.cursor.description]
+        # self.cursor.description = ""
+        # return [converter(column_meta[0]) for column_meta in self.cursor.description]
+        return [converter(column_name) for column_name in self.model._meta.get_fields()]
 
 
 class MongoRawQuerySet(RawQuerySet):
@@ -391,6 +391,8 @@ def __init__(
         self.model = model
         self._db = using
         self._hints = hints or {}
-        self.query = query or MongoRawQuery(sql=raw_query, using=self.db, model=self.model, params=params)
+        self.query = query or MongoRawQuery(
+            sql=raw_query, using=self.db, model=self.model, params=params
+        )
         self._result_cache = None
         self.translations = translations or {}