Fix for CSV export.
[fnpdjango.git] / fnpdjango / middleware.py
1 from . import app_settings
2
3 try:
4     # Django >= 1.10
5     from django.utils.deprecation import MiddlewareMixin
6 except ImportError:
7     # Django <= 1.9
8     MiddlewareMixin = object
9
10     from django.conf import settings
11     from django.http import Http404
12     from django.utils import translation
13
14     class URLLocaleMiddleware(MiddlewareMixin):
15         """Decides which translation to use, based on path only."""
16         def process_request(self, request):
17             language = translation.get_language_from_path(request.path_info)
18             if language:
19                 translation.activate(language)
20             request.LANGUAGE_CODE = translation.get_language()
21             if language == settings.LANGUAGE_CODE:
22                 raise Http404
23
24         def process_response(self, request, response):
25             language = translation.get_language()
26             translation.deactivate()
27             if 'Content-Language' not in response:
28                 response['Content-Language'] = language
29             return response
30 else:
31     # Django >= 1.10
32     import warnings
33     from django.middleware.locale import LocaleMiddleware
34
35     class URLLocaleMiddleware(LocaleMiddleware):
36         def __init__(self, *args, **kwargs):
37             warnings.warn(
38                 "As of Django 1.10, fnpdjango.middleware.URLLocaleMiddleware "
39                 "is deprecated in favor of "
40                 "django.middleware.locale.LocaleMiddleware.",
41                 DeprecationWarning)
42             super(URLLocaleMiddleware, self).__init__(*args, **kwargs)
43
44
45 class SetRemoteAddrFromXRealIP(MiddlewareMixin):
46     """Sets REMOTE_ADDR from the X-Real-IP header, as set by Nginx."""
47     def process_request(self, request):
48         if app_settings.REALIP:
49             try:
50                 request.META['REMOTE_ADDR'] = request.META['HTTP_X_REAL_IP']
51             except KeyError:
52                 pass
53         return None