encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-02-02 20:54:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

set the action when override_method regardless of its None-ness

Browse Source
This commit is contained in:
Alex Rothberg 2015-05-13 22:53:32 -04:00
parent ea1145c5aa
commit a804d0bd75
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_framework/request.py
View File

@ -48,16 +48,14 @@ 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)
if self.action is not None: # For viewsets we also set the `.action` attribute.
# For viewsets we also set the `.action` attribute. action_map = getattr(self.view, 'action_map', {})
action_map = getattr(self.view, 'action_map', {}) self.view.action = action_map.get(self.method.lower())
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.request = self.request self.view.request = self.request
if self.action is not None: self.view.action = self.action
self.view.action = self.action
class Empty(object): class Empty(object):