encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-06-29 09:53:15 +03:00
Code Issues Packages Projects Releases Wiki Activity

Only set .action attribute in override_method if it already existed on the view

Browse Source
This commit is contained in:
Tom Christie 2014-08-18 20:56:17 +01:00
parent 01986fc80e
commit 97d8f037cc
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rest_framework/request.py
View File

@ -46,13 +46,16 @@ class override_method(object):
def __enter__(self): def __enter__(self):
self.view.request = clone_request(self.request, self.method) self.view.request = clone_request(self.request, self.method)
action_map = getattr(self.view, 'action_map', {}) if self.action is not None:
self.view.action = action_map.get(self.method.lower()) # For viewsets we also set the `.action` attribute.
action_map = getattr(self.view, 'action_map', {})
self.view.action = action_map.get(self.method.lower())
return self.view.request return self.view.request
def __exit__(self, *args, **kwarg): def __exit__(self, *args, **kwarg):
self.view.action = self.action
self.view.request = self.request self.view.request = self.request
if self.action is not None:
self.view.action = self.action
class Empty(object): class Empty(object):