Fixed #24829 -- Allowed use of TemplateResponse in view error handlers.

This commit is contained in:
ana-balica 2015-06-15 15:55:55 +03:00 committed by Tim Graham
parent 64a4211aa8
commit 2f615b10e6
4 changed files with 44 additions and 0 deletions

View File

@ -113,6 +113,9 @@ class BaseHandler(object):
urlconf = settings.ROOT_URLCONF
urlresolvers.set_urlconf(urlconf)
resolver = urlresolvers.RegexURLResolver(r'^/', urlconf)
# Use a flag to check if the response was rendered to prevent
# multiple renderings or to force rendering if necessary.
response_is_rendered = False
try:
response = None
# Apply request middleware
@ -174,6 +177,7 @@ class BaseHandler(object):
"HttpResponse object. It returned None instead."
% (middleware_method.__self__.__class__.__name__))
response = response.render()
response_is_rendered = True
except http.Http404 as exc:
logger.warning('Not Found: %s', request.path,
@ -246,6 +250,11 @@ class BaseHandler(object):
response._closable_objects.append(request)
# If the exception handler returns a TemplateResponse that has not
# been rendered, force it to be rendered.
if not response_is_rendered and callable(getattr(response, 'render', None)):
response = response.render()
return response
def handle_uncaught_exception(self, request, resolver, exc_info):

View File

@ -385,6 +385,10 @@ Requests and Responses
* The default 40x error views now accept a second positional parameter, the
exception that triggered the view.
* View error handlers now support
:class:`~django.template.response.TemplateResponse`, commonly used with
class-based views.
Tests
^^^^^

View File

@ -0,0 +1 @@
Error handler content

View File

@ -0,0 +1,30 @@
from django.conf.urls import url
from django.core.exceptions import PermissionDenied
from django.template.response import TemplateResponse
from django.test import SimpleTestCase, override_settings
def template_response_error_handler(request, exception=None):
return TemplateResponse(request, 'test_handler.html', status=403)
def permission_denied_view(request):
raise PermissionDenied
urlpatterns = [
url(r'^$', permission_denied_view),
]
handler403 = template_response_error_handler
@override_settings(ROOT_URLCONF='handlers.tests_custom_error_handlers')
class CustomErrorHandlerTests(SimpleTestCase):
def test_handler_renders_template_response(self):
"""
BaseHandler should render TemplateResponse if necessary.
"""
response = self.client.get('/')
self.assertContains(response, 'Error handler content', status_code=403)