locale.py 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. "This is the locale selecting middleware that will look at accept headers"
  2. from django.conf import settings
  3. from django.core.urlresolvers import (is_valid_path, get_resolver,
  4. LocaleRegexURLResolver)
  5. from django.http import HttpResponseRedirect
  6. from django.utils.cache import patch_vary_headers
  7. from django.utils import translation
  8. class LocaleMiddleware(object):
  9. """
  10. This is a very simple middleware that parses a request
  11. and decides what translation object to install in the current
  12. thread context. This allows pages to be dynamically
  13. translated to the language the user desires (if the language
  14. is available, of course).
  15. """
  16. response_redirect_class = HttpResponseRedirect
  17. def __init__(self):
  18. self._is_language_prefix_patterns_used = False
  19. for url_pattern in get_resolver(None).url_patterns:
  20. if isinstance(url_pattern, LocaleRegexURLResolver):
  21. self._is_language_prefix_patterns_used = True
  22. break
  23. def process_request(self, request):
  24. check_path = self.is_language_prefix_patterns_used()
  25. language = translation.get_language_from_request(
  26. request, check_path=check_path)
  27. translation.activate(language)
  28. request.LANGUAGE_CODE = translation.get_language()
  29. def process_response(self, request, response):
  30. language = translation.get_language()
  31. language_from_path = translation.get_language_from_path(request.path_info)
  32. if (response.status_code == 404 and not language_from_path
  33. and self.is_language_prefix_patterns_used()):
  34. urlconf = getattr(request, 'urlconf', None)
  35. language_path = '/%s%s' % (language, request.path_info)
  36. path_valid = is_valid_path(language_path, urlconf)
  37. if (not path_valid and settings.APPEND_SLASH
  38. and not language_path.endswith('/')):
  39. path_valid = is_valid_path("%s/" % language_path, urlconf)
  40. if path_valid:
  41. language_url = "%s://%s/%s%s" % (
  42. request.scheme, request.get_host(), language,
  43. request.get_full_path())
  44. return self.response_redirect_class(language_url)
  45. if not (self.is_language_prefix_patterns_used()
  46. and language_from_path):
  47. patch_vary_headers(response, ('Accept-Language',))
  48. if 'Content-Language' not in response:
  49. response['Content-Language'] = language
  50. return response
  51. def is_language_prefix_patterns_used(self):
  52. """
  53. Returns `True` if the `LocaleRegexURLResolver` is used
  54. at root level of the urlpatterns, else it returns `False`.
  55. """
  56. return self._is_language_prefix_patterns_used