Refactored common code in handlers/base.py and urlresolvers.py

This commit is contained in:
Anubhav Joshi 2014-06-22 09:02:14 +05:30 committed by Tim Graham
parent bcd2e6c697
commit 7f76251fb7
3 changed files with 21 additions and 42 deletions

View File

@ -70,6 +70,16 @@ class BaseHandler(object):
view = transaction.atomic(using=db.alias)(view) view = transaction.atomic(using=db.alias)(view)
return view return view
def get_exception_response(self, request, resolver, status_code):
try:
callback, param_dict = resolver.resolve_error_handler(status_code)
response = callback(request, **param_dict)
except:
signals.got_request_exception.send(sender=self.__class__, request=request)
response = self.handle_uncaught_exception(request, resolver, sys.exc_info())
return response
def get_response(self, request): def get_response(self, request):
"Returns an HttpResponse object for the given HttpRequest" "Returns an HttpResponse object for the given HttpRequest"
@ -151,12 +161,7 @@ class BaseHandler(object):
if settings.DEBUG: if settings.DEBUG:
response = debug.technical_404_response(request, e) response = debug.technical_404_response(request, e)
else: else:
try: response = self.get_exception_response(request, resolver, 404)
callback, param_dict = resolver.resolve404()
response = callback(request, **param_dict)
except:
signals.got_request_exception.send(sender=self.__class__, request=request)
response = self.handle_uncaught_exception(request, resolver, sys.exc_info())
except PermissionDenied: except PermissionDenied:
logger.warning( logger.warning(
@ -165,14 +170,7 @@ class BaseHandler(object):
'status_code': 403, 'status_code': 403,
'request': request 'request': request
}) })
try: response = self.get_exception_response(request, resolver, 403)
callback, param_dict = resolver.resolve403()
response = callback(request, **param_dict)
except:
signals.got_request_exception.send(
sender=self.__class__, request=request)
response = self.handle_uncaught_exception(request,
resolver, sys.exc_info())
except SuspiciousOperation as e: except SuspiciousOperation as e:
# The request logger receives events for any problematic request # The request logger receives events for any problematic request
@ -186,14 +184,7 @@ class BaseHandler(object):
'request': request 'request': request
}) })
try: response = self.get_exception_response(request, resolver, 400)
callback, param_dict = resolver.resolve400()
response = callback(request, **param_dict)
except:
signals.got_request_exception.send(
sender=self.__class__, request=request)
response = self.handle_uncaught_exception(request,
resolver, sys.exc_info())
except SystemExit: except SystemExit:
# Allow sys.exit() to actually exit. See tickets #1023 and #4701 # Allow sys.exit() to actually exit. See tickets #1023 and #4701
@ -251,7 +242,7 @@ class BaseHandler(object):
if resolver.urlconf_module is None: if resolver.urlconf_module is None:
six.reraise(*exc_info) six.reraise(*exc_info)
# Return an HttpResponse that displays a friendly error message. # Return an HttpResponse that displays a friendly error message.
callback, param_dict = resolver.resolve500() callback, param_dict = resolver.resolve_error_handler(500)
return callback(request, **param_dict) return callback(request, **param_dict)
def apply_response_fixes(self, request, response): def apply_response_fixes(self, request, response):

View File

@ -393,7 +393,7 @@ class RegexURLResolver(LocaleRegexProvider):
raise ImproperlyConfigured(msg.format(name=self.urlconf_name)) raise ImproperlyConfigured(msg.format(name=self.urlconf_name))
return patterns return patterns
def _resolve_special(self, view_type): def resolve_error_handler(self, view_type):
callback = getattr(self.urlconf_module, 'handler%s' % view_type, None) callback = getattr(self.urlconf_module, 'handler%s' % view_type, None)
if not callback: if not callback:
# No handler specified in file; use default # No handler specified in file; use default
@ -402,18 +402,6 @@ class RegexURLResolver(LocaleRegexProvider):
callback = getattr(urls, 'handler%s' % view_type) callback = getattr(urls, 'handler%s' % view_type)
return get_callable(callback), {} return get_callable(callback), {}
def resolve400(self):
return self._resolve_special('400')
def resolve403(self):
return self._resolve_special('403')
def resolve404(self):
return self._resolve_special('404')
def resolve500(self):
return self._resolve_special('500')
def reverse(self, lookup_view, *args, **kwargs): def reverse(self, lookup_view, *args, **kwargs):
return self._reverse_with_prefix(lookup_view, '', *args, **kwargs) return self._reverse_with_prefix(lookup_view, '', *args, **kwargs)

View File

@ -614,15 +614,15 @@ class ErrorHandlerResolutionTests(TestCase):
def test_named_handlers(self): def test_named_handlers(self):
handler = (empty_view, {}) handler = (empty_view, {})
self.assertEqual(self.resolver.resolve400(), handler) self.assertEqual(self.resolver.resolve_error_handler(400), handler)
self.assertEqual(self.resolver.resolve404(), handler) self.assertEqual(self.resolver.resolve_error_handler(404), handler)
self.assertEqual(self.resolver.resolve500(), handler) self.assertEqual(self.resolver.resolve_error_handler(500), handler)
def test_callable_handers(self): def test_callable_handers(self):
handler = (empty_view, {}) handler = (empty_view, {})
self.assertEqual(self.callable_resolver.resolve400(), handler) self.assertEqual(self.callable_resolver.resolve_error_handler(400), handler)
self.assertEqual(self.callable_resolver.resolve404(), handler) self.assertEqual(self.callable_resolver.resolve_error_handler(404), handler)
self.assertEqual(self.callable_resolver.resolve500(), handler) self.assertEqual(self.callable_resolver.resolve_error_handler(500), handler)
@override_settings(ROOT_URLCONF='urlpatterns_reverse.urls_without_full_import') @override_settings(ROOT_URLCONF='urlpatterns_reverse.urls_without_full_import')