import os import json from functools import lru_cache from flask_babel import gettext as _ from flask_babel import lazy_gettext as _lazy from markupsafe import escape, Markup @lru_cache(maxsize=None) def get_available_locales(only_reviewed=True, sort_by_name=False): locales_dir = os.path.join(os.path.dirname(__file__), 'locales') dirs = [os.path.join(locales_dir, d) for d in os.listdir(locales_dir)] res = [{'code': 'en', 'name': 'English', 'reviewed': True}] for d in dirs: if d == 'en': continue meta_file = os.path.join(d, 'meta.json') if os.path.isdir(os.path.join(d, 'LC_MESSAGES')) and os.path.isfile(meta_file): try: with open(meta_file) as f: j = json.loads(f.read()) except Exception as e: print(e) continue if j.get('reviewed') or not only_reviewed: res.append({'code': os.path.basename(d), 'name': j.get('name', ''), 'reviewed': j.get('reviewed', False)}) if sort_by_name: res.sort(key=lambda s: s['name']) return res @lru_cache(maxsize=None) def get_available_locale_codes(only_reviewed=True): return [l['code'] for l in get_available_locales(only_reviewed=only_reviewed)] @lru_cache(maxsize=None) def get_alternate_locale_links(): tmpl = os.environ.get("LT_LOCALE_LINK_TEMPLATE") if tmpl is None: return [] locales = get_available_locale_codes() result = [] for l in locales: link = tmpl.replace("{LANG}", l) if l == 'en': link = link.replace("en.", "") result.append({ 'link': link,'lang': l }) return result # Javascript code should use _e instead of _ def gettext_escaped(text, **variables): return json.dumps(_(text, **variables)) # HTML should be escaped using _h instead of _ def gettext_html(text, **variables): # Translate text without args s = str(escape(_(text))) v = {} if variables: for k in variables: if hasattr(variables[k], 'unescape'): v[k] = variables[k].unescape() else: v[k] = Markup(variables[k]) # Variables are assumed to be already escaped and thus safe return Markup(s if not v else s % v) def swag_eval(swag, func): # Traverse the swag spec structure # and call func on summary and description keys for k in swag: if k in ['summary', 'description'] and isinstance(swag[k], str) and swag[k] != "": swag[k] = func(swag[k]) elif k == 'tags' and isinstance(swag[k], list): swag[k] = [func(v) for v in swag[k]] elif isinstance(swag[k], dict): swag_eval(swag[k], func) elif isinstance(swag[k], list) and k != 'consumes': for i in swag[k]: if isinstance(i, str): func(i) elif isinstance(i, dict): swag_eval(i, func) return swag def lazy_swag(swag): return swag_eval(swag, _lazy)