Skip to content

Commit 3b75cc0

Browse files
author
Umut Bozkurt
committed
fix isinstance() check on get_queryset()
1 parent 18b9dc5 commit 3b75cc0

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

rest_framework_mongoengine/generics.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from rest_framework import generics as drf_generics
33

44
from mongoengine.django.shortcuts import get_document_or_404
5+
from mongoengine.queryset import BaseQuerySet
56

67

78
class 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() ***

0 commit comments

Comments
 (0)