Fixed #24836 -- Made force_text() resolve lazy objects.

This commit is contained in:
Tim Graham 2015-05-26 16:46:13 -04:00
parent 10945ebeb8
commit 70be31bba7
4 changed files with 25 additions and 2 deletions

View File

@ -77,12 +77,12 @@ def force_text(s, encoding='utf-8', strings_only=False, errors='strict'):
If strings_only is True, don't convert (some) non-string-like objects. If strings_only is True, don't convert (some) non-string-like objects.
""" """
# Handle the common case first for performance reasons. # Handle the common case first for performance reasons.
if isinstance(s, six.text_type): if issubclass(type(s), six.text_type):
return s return s
if strings_only and is_protected_type(s): if strings_only and is_protected_type(s):
return s return s
try: try:
if not isinstance(s, six.string_types): if not issubclass(type(s), six.string_types):
if six.PY3: if six.PY3:
if isinstance(s, bytes): if isinstance(s, bytes):
s = six.text_type(s, encoding, errors) s = six.text_type(s, encoding, errors)

View File

@ -19,3 +19,6 @@ Bugfixes
``Count()`` (:ticket:`24835`). ``Count()`` (:ticket:`24835`).
* Corrected ``HStoreField.has_changed()`` (:ticket:`24844`). * Corrected ``HStoreField.has_changed()`` (:ticket:`24844`).
* Reverted an optimization to the CSRF template context processor which caused
a regression (:ticket:`24836`).

View File

@ -0,0 +1,15 @@
import json
from django.http import HttpRequest
from django.template.context_processors import csrf
from django.test import SimpleTestCase
from django.utils.encoding import force_text
class TestContextProcessor(SimpleTestCase):
def test_force_text_on_token(self):
request = HttpRequest()
request.META['CSRF_COOKIE'] = 'test-token'
token = csrf(request).get('csrf_token')
self.assertEqual(json.dumps(force_text(token)), '"test-token"')

View File

@ -9,6 +9,7 @@ from django.utils.encoding import (
escape_uri_path, filepath_to_uri, force_bytes, force_text, iri_to_uri, escape_uri_path, filepath_to_uri, force_bytes, force_text, iri_to_uri,
smart_text, uri_to_iri, smart_text, uri_to_iri,
) )
from django.utils.functional import SimpleLazyObject
from django.utils.http import urlquote_plus from django.utils.http import urlquote_plus
@ -28,6 +29,10 @@ class TestEncodingUtils(unittest.TestCase):
exception = TypeError if six.PY3 else UnicodeError exception = TypeError if six.PY3 else UnicodeError
self.assertRaises(exception, force_text, MyString()) self.assertRaises(exception, force_text, MyString())
def test_force_text_lazy(self):
s = SimpleLazyObject(lambda: 'x')
self.assertTrue(issubclass(type(force_text(s)), six.text_type))
def test_force_bytes_exception(self): def test_force_bytes_exception(self):
""" """
Test that force_bytes knows how to convert to bytes an exception Test that force_bytes knows how to convert to bytes an exception