mirror of https://github.com/morpheus65535/bazarr
287 lines
8.6 KiB
Python
287 lines
8.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# BSD 2-Clause License
|
|
#
|
|
# Apprise - Push Notification Library.
|
|
# Copyright (c) 2023, Chris Caron <lead2gold@gmail.com>
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions and the following disclaimer.
|
|
#
|
|
# 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions and the following disclaimer in the documentation
|
|
# and/or other materials provided with the distribution.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
import ctypes
|
|
import locale
|
|
import contextlib
|
|
import os
|
|
import re
|
|
from os.path import join
|
|
from os.path import dirname
|
|
from os.path import abspath
|
|
from .logger import logger
|
|
|
|
|
|
# This gets toggled to True if we succeed
|
|
GETTEXT_LOADED = False
|
|
|
|
try:
|
|
# Initialize gettext
|
|
import gettext
|
|
|
|
# Toggle our flag
|
|
GETTEXT_LOADED = True
|
|
|
|
except ImportError:
|
|
# gettext isn't available; no problem; Use the library features without
|
|
# multi-language support.
|
|
pass
|
|
|
|
|
|
class AppriseLocale:
|
|
"""
|
|
A wrapper class to gettext so that we can manipulate multiple lanaguages
|
|
on the fly if required.
|
|
|
|
"""
|
|
|
|
# Define our translation domain
|
|
_domain = 'apprise'
|
|
|
|
# The path to our translations
|
|
_locale_dir = abspath(join(dirname(__file__), 'i18n'))
|
|
|
|
# Locale regular expression
|
|
_local_re = re.compile(
|
|
r'^((?P<ansii>C)|(?P<lang>([a-z]{2}))([_:](?P<country>[a-z]{2}))?)'
|
|
r'(\.(?P<enc>[a-z0-9-]+))?$', re.IGNORECASE)
|
|
|
|
# Define our default encoding
|
|
_default_encoding = 'utf-8'
|
|
|
|
# The function to assign `_` by default
|
|
_fn = 'gettext'
|
|
|
|
# The language we should fall back to if all else fails
|
|
_default_language = 'en'
|
|
|
|
def __init__(self, language=None):
|
|
"""
|
|
Initializes our object, if a language is specified, then we
|
|
initialize ourselves to that, otherwise we use whatever we detect
|
|
from the local operating system. If all else fails, we resort to the
|
|
defined default_language.
|
|
|
|
"""
|
|
|
|
# Cache previously loaded translations
|
|
self._gtobjs = {}
|
|
|
|
# Get our language
|
|
self.lang = AppriseLocale.detect_language(language)
|
|
|
|
# Our mapping to our _fn
|
|
self.__fn_map = None
|
|
|
|
if GETTEXT_LOADED is False:
|
|
# We're done
|
|
return
|
|
|
|
# Add language
|
|
self.add(self.lang)
|
|
|
|
def add(self, lang=None, set_default=True):
|
|
"""
|
|
Add a language to our list
|
|
"""
|
|
lang = lang if lang else self._default_language
|
|
if lang not in self._gtobjs:
|
|
# Load our gettext object and install our language
|
|
try:
|
|
self._gtobjs[lang] = gettext.translation(
|
|
self._domain, localedir=self._locale_dir, languages=[lang],
|
|
fallback=False)
|
|
|
|
# The non-intrusive method of applying the gettext change to
|
|
# the global namespace only
|
|
self.__fn_map = getattr(self._gtobjs[lang], self._fn)
|
|
|
|
except FileNotFoundError:
|
|
# The translation directory does not exist
|
|
logger.debug(
|
|
'Could not load translation path: %s',
|
|
join(self._locale_dir, lang))
|
|
|
|
# Fallback (handle case where self.lang does not exist)
|
|
if self.lang not in self._gtobjs:
|
|
self._gtobjs[self.lang] = gettext
|
|
self.__fn_map = getattr(self._gtobjs[self.lang], self._fn)
|
|
|
|
return False
|
|
|
|
logger.trace('Loaded language %s', lang)
|
|
|
|
if set_default:
|
|
logger.debug('Language set to %s', lang)
|
|
self.lang = lang
|
|
|
|
return True
|
|
|
|
@contextlib.contextmanager
|
|
def lang_at(self, lang, mapto=_fn):
|
|
"""
|
|
The syntax works as:
|
|
with at.lang_at('fr'):
|
|
# apprise works as though the french language has been
|
|
# defined. afterwards, the language falls back to whatever
|
|
# it was.
|
|
"""
|
|
|
|
if GETTEXT_LOADED is False:
|
|
# Do nothing
|
|
yield None
|
|
|
|
# we're done
|
|
return
|
|
|
|
# Tidy the language
|
|
lang = AppriseLocale.detect_language(lang, detect_fallback=False)
|
|
if lang not in self._gtobjs and not self.add(lang, set_default=False):
|
|
# Do Nothing
|
|
yield getattr(self._gtobjs[self.lang], mapto)
|
|
else:
|
|
# Yield
|
|
yield getattr(self._gtobjs[lang], mapto)
|
|
|
|
return
|
|
|
|
@property
|
|
def gettext(self):
|
|
"""
|
|
Return the current language gettext() function
|
|
|
|
Useful for assigning to `_`
|
|
"""
|
|
return self._gtobjs[self.lang].gettext
|
|
|
|
@staticmethod
|
|
def detect_language(lang=None, detect_fallback=True):
|
|
"""
|
|
Returns the language (if it's retrievable)
|
|
"""
|
|
# We want to only use the 2 character version of this language
|
|
# hence en_CA becomes en, en_US becomes en.
|
|
if not isinstance(lang, str):
|
|
if detect_fallback is False:
|
|
# no detection enabled; we're done
|
|
return None
|
|
|
|
# Posix lookup
|
|
lookup = os.environ.get
|
|
localename = None
|
|
for variable in ('LC_ALL', 'LC_CTYPE', 'LANG', 'LANGUAGE'):
|
|
localename = lookup(variable, None)
|
|
if localename:
|
|
result = AppriseLocale._local_re.match(localename)
|
|
if result and result.group('lang'):
|
|
return result.group('lang').lower()
|
|
|
|
# Windows handling
|
|
if hasattr(ctypes, 'windll'):
|
|
windll = ctypes.windll.kernel32
|
|
try:
|
|
lang = locale.windows_locale[
|
|
windll.GetUserDefaultUILanguage()]
|
|
|
|
# Our detected windows language
|
|
return lang[0:2].lower()
|
|
|
|
except (TypeError, KeyError):
|
|
# Fallback to posix detection
|
|
pass
|
|
|
|
# Built in locale library check
|
|
try:
|
|
# Acquire our locale
|
|
lang = locale.getlocale()[0]
|
|
|
|
except (ValueError, TypeError) as e:
|
|
# This occurs when an invalid locale was parsed from the
|
|
# environment variable. While we still return None in this
|
|
# case, we want to better notify the end user of this. Users
|
|
# receiving this error should check their environment
|
|
# variables.
|
|
logger.warning(
|
|
'Language detection failure / {}'.format(str(e)))
|
|
return None
|
|
|
|
return None if not lang else lang[0:2].lower()
|
|
|
|
def __getstate__(self):
|
|
"""
|
|
Pickle Support dumps()
|
|
"""
|
|
state = self.__dict__.copy()
|
|
|
|
# Remove the unpicklable entries.
|
|
del state['_gtobjs']
|
|
del state['_AppriseLocale__fn_map']
|
|
return state
|
|
|
|
def __setstate__(self, state):
|
|
"""
|
|
Pickle Support loads()
|
|
"""
|
|
self.__dict__.update(state)
|
|
# Our mapping to our _fn
|
|
self.__fn_map = None
|
|
self._gtobjs = {}
|
|
self.add(state['lang'], set_default=True)
|
|
|
|
|
|
#
|
|
# Prepare our default LOCALE Singleton
|
|
#
|
|
LOCALE = AppriseLocale()
|
|
|
|
|
|
class LazyTranslation:
|
|
"""
|
|
Doesn't translate anything until str() or unicode() references
|
|
are made.
|
|
|
|
"""
|
|
def __init__(self, text, *args, **kwargs):
|
|
"""
|
|
Store our text
|
|
"""
|
|
self.text = text
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return LOCALE.gettext(self.text) if GETTEXT_LOADED else self.text
|
|
|
|
|
|
# Lazy translation handling
|
|
def gettext_lazy(text):
|
|
"""
|
|
A dummy function that can be referenced
|
|
"""
|
|
return LazyTranslation(text=text)
|