diff options
| author | Hiroki KIYOHARA <hirokiky@gmail.com> | 2014-07-26 12:06:48 +0200 |
|---|---|---|
| committer | Hiroki KIYOHARA <hirokiky@gmail.com> | 2014-07-26 12:35:51 +0200 |
| commit | d0889863de50d65659f56f0b9ea0672a8b6482a1 (patch) | |
| tree | 91dd500ab81b9f25f115e61f81fe5876622ef893 | |
| parent | e569144910e336d2f76a3a31998ec672377d31cc (diff) | |
Fixed code to solve #23070 problem
Added a class to wrap callable in settings:
* Not to call in the debug page (#21345).
* Not to break the debug page if the callable forbidding to set attributes (#23070).
Thanks @bmispelon for giving me some advice.
| -rw-r--r-- | django/views/debug.py | 16 | ||||
| -rw-r--r-- | tests/view_tests/tests/test_debug.py | 17 |
2 files changed, 31 insertions, 2 deletions
diff --git a/django/views/debug.py b/django/views/debug.py index ac81ecaf9e..23bf16604d 100644 --- a/django/views/debug.py +++ b/django/views/debug.py @@ -33,6 +33,19 @@ def linebreak_iter(template_source): yield len(template_source) + 1 +class CallableSettingWrapper(object): + """ Object to wrap callable appearing in settings + + * Not to call in the debug page (#21345). + * Not to break the debug page if the callable forbidding to set attributes (#23070). + """ + def __init__(self, callable_setting): + self._wrapped = callable_setting + + def __repr__(self): + return repr(self._wrapped) + + def cleanse_setting(key, value): """Cleanse an individual setting key/value of sensitive content. @@ -52,7 +65,8 @@ def cleanse_setting(key, value): cleansed = value if callable(cleansed): - cleansed.do_not_call_in_templates = True + # For fixing #21345 and #23070 + cleansed = CallableSettingWrapper(cleansed) return cleansed diff --git a/tests/view_tests/tests/test_debug.py b/tests/view_tests/tests/test_debug.py index b41ce8b75c..3d51da9f94 100644 --- a/tests/view_tests/tests/test_debug.py +++ b/tests/view_tests/tests/test_debug.py @@ -20,7 +20,7 @@ from django.test import TestCase, RequestFactory, override_settings from django.test.utils import override_with_test_loader from django.utils.encoding import force_text, force_bytes from django.utils import six -from django.views.debug import ExceptionReporter +from django.views.debug import CallableSettingWrapper, ExceptionReporter from .. import BrokenException, except_args from ..views import (sensitive_view, non_sensitive_view, paranoid_view, @@ -29,6 +29,21 @@ from ..views import (sensitive_view, non_sensitive_view, paranoid_view, multivalue_dict_key_error) +class CallableSettingWrapperTests(TestCase): + """ Unittests for CallableSettingWrapper + """ + def test_repr(self): + class WrappedCallable(object): + def __repr__(self): + return "repr from the wrapped callable" + + def __call__(self): + pass + + actual = repr(CallableSettingWrapper(WrappedCallable())) + self.assertEqual(actual, "repr from the wrapped callable") + + @override_settings(DEBUG=True, TEMPLATE_DEBUG=True, ROOT_URLCONF="view_tests.urls") class DebugViewTests(TestCase): |
