Commit e3f8d06b by Tom Christie Committed by GitHub

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

parent 101fd290
...@@ -79,6 +79,13 @@ class SchemaGenerator(object): ...@@ -79,6 +79,13 @@ class SchemaGenerator(object):
view.kwargs = {} view.kwargs = {}
view.format_kwarg = None 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: if request is not None:
view.request = clone_request(request, method) view.request = clone_request(request, method)
try: try:
......
...@@ -49,6 +49,7 @@ class ExampleViewSet(ModelViewSet): ...@@ -49,6 +49,7 @@ class ExampleViewSet(ModelViewSet):
def get_serializer(self, *args, **kwargs): def get_serializer(self, *args, **kwargs):
assert self.request assert self.request
assert self.action
return super(ExampleViewSet, self).get_serializer(*args, **kwargs) return super(ExampleViewSet, self).get_serializer(*args, **kwargs)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment