You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.1 KiB
113 lines
3.1 KiB
6 years ago
|
# coding=utf-8
|
||
|
|
from babelfish.exceptions import LanguageError
|
||
|
|
|
||
|
|
from babelfish import Language as Language_, basestr
|
||
|
|
|
||
|
|
|
||
|
|
repl_map = {
|
||
|
|
"dk": "da",
|
||
|
|
"nld": "nl",
|
||
|
|
"english": "en",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def language_from_stream(l):
|
||
|
|
if not l:
|
||
|
|
raise LanguageError()
|
||
|
|
for method in ("fromietf", "fromalpha3t", "fromalpha3b"):
|
||
|
|
try:
|
||
|
|
return getattr(Language, method)(l)
|
||
|
|
except (LanguageError, ValueError):
|
||
|
|
pass
|
||
|
|
raise LanguageError()
|
||
|
|
|
||
|
|
|
||
|
|
def wrap_forced(f):
|
||
|
|
def inner(*args, **kwargs):
|
||
|
|
"""
|
||
|
|
classmethod wrapper
|
||
|
|
:param args: args[0] = cls
|
||
|
|
:param kwargs:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
args = list(args)
|
||
|
|
cls = args[0]
|
||
|
|
args = args[1:]
|
||
|
|
s = args.pop(0)
|
||
|
|
base, forced = s.split(":") if ":" in s else (s, False)
|
||
|
|
instance = f(cls, base, *args, **kwargs)
|
||
|
|
if isinstance(instance, Language):
|
||
|
|
instance.forced = forced == "forced"
|
||
|
|
return instance
|
||
|
|
|
||
|
|
return inner
|
||
|
|
|
||
|
|
|
||
|
|
class Language(Language_):
|
||
|
|
forced = False
|
||
|
|
|
||
|
|
def __init__(self, language, country=None, script=None, unknown=None, forced=False):
|
||
|
|
self.forced = forced
|
||
|
|
super(Language, self).__init__(language, country=country, script=script, unknown=unknown)
|
||
|
|
|
||
|
|
def __getstate__(self):
|
||
|
|
return self.alpha3, self.country, self.script, self.forced
|
||
|
|
|
||
|
|
def __setstate__(self, state):
|
||
|
|
self.alpha3, self.country, self.script, self.forced = state
|
||
|
|
|
||
|
|
def __eq__(self, other):
|
||
|
|
if isinstance(other, basestr):
|
||
|
|
return str(self) == other
|
||
|
|
if not isinstance(other, Language):
|
||
|
|
return False
|
||
|
|
return (self.alpha3 == other.alpha3 and
|
||
|
|
self.country == other.country and
|
||
|
|
self.script == other.script and
|
||
|
|
bool(self.forced) == bool(other.forced))
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return super(Language, self).__str__() + (":forced" if self.forced else "")
|
||
|
|
|
||
|
|
@property
|
||
|
|
def basename(self):
|
||
|
|
return super(Language, self).__str__()
|
||
|
|
|
||
|
|
def __getattr__(self, name):
|
||
|
|
ret = super(Language, self).__getattr__(name)
|
||
|
|
if isinstance(ret, Language):
|
||
|
|
ret.forced = self.forced
|
||
|
|
return ret
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def rebuild(cls, instance, **replkw):
|
||
|
|
state = instance.__getstate__()
|
||
|
|
attrs = ("country", "script", "forced")
|
||
|
|
language = state[0]
|
||
|
|
kwa = dict(zip(attrs, state[1:]))
|
||
|
|
kwa.update(replkw)
|
||
|
|
return cls(language, **kwa)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
@wrap_forced
|
||
|
|
def fromcode(cls, code, converter):
|
||
|
|
return Language(*Language_.fromcode(code, converter).__getstate__())
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
@wrap_forced
|
||
|
|
def fromietf(cls, ietf):
|
||
|
|
ietf_lower = ietf.lower()
|
||
|
|
if ietf_lower in repl_map:
|
||
|
|
ietf = repl_map[ietf_lower]
|
||
|
|
|
||
|
|
return Language(*Language_.fromietf(ietf).__getstate__())
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
@wrap_forced
|
||
|
|
def fromalpha3b(cls, s):
|
||
|
|
if s in repl_map:
|
||
|
|
s = repl_map[s]
|
||
|
|
return Language(*Language_.fromietf(s).__getstate__())
|
||
|
|
|
||
|
|
return Language(*Language_.fromalpha3b(s).__getstate__())
|