Commit 142c2721 authored by Claude Paroz's avatar Claude Paroz
Browse files

Fixed #22565 -- Prevented pgettext_lazy crash with bytestring input

Thanks ygbo for the report.
parent d1799233
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -314,7 +314,8 @@ def pgettext(context, message):
    result = ugettext(msg_with_ctxt)
    if CONTEXT_SEPARATOR in result:
        # Translation not found
        result = message
        # force unicode, because lazy version expects unicode
        result = force_text(message)
    return result


+3 −0
Original line number Diff line number Diff line
@@ -11,3 +11,6 @@ Bugfixes

* Made the ``year_lookup_bounds_for_datetime_field`` Oracle backend method
  Python 3 compatible (`#22551 <http://code.djangoproject.com/ticket/22551>`_).

* Fixed ``pgettext_lazy`` crash when receiving bytestring content on Python 2
  (`#22565 <http://code.djangoproject.com/ticket/22565>`_).
+16 −4
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ from importlib import import_module
import os
import pickle
from threading import local
from unittest import skipUnless

from django.conf import settings
from django.template import Template, Context
@@ -30,7 +31,7 @@ from django.utils.translation import (activate, deactivate,
    ugettext, ugettext_lazy,
    ngettext_lazy,
    ungettext_lazy,
    pgettext,
    pgettext, pgettext_lazy,
    npgettext, npgettext_lazy,
    check_for_language,
    string_concat, LANGUAGE_SESSION_KEY)
@@ -94,9 +95,20 @@ class TranslationTests(TestCase):
        s4 = ugettext_lazy('Some other string')
        self.assertEqual(False, s == s4)

        if six.PY2:
            # On Python 2, gettext_lazy should not transform a bytestring to unicode
    @skipUnless(six.PY2, "No more bytestring translations on PY3")
    def test_lazy_and_bytestrings(self):
        # On Python 2, (n)gettext_lazy should not transform a bytestring to unicode
        self.assertEqual(gettext_lazy(b"test").upper(), b"TEST")
        self.assertEqual((ngettext_lazy(b"%d test", b"%d tests") % 1).upper(), b"1 TEST")

        # Other versions of lazy functions always return unicode
        self.assertEqual(ugettext_lazy(b"test").upper(), "TEST")
        self.assertEqual((ungettext_lazy(b"%d test", b"%d tests") % 1).upper(), "1 TEST")
        self.assertEqual(pgettext_lazy(b"context", b"test").upper(), "TEST")
        self.assertEqual(
            (npgettext_lazy(b"context", b"%d test", b"%d tests") % 1).upper(),
            "1 TEST"
        )

    def test_lazy_pickle(self):
        s1 = ugettext_lazy("test")