forked from Github-Mirrors/canaille
70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
import gettext
|
|
|
|
import pytz
|
|
from babel.dates import LOCALTZ
|
|
from flask import current_app
|
|
from flask import g
|
|
from flask import request
|
|
from flask_babel import Babel
|
|
from flask_babel import get_locale
|
|
|
|
DEFAULT_LANGUAGE_CODE = "en"
|
|
|
|
babel = Babel()
|
|
|
|
|
|
def setup_i18n(app):
|
|
babel.init_app(
|
|
app, locale_selector=locale_selector, timezone_selector=timezone_selector
|
|
)
|
|
|
|
@app.before_request
|
|
def before_request():
|
|
g.available_language_codes = available_language_codes()
|
|
|
|
@app.context_processor
|
|
def global_processor():
|
|
return {
|
|
"locale": get_locale(),
|
|
}
|
|
|
|
|
|
def locale_selector():
|
|
from .flask import current_user
|
|
|
|
user = current_user()
|
|
available_language_codes = getattr(g, "available_language_codes", [])
|
|
if user is not None and user.preferred_language in available_language_codes:
|
|
return user.preferred_language
|
|
|
|
if current_app.config.get("LANGUAGE"):
|
|
return current_app.config.get("LANGUAGE")
|
|
|
|
return request.accept_languages.best_match(available_language_codes)
|
|
|
|
|
|
def timezone_selector():
|
|
try:
|
|
return pytz.timezone(current_app.config.get("TIMEZONE"))
|
|
except pytz.exceptions.UnknownTimeZoneError:
|
|
return LOCALTZ
|
|
|
|
|
|
def native_language_name_from_code(code):
|
|
try:
|
|
import pycountry
|
|
except ImportError: # pragma: no cover
|
|
return code
|
|
|
|
language = pycountry.languages.get(alpha_2=code[:2])
|
|
if code == DEFAULT_LANGUAGE_CODE:
|
|
return language.name
|
|
|
|
translation = gettext.translation(
|
|
"iso639-3", pycountry.LOCALES_DIR, languages=[code]
|
|
)
|
|
return translation.gettext(language.name)
|
|
|
|
|
|
def available_language_codes():
|
|
return [str(translation) for translation in babel.list_translations()]
|