Add swagger_fake_view marker to help detect fake views in get_serializer

Cleaner fix for #154
This commit is contained in:
Cristi Vîjdea
2018-06-29 17:46:05 +03:00
parent e0aec3ff45
commit 696ec3a94a
3 changed files with 17 additions and 1 deletions
+6 -1
View File
@@ -33,7 +33,12 @@ class NestedTodoView(RetrieveAPIView):
class TodoTreeView(viewsets.ReadOnlyModelViewSet):
queryset = TodoTree.objects.all()
serializer_class = TodoTreeSerializer
def get_serializer_class(self):
if getattr(self, 'swagger_fake_view', False):
return TodoTreeSerializer
raise NotImplementedError("must not call this")
class TodoRecursiveView(viewsets.ModelViewSet):