-
Notifications
You must be signed in to change notification settings - Fork 79
/
wsgi.py
110 lines (91 loc) · 3.97 KB
/
wsgi.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from webob import Request
import re
import settings
def get_language_map():
"""
Return a complete dict of language -> URL mappings, including the canonical
short locale maps (e.g. es -> es-ES and en -> en-US).
:return: dict
"""
LUM = {i.lower(): i for i in settings.PROD_LANGUAGES}
langs = dict(list(LUM.items()) + list(settings.CANONICAL_LOCALES.items()))
# Add missing short locales to the list. This will automatically map
# en to en-GB (not en-US), es to es-AR (not es-ES), etc. in alphabetical
# order. To override this behavior, explicitly define a preferred locale
# map with the CANONICAL_LOCALES setting.
langs.update((k.split('-')[0], v) for k, v in LUM.items() if
k.split('-')[0] not in langs)
return langs
def parse_accept_lang_header(lang_string):
"""
Parse the lang_string, which is the body of an HTTP Accept-Language
header, and return a list of (lang, q-value), ordered by 'q' values.
Return an empty list if there are any format errors in lang_string.
"""
# Format of Accept-Language header values. From RFC 2616, section 14.4 and 3.9
# and RFC 3066, section 2.1
accept_language_re = re.compile(r'''
([A-Za-z]{1,8}(?:-[A-Za-z0-9]{1,8})*|\*) # "en", "en-au", "x-y-z", "es-419", "*"
(?:\s*;\s*q=(0(?:\.\d{,3})?|1(?:\.0{,3})?))? # Optional "q=1.00", "q=0.8"
(?:\s*,\s*|$) # Multiple accepts per header.
''', re.VERBOSE)
result = []
pieces = accept_language_re.split(lang_string.lower())
if pieces[-1]:
return []
for i in range(0, len(pieces) - 1, 3):
first, lang, priority = pieces[i:i + 3]
if first:
return []
if priority:
priority = float(priority)
else:
priority = 1.0
result.append((lang, priority))
result.sort(key=lambda k: k[1], reverse=True)
return result
def get_best_language(accept_lang):
"""Given an Accept-Language header, return the best-matching language."""
ranked = parse_accept_lang_header(accept_lang)
FULL_LANGUAGE_MAP = get_language_map()
for lang, _ in ranked:
lang = lang.lower()
if lang in FULL_LANGUAGE_MAP:
return FULL_LANGUAGE_MAP[lang]
pre = lang.split('-')[0]
if pre in FULL_LANGUAGE_MAP:
return FULL_LANGUAGE_MAP[pre]
return settings.LANGUAGE_CODE
def application(environ, start_response):
"""Main wsgi application to redirect incoming users to the correct locale."""
req = Request(environ)
# Are we updates.thunderbird.net?
is_utn = any(['updates.' in req.host_url, 'updates-stage.' in req.host_url])
if (not is_utn
and 'thunderbird' in req.path
and not any(s in req.path for s in settings.ALWAYS_LOCALIZE)):
# Release notes, system requirements, and 'all' builds pages are only available in English.
language_code = 'en-US'
else:
language_code = get_best_language(environ.get('HTTP_ACCEPT_LANGUAGE', 'en-US'))
location = "{0}/{1}{2}".format(req.host_url, language_code, req.path_qs)
start_response('302 Found', [
('Content-type', 'text/html; charset=utf-8'),
('Cache-Control', 'private, s-maxage=0, max-age=604800'),
('Vary', 'Accept-Language'),
('Location', location)
])
return ''
if __name__ == '__main__':
# This runs when script is started directly from commandline.
try:
# Create a simple WSGI server and run the application.
from wsgiref import simple_server
print("Running test application - point your browser at http://localhost:8000/ ...")
httpd = simple_server.WSGIServer(('', 8000), simple_server.WSGIRequestHandler)
httpd.set_app(application)
httpd.serve_forever()
except ImportError:
# wsgiref not installed, just output html to stdout
for content in application({}, lambda status, headers: None):
print(content)