File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
rest_framework_mongoengine Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change 22from rest_framework import generics as drf_generics
33
44from mongoengine .django .shortcuts import get_document_or_404
5+ from mongoengine .queryset import BaseQuerySet
56
67
78class GenericAPIView (drf_generics .GenericAPIView ):
@@ -10,6 +11,17 @@ class GenericAPIView(drf_generics.GenericAPIView):
1011 """
1112 lookup_field = 'id'
1213
14+ def get_queryset (self ):
15+ """
16+ Re evaluate queryset, fixes #63
17+ """
18+ queryset = super (GenericAPIView , self ).get_queryset ()
19+
20+ if isinstance (queryset , BaseQuerySet ):
21+ queryset = queryset .all ()
22+
23+ return queryset
24+
1325 def get_object (self ):
1426 """
1527 *** Inherited from DRF 3 GenericAPIView, swapped get_object_or_404() with get_document_or_404() ***
You can’t perform that action at this time.
0 commit comments