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 committed 2018-06-29 18:00:54 +03:00
1 parent e0aec3ff45
commit 696ec3a94a
3 files changed
+17 -1

No files matched your search

+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):