pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

LRU cache

Browse Source
This commit is contained in:
George Drummond
2019-01-03 12:04:57 -05:00
parent 1b0bdff90b
commit e630a7868f
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
atst/utils/localization.py
View File

@@ -1,7 +1,11 @@
import yaml import yaml
import os
from functools import lru_cache
from flask import current_app as app from flask import current_app as app
from atst.utils import getattr_path from atst.utils import getattr_path
ENV = os.getenv("FLASK_ENV", "dev")
class LocalizationInvalidKeyError(Exception): class LocalizationInvalidKeyError(Exception):
def __init__(self, key, variables): def __init__(self, key, variables):
@@ -14,18 +18,23 @@ class LocalizationInvalidKeyError(Exception):
) )
translations_yaml_max_cache = 0 if ENV == "dev" else None
@lru_cache(maxsize=translations_yaml_max_cache)
def _translations_file(): def _translations_file():
file_name = "translations.yaml" file_name = "translations.yaml"
if app: if app:
file_name = app.config.get("DEFAULT_TRANSLATIONS_FILE", file_name) file_name = app.config.get("DEFAULT_TRANSLATIONS_FILE", file_name)
return open(file_name) file = open(file_name)
return yaml.safe_load(file)
def translate(key, variables=None): def translate(key, variables=None):
translations_file = _translations_file() translations = _translations_file()
translations = yaml.safe_load(translations_file)
value = getattr_path(translations, key) value = getattr_path(translations, key)
if variables is None: if variables is None: