Include .action attribute on viewsets when generating schemas (#4408)

tomchristie · web-flow · commit e3f8d06bafd3 · 2016-08-15T17:10:55.000+01:00
diff --git a/rest_framework/schemas.py b/rest_framework/schemas.py
@@ -79,6 +79,13 @@ def get_schema(self, request=None):
             view.kwargs = {}
             view.format_kwarg = None
 
+            actions = getattr(callback, 'actions', None)
+            if actions is not None:
+                if method == 'OPTIONS':
+                    view.action = 'metadata'
+                else:
+                    view.action = actions.get(method.lower())
+
             if request is not None:
                 view.request = clone_request(request, method)
                 try:
diff --git a/tests/test_schemas.py b/tests/test_schemas.py
@@ -49,6 +49,7 @@ def custom_list_action(self, request):
 
     def get_serializer(self, *args, **kwargs):
         assert self.request
+        assert self.action
         return super(ExampleViewSet, self).get_serializer(*args, **kwargs)
 
 
