orbit-oss/flask
4
4
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added caching

Browse source
This commit is contained in:
Phil Schaf 2015-01-03 01:31:38 +01:00
parent fd8e6b26f9
commit 8c054f04b8
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
flask/app.py
View file

@ -1079,14 +1079,12 @@ class Flask(_PackageBoundObject):
return decorator return decorator
@staticmethod @staticmethod
def _ensure_exc_class(exc_class_or_code): def _get_exc_class_and_code(exc_class_or_code):
"""ensure that we register only exceptions as handler keys""" """ensure that we register only exceptions as handler keys"""
if isinstance(exc_class_or_code, integer_types): if isinstance(exc_class_or_code, integer_types):
exc_class = default_exceptions[exc_class_or_code] exc_class = default_exceptions[exc_class_or_code]
elif isinstance(exc_class_or_code, type):
exc_class = exc_class_or_code
else: else:
exc_class = type(exc_class_or_code) exc_class = exc_class_or_code
assert issubclass(exc_class, Exception) assert issubclass(exc_class, Exception)
@ -1164,7 +1162,7 @@ class Flask(_PackageBoundObject):
'Handlers can only be registered for exception classes or HTTP error codes.' 'Handlers can only be registered for exception classes or HTTP error codes.'
.format(code_or_exception)) .format(code_or_exception))
exc_class, code = self._ensure_exc_class(code_or_exception) exc_class, code = self._get_exc_class_and_code(code_or_exception)
handlers = self.error_handler_spec.setdefault(key, {}).setdefault(code, {}) handlers = self.error_handler_spec.setdefault(key, {}).setdefault(code, {})
handlers[exc_class] = f handlers[exc_class] = f
@ -1412,7 +1410,7 @@ class Flask(_PackageBoundObject):
"""Finds a registered error handler for the request’s blueprint. """Finds a registered error handler for the request’s blueprint.
If neither blueprint nor App has a suitable handler registered, returns None If neither blueprint nor App has a suitable handler registered, returns None
""" """
exc_class, code = self._ensure_exc_class(e) exc_class, code = self._get_exc_class_and_code(type(e))
def find_superclass(d): def find_superclass(d):
if not d: if not d:
@ -1422,6 +1420,7 @@ class Flask(_PackageBoundObject):
return None return None
handler = d.get(superclass) handler = d.get(superclass)
if handler is not None: if handler is not None:
d[exc_class] = handler # cache for next time exc_class is raised
return handler return handler
return None return None