mirror of https://github.com/morpheus65535/bazarr
23 lines
886 B
Python
23 lines
886 B
Python
import typing
|
|
|
|
import babelfish
|
|
|
|
|
|
class Context(dict):
|
|
def __init__(self, options: typing.Optional[typing.Mapping[str, typing.Any]] = None):
|
|
super().__init__(options or {})
|
|
language = self['expected_language'] if 'expected_language' in self else None
|
|
if language and not isinstance(language, babelfish.Language):
|
|
language = babelfish.Language.fromietf(str(language))
|
|
self.expected_language: typing.Optional[babelfish.Language] = language
|
|
|
|
def accept(self, lang: babelfish.Language):
|
|
if self.expected_language is None:
|
|
return True
|
|
if self.expected_language.alpha3 != lang.alpha3:
|
|
return False
|
|
if self.expected_language.script and self.expected_language != lang.script:
|
|
return False
|
|
|
|
return not self.expected_language.country or self.expected_language == lang.country
|