204aba376d
- Make babel dependent primitive types to use Locale('en') for data
validation instead of current locale. Using current locale leads to
infinite recursion in cases where the loaded data has dependency to
the loaded object's locale.
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
import sqlalchemy as sa
|
|
from sqlalchemy.ext.hybrid import hybrid_property
|
|
|
|
from .exceptions import ImproperlyConfigured
|
|
|
|
try:
|
|
import babel
|
|
except ImportError:
|
|
babel = None
|
|
|
|
try:
|
|
from flask.ext.babel import get_locale
|
|
except ImportError:
|
|
def get_locale():
|
|
raise ImproperlyConfigured(
|
|
'Could not load get_locale function from Flask-Babel. Either '
|
|
'install babel or make a similar function and override it '
|
|
'in this module.'
|
|
)
|
|
|
|
|
|
class TranslationHybrid(object):
|
|
def __init__(self, current_locale, default_locale, default_value=None):
|
|
if babel is None:
|
|
raise ImproperlyConfigured(
|
|
'You need to install babel in order to use TranslationHybrid.'
|
|
)
|
|
self.current_locale = current_locale
|
|
self.default_locale = default_locale
|
|
self.default_value = default_value
|
|
|
|
def cast_locale(self, obj, locale):
|
|
"""
|
|
Cast given locale to string. Supports also callbacks that return
|
|
locales.
|
|
"""
|
|
if callable(locale):
|
|
try:
|
|
locale = locale()
|
|
except TypeError:
|
|
locale = locale(obj)
|
|
if isinstance(locale, babel.Locale):
|
|
return str(locale)
|
|
|
|
return locale
|
|
|
|
def getter_factory(self, attr):
|
|
"""
|
|
Return a hybrid_property getter function for given attribute. The
|
|
returned getter first checks if object has translation for current
|
|
locale. If not it tries to get translation for default locale. If there
|
|
is no translation found for default locale it returns None.
|
|
"""
|
|
def getter(obj):
|
|
current_locale = self.cast_locale(obj, self.current_locale)
|
|
try:
|
|
return getattr(obj, attr.key)[current_locale]
|
|
except (TypeError, KeyError):
|
|
default_locale = self.cast_locale(
|
|
obj, self.default_locale
|
|
)
|
|
try:
|
|
return getattr(obj, attr.key)[default_locale]
|
|
except (TypeError, KeyError):
|
|
return self.default_value
|
|
return getter
|
|
|
|
def setter_factory(self, attr):
|
|
def setter(obj, value):
|
|
if getattr(obj, attr.key) is None:
|
|
setattr(obj, attr.key, {})
|
|
locale = self.cast_locale(obj, self.current_locale)
|
|
getattr(obj, attr.key)[locale] = value
|
|
return setter
|
|
|
|
def expr_factory(self, attr):
|
|
def expr(cls):
|
|
current_locale = self.cast_locale(cls, self.current_locale)
|
|
default_locale = self.cast_locale(cls, self.default_locale)
|
|
return sa.func.coalesce(attr[current_locale], attr[default_locale])
|
|
return expr
|
|
|
|
def __call__(self, attr):
|
|
return hybrid_property(
|
|
fget=self.getter_factory(attr),
|
|
fset=self.setter_factory(attr),
|
|
expr=self.expr_factory(attr)
|
|
)
|