Merge pull request #34 from python-babel/akx-one-ten

Akx one ten

Author: graingert

.gitignore

@@ -5,3 +5,4 @@ dist
 *.pyc
 django_babel.egg-info
 .tox
+htmlcov

.travis.yml

@@ -2,26 +2,17 @@ language: python
 python: 3.4
 sudo: false
 env:
-  - TOX_ENV=py27-django15
-  - TOX_ENV=py27-django16
-  - TOX_ENV=py27-django17
+  - TOX_ENV=docs
+  - TOX_ENV=lint
   - TOX_ENV=py27-django18
   - TOX_ENV=py27-django19
+  - TOX_ENV=py27-django110
   - TOX_ENV=py27-djangomaster
-  - TOX_ENV=py34-django15
-  - TOX_ENV=py34-django16
-  - TOX_ENV=py34-django17
+  - TOX_ENV=py33-django18
   - TOX_ENV=py34-django18
   - TOX_ENV=py34-django19
+  - TOX_ENV=py34-django110
   - TOX_ENV=py34-djangomaster
-  - TOX_ENV=py33-django15
-  - TOX_ENV=py33-django16
-  - TOX_ENV=py33-django17
-  - TOX_ENV=py33-django18
-  - TOX_ENV=py26-django15
-  - TOX_ENV=py26-django16
-  - TOX_ENV=lint
-  - TOX_ENV=docs
 install:
 - pip install tox
 script:

django_babel/extract.py

@@ -1,31 +1,11 @@
 # -*- coding: utf-8 -*-
-try:
-    from django.template import Lexer, TOKEN_TEXT, TOKEN_VAR, TOKEN_BLOCK
-except ImportError:
-    # Django 1.8 moved most stuff to .base
-    from django.template.base import Lexer, TOKEN_TEXT, TOKEN_VAR, TOKEN_BLOCK
-
-try:
-    from django.utils.translation import trim_whitespace as trim_django
-except ImportError:
-    trim_django = False
-
+from django.template.base import Lexer, TOKEN_TEXT, TOKEN_VAR, TOKEN_BLOCK
+from django.utils.translation import trim_whitespace
 from django.utils.encoding import smart_text
 from django.utils.translation.trans_real import (
     inline_re, block_re, endblock_re, plural_re, constant_re)
 
 
-def trim_whitespace(string):
-    """Trim whitespace.
-
-    This is only supported in Django>=1.7. This method help in cases of older
-    Django versions.
-    """
-    if trim_django:
-        return trim_django(string)
-    return string
-
-
 def join_tokens(tokens, trim=False):
     message = ''.join(tokens)
     if trim:

django_babel/management/commands/babel.py

@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
 import os
 from distutils.dist import Distribution
-from optparse import make_option
 from subprocess import call
 
 from django.core.management.base import LabelCommand, CommandError
@@ -15,23 +14,23 @@ class Command(LabelCommand):
 
     args = '[makemessages] [compilemessages]'
 
-    option_list = LabelCommand.option_list + (
-        make_option(
-            '--locale', '-l', default=None, dest='locale', action='append',
+    def add_arguments(self, parser):
+        super(Command, self).add_arguments(parser)
+        parser.add_argument(
+            '--locale', '-l', default=[], dest='locale', action='append',
             help=(
                 'Creates or updates the message files for the given locale(s)'
                 ' (e.g pt_BR). Can be used multiple times.'
             ),
-        ),
-        make_option(
+        )
+        parser.add_argument(
             '--domain', '-d', default='django', dest='domain',
             help='The domain of the message files (default: "django").',
         ),
-        make_option(
+        parser.add_argument(
             '--mapping-file', '-F', default=None, dest='mapping_file',
             help='Mapping file',
         )
-    )
 
     def handle_label(self, command, **options):
         if command not in ('makemessages', 'compilemessages'):

django_babel/middleware.py

@@ -2,10 +2,14 @@
 
 from babel import Locale, UnknownLocaleError
 from django.utils.translation import get_language
+from threading import local
+
 try:
-    from threading import local
+    from django.utils.deprecation import MiddlewareMixin
 except ImportError:
-    from django.utils._threading_local import local
+    # Not required for Django <= 1.9, see:
+    # https://docs.djangoproject.com/en/1.10/topics/http/middleware/#upgrading-pre-django-1-10-style-middleware
+    MiddlewareMixin = object
 
 
 __all__ = ['get_current_locale', 'LocaleMiddleware']
@@ -22,7 +26,7 @@ def get_current_locale():
     return getattr(_thread_locals, 'locale', None)
 
 
-class LocaleMiddleware(object):
+class LocaleMiddleware(MiddlewareMixin):
 
     """Simple Django middleware that makes available a Babel `Locale` object
     via the `request.locale` attribute.

setup.py

@@ -24,7 +24,7 @@ def read(*parts):
     url='https://github.com/python-babel/django-babel/',
     packages=find_packages(exclude=('tests',)),
     install_requires=[
-        'django>=1.4,<1.10',
+        'django>=1.4,<1.11',
         'babel>=1.3',
     ],
     classifiers=[

tests/babel.cfg

@@ -0,0 +1 @@
+[django: **/templates/**.*]

tests/conftest.py

@@ -0,0 +1,7 @@
+from django.conf import settings
+
+from testproject import settings as testproject_settings
+
+
+def pytest_configure():
+    settings.configure(**vars(testproject_settings))

tests/test_command.py

@@ -0,0 +1,37 @@
+import os
+
+import pkg_resources
+from django.core.management import call_command
+
+TEST_LOCALE_DIR = pkg_resources.resource_filename(
+    'testproject', 'locale'
+)
+
+
+def test_babel_compilemessages():
+    call_command(
+        'babel',
+        'compilemessages',
+        '-l', 'fi',
+    )
+    # Assert that the .mo file was created by attempting to delete it.
+    os.unlink(
+        os.path.join(TEST_LOCALE_DIR, 'fi', 'LC_MESSAGES', 'django.mo')
+    )
+
+
+def test_babel_makemessages():
+    call_command(
+        'babel',
+        'makemessages',
+        '-l', 'en',
+        '-F', pkg_resources.resource_filename(__name__, 'babel.cfg'),
+    )
+    # See that the expected files get populated with the discovered message
+    for path in [
+        os.path.join(TEST_LOCALE_DIR, 'django.pot'),
+        os.path.join(TEST_LOCALE_DIR, 'en', 'LC_MESSAGES', 'django.po'),
+    ]:
+        with open(path) as infp:
+            assert '"This could be translated."' in infp.read()
+        os.unlink(path)  # clean up

tests/test_render.py

@@ -0,0 +1,26 @@
+# -- encoding: utf-8 --
+from __future__ import unicode_literals
+
+import pytest
+
+
+@pytest.mark.parametrize('locale', ('en', 'fi', 'sv', 'pt-BR'))
+def test_babel_render(client, locale):
+    """
+    Test the middleware and the rendery bits.
+    """
+    response = client.get('/', HTTP_ACCEPT_LANGUAGE=locale)
+    # "Parse" the key-value format
+    lines = response.content.decode('utf-8').strip().splitlines()
+    content = dict(kv.split('=', 1) for kv in lines)
+    # See that we're rendering in the locale we expect
+    assert content['language_code'] == locale.lower()
+    # check that we could access `babel.Locale.language_name`
+    assert content['language_name'] == {
+        'en': 'English',
+        'fi': 'suomi',
+        'sv': 'svenska',
+        'pt-BR': 'português',
+    }[locale]
+    # The rest are not really tested (aside from smoke tests) further;
+    # the Babel test suite has taken care of that.