Fixes request.QUERY_PARAMS raising AttributeError since RestFramework…

… 3.2.
milashensky · Dec 1, 2015 · 2d25da2 · 2d25da2
1 parent 62fab53
commit 2d25da2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/django_elasticsearch/contrib/restframework/base.py b/django_elasticsearch/contrib/restframework/base.py
@@ -12,8 +12,14 @@ class AutoCompletionMixin(ListModelMixin):
 
     @list_route()
     def autocomplete(self, request, **kwargs):
-        field_name = request.QUERY_PARAMS.get('f', None)
-        query = request.QUERY_PARAMS.get('q', '')
+        try:
+            qp = request.query_params
+        except AttributeError:
+            # restframework 2
+            qp = request.QUERY_PARAMS
+
+        field_name = qp.get('f', None)
+        query = qp.get('q', '')
 
         try:
             data = self.model.es.complete(field_name, query)