From 76c5338930c5fe9855f2f4219b2ace2240e8f3b0 Mon Sep 17 00:00:00 2001 From: vkylamba Date: Wed, 27 Dec 2017 12:09:18 +0530 Subject: [PATCH] Updated to work with ViewSets --- rest_framework/schemas/inspectors.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/rest_framework/schemas/inspectors.py b/rest_framework/schemas/inspectors.py index b2a5320bd..36ca716fc 100644 --- a/rest_framework/schemas/inspectors.py +++ b/rest_framework/schemas/inspectors.py @@ -291,15 +291,17 @@ class AutoSchema(ViewInspector): request body input, as determined by the serializer class. """ view = self.view - if method not in ('PUT', 'PATCH', 'POST'): return [] - if not hasattr(view, 'get_serializer'): + if not hasattr(view, 'get_serializer') and not hasattr(view, 'serializer_class'): return [] try: - serializer = view.get_serializer() + if hasattr(view, 'get_serializer'): + serializer = view.get_serializer() + elif hasattr(view, 'serializer_class'): + serializer = view.serializer_class() except exceptions.APIException: serializer = None warnings.warn('{}.get_serializer() raised an exception during '