forked from mkdocs/mkdocs
-
Notifications
You must be signed in to change notification settings - Fork 0
/
localization.py
77 lines (61 loc) · 2.53 KB
/
localization.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
import os
import logging
from jinja2.ext import Extension, InternationalizationExtension
from mkdocs.config.base import ValidationError
try:
from babel.core import Locale, UnknownLocaleError
from babel.support import Translations, NullTranslations
has_babel = True
except ImportError: # pragma: no cover
from mkdocs.utils.babel_stub import Locale, UnknownLocaleError
has_babel = False
log = logging.getLogger(__name__)
base_path = os.path.dirname(os.path.abspath(__file__))
class NoBabelExtension(InternationalizationExtension): # pragma: no cover
def __init__(self, environment):
Extension.__init__(self, environment)
environment.extend(
install_null_translations=self._install_null,
newstyle_gettext=False,
)
def parse_locale(locale):
try:
return Locale.parse(locale, sep='_')
except (ValueError, UnknownLocaleError, TypeError) as e:
raise ValidationError(f'Invalid value for locale: {str(e)}')
def install_translations(env, locale, theme_dirs):
if has_babel:
env.add_extension('jinja2.ext.i18n')
translations = _get_merged_translations(theme_dirs, 'locales', locale)
if translations is not None:
env.install_gettext_translations(translations)
else:
env.install_null_translations()
if locale.language != 'en':
log.warning(
f"No translations could be found for the locale '{locale}'. "
'Defaulting to English.'
)
else: # pragma: no cover
# no babel installed, add dummy support for trans/endtrans blocks
env.add_extension(NoBabelExtension)
env.install_null_translations()
def _get_merged_translations(theme_dirs, locales_dir, locale):
merged_translations = None
log.debug(f"Looking for translations for locale '{locale}'")
if locale.territory:
locale_str = f"{locale.language}_{locale.territory}"
else:
locale_str = locale.language
for theme_dir in reversed(theme_dirs):
dirname = os.path.join(theme_dir, locales_dir)
translations = Translations.load(dirname, [locale_str])
if type(translations) is NullTranslations:
log.debug(f"No translations found here: '{dirname}'")
continue
log.debug(f"Translations found here: '{dirname}'")
if merged_translations is None:
merged_translations = translations
else:
merged_translations.merge(translations)
return merged_translations