Merge development into master

This commit is contained in:
github-actions[bot] 2023-07-22 13:49:00 +00:00 committed by GitHub
commit 64af56cb80
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 61 additions and 43 deletions

View File

@ -391,6 +391,7 @@ def save_settings(settings_items):
undefined_audio_track_default_changed = False undefined_audio_track_default_changed = False
undefined_subtitles_track_default_changed = False undefined_subtitles_track_default_changed = False
audio_tracks_parsing_changed = False audio_tracks_parsing_changed = False
reset_providers = False
# Subzero Mods # Subzero Mods
update_subzero = False update_subzero = False
@ -491,46 +492,62 @@ def save_settings(settings_items):
if key == 'settings-addic7ed-username': if key == 'settings-addic7ed-username':
if key != settings.addic7ed.username: if key != settings.addic7ed.username:
reset_providers = True
region.delete('addic7ed_data') region.delete('addic7ed_data')
elif key == 'settings-addic7ed-password': elif key == 'settings-addic7ed-password':
if key != settings.addic7ed.password: if key != settings.addic7ed.password:
reset_providers = True
region.delete('addic7ed_data') region.delete('addic7ed_data')
if key == 'settings-legendasdivx-username': if key == 'settings-legendasdivx-username':
if key != settings.legendasdivx.username: if key != settings.legendasdivx.username:
reset_providers = True
region.delete('legendasdivx_cookies2') region.delete('legendasdivx_cookies2')
elif key == 'settings-legendasdivx-password': elif key == 'settings-legendasdivx-password':
if key != settings.legendasdivx.password: if key != settings.legendasdivx.password:
reset_providers = True
region.delete('legendasdivx_cookies2') region.delete('legendasdivx_cookies2')
if key == 'settings-opensubtitles-username': if key == 'settings-opensubtitles-username':
if key != settings.opensubtitles.username: if key != settings.opensubtitles.username:
reset_providers = True
region.delete('os_token') region.delete('os_token')
elif key == 'settings-opensubtitles-password': elif key == 'settings-opensubtitles-password':
if key != settings.opensubtitles.password: if key != settings.opensubtitles.password:
reset_providers = True
region.delete('os_token') region.delete('os_token')
if key == 'settings-opensubtitlescom-username': if key == 'settings-opensubtitlescom-username':
if key != settings.opensubtitlescom.username: if key != settings.opensubtitlescom.username:
reset_providers = True
region.delete('oscom_token') region.delete('oscom_token')
elif key == 'settings-opensubtitlescom-password': elif key == 'settings-opensubtitlescom-password':
if key != settings.opensubtitlescom.password: if key != settings.opensubtitlescom.password:
reset_providers = True
region.delete('oscom_token') region.delete('oscom_token')
if key == 'settings-subscene-username': if key == 'settings-subscene-username':
if key != settings.subscene.username: if key != settings.subscene.username:
reset_providers = True
region.delete('subscene_cookies2') region.delete('subscene_cookies2')
elif key == 'settings-subscene-password': elif key == 'settings-subscene-password':
if key != settings.subscene.password: if key != settings.subscene.password:
reset_providers = True
region.delete('subscene_cookies2') region.delete('subscene_cookies2')
if key == 'settings-titlovi-username': if key == 'settings-titlovi-username':
if key != settings.titlovi.username: if key != settings.titlovi.username:
reset_providers = True
region.delete('titlovi_token') region.delete('titlovi_token')
elif key == 'settings-titlovi-password': elif key == 'settings-titlovi-password':
if key != settings.titlovi.password: if key != settings.titlovi.password:
reset_providers = True
region.delete('titlovi_token') region.delete('titlovi_token')
if reset_providers:
from .get_providers import reset_throttled_providers
reset_throttled_providers(only_auth_or_conf_error=True)
if settings_keys[0] == 'settings': if settings_keys[0] == 'settings':
settings[settings_keys[1]][settings_keys[2]] = str(value) settings[settings_keys[1]][settings_keys[2]] = str(value)

View File

@ -14,7 +14,7 @@ import requests
from subliminal_patch.exceptions import TooManyRequests, APIThrottled, ParseResponseError, IPAddressBlocked, \ from subliminal_patch.exceptions import TooManyRequests, APIThrottled, ParseResponseError, IPAddressBlocked, \
MustGetBlacklisted, SearchLimitReached MustGetBlacklisted, SearchLimitReached
from subliminal.providers.opensubtitles import DownloadLimitReached from subliminal.providers.opensubtitles import DownloadLimitReached
from subliminal.exceptions import DownloadLimitExceeded, ServiceUnavailable from subliminal.exceptions import DownloadLimitExceeded, ServiceUnavailable, AuthenticationError, ConfigurationError
from subliminal import region as subliminal_cache_region from subliminal import region as subliminal_cache_region
from subliminal_patch.extensions import provider_registry from subliminal_patch.extensions import provider_registry
@ -76,6 +76,8 @@ def provider_throttle_map():
APIThrottled: (datetime.timedelta(seconds=15), "15 seconds"), APIThrottled: (datetime.timedelta(seconds=15), "15 seconds"),
}, },
"opensubtitlescom": { "opensubtitlescom": {
AuthenticationError: (datetime.timedelta(hours=12), "12 hours"),
ConfigurationError: (datetime.timedelta(hours=12), "12 hours"),
TooManyRequests: (datetime.timedelta(minutes=1), "1 minute"), TooManyRequests: (datetime.timedelta(minutes=1), "1 minute"),
DownloadLimitExceeded: (datetime.timedelta(hours=24), "24 hours"), DownloadLimitExceeded: (datetime.timedelta(hours=24), "24 hours"),
}, },
@ -413,12 +415,17 @@ def list_throttled_providers():
return throttled_providers return throttled_providers
def reset_throttled_providers(): def reset_throttled_providers(only_auth_or_conf_error=False):
for provider in list(tp): for provider in list(tp):
if only_auth_or_conf_error and tp[provider][0] not in ['AuthenticationError', 'ConfigurationError']:
continue
del tp[provider] del tp[provider]
set_throttled_providers(str(tp)) set_throttled_providers(str(tp))
update_throttled_provider() update_throttled_provider()
logging.info('BAZARR throttled providers have been reset.') if only_auth_or_conf_error:
logging.info('BAZARR throttled providers have been reset (only AuthenticationError and ConfigurationError).')
else:
logging.info('BAZARR throttled providers have been reset.')
def get_throttled_providers(): def get_throttled_providers():

View File

@ -183,7 +183,12 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
def initialize(self): def initialize(self):
self._started = time.time() self._started = time.time()
self.login()
if region.get("oscom_token", expiration_time=TOKEN_EXPIRATION_TIME) is NO_VALUE:
logger.debug("No cached token, we'll try to login again.")
self.login()
else:
self.token = region.get("oscom_token", expiration_time=TOKEN_EXPIRATION_TIME)
def terminate(self): def terminate(self):
self.session.close() self.session.close()
@ -191,26 +196,21 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
def ping(self): def ping(self):
return self._started and (time.time() - self._started) < TOKEN_EXPIRATION_TIME return self._started and (time.time() - self._started) < TOKEN_EXPIRATION_TIME
def login(self): def login(self, is_retry=False):
r = self.retry( r = self.checked(
lambda: self.checked( lambda: self.session.post(self.server_url + 'login',
lambda: self.session.post(self.server_url + 'login', json={"username": self.username, "password": self.password},
json={"username": self.username, "password": self.password}, allow_redirects=False,
allow_redirects=False, timeout=30),
timeout=30), is_retry=is_retry)
validate_json=True,
json_key_name='token'
),
amount=retry_amount
)
try: try:
self.token = r.json()['token'] self.token = r.json()['token']
except (ValueError, JSONDecodeError): except (ValueError, JSONDecodeError):
return False log_request_response(r)
raise ProviderError("Cannot get token from provider login response")
else: else:
region.set("oscom_token", self.token) region.set("oscom_token", self.token)
return True
@staticmethod @staticmethod
def sanitize_external_ids(external_id): def sanitize_external_ids(external_id):
@ -262,9 +262,6 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
logger.debug(f'No match found for {title}') logger.debug(f'No match found for {title}')
def query(self, languages, video): def query(self, languages, video):
if region.get("oscom_token", expiration_time=TOKEN_EXPIRATION_TIME) is NO_VALUE:
logger.debug("No cached token, we'll try to login again.")
self.login()
self.video = video self.video = video
if self.use_hash: if self.use_hash:
file_hash = self.video.hashes.get('opensubtitlescom') file_hash = self.video.hashes.get('opensubtitlescom')
@ -289,9 +286,11 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
if not title_id: if not title_id:
return [] return []
lang_strings = [to_opensubtitlescom(lang.basename) for lang in languages] # be sure to remove duplicates
lang_strings = list(set([to_opensubtitlescom(lang.basename) for lang in languages]))
langs = ','.join(lang_strings) langs = ','.join(lang_strings)
logging.debug(f'Searching for this languages: {lang_strings}') logging.debug(f'Searching for those languages: {lang_strings}')
# query the server # query the server
if isinstance(self.video, Episode): if isinstance(self.video, Episode):
@ -400,13 +399,6 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
return self.query(languages, video) return self.query(languages, video)
def download_subtitle(self, subtitle): def download_subtitle(self, subtitle):
if region.get("oscom_token", expiration_time=TOKEN_EXPIRATION_TIME) is NO_VALUE:
logger.debug("No cached token, we'll try to login again.")
self.login()
if self.token is NO_VALUE:
logger.debug("Unable to obtain an authentication token right now, we'll try again later.")
raise ProviderError("Unable to obtain an authentication token")
logger.info('Downloading subtitle %r', subtitle) logger.info('Downloading subtitle %r', subtitle)
headers = {'Accept': 'application/json', 'Content-Type': 'application/json', headers = {'Accept': 'application/json', 'Content-Type': 'application/json',
@ -442,12 +434,14 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
subtitle_content = r.content subtitle_content = r.content
subtitle.content = fix_line_ending(subtitle_content) subtitle.content = fix_line_ending(subtitle_content)
def reset_token(self): @staticmethod
def reset_token():
logging.debug('Authentication failed: clearing cache and attempting to login.') logging.debug('Authentication failed: clearing cache and attempting to login.')
region.delete("oscom_token") region.delete("oscom_token")
return self.login() return
def checked(self, fn, raise_api_limit=False, validate_json=False, json_key_name=None, validate_content=False): def checked(self, fn, raise_api_limit=False, validate_json=False, json_key_name=None, validate_content=False,
is_retry=False):
"""Run :fn: and check the response status before returning it. """Run :fn: and check the response status before returning it.
:param fn: the function to make an API call to OpenSubtitles.com. :param fn: the function to make an API call to OpenSubtitles.com.
@ -455,6 +449,7 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
:param validate_json: test if response is valid json. :param validate_json: test if response is valid json.
:param json_key_name: test if returned json contain a specific key. :param json_key_name: test if returned json contain a specific key.
:param validate_content: test if response have a content (used with download). :param validate_content: test if response have a content (used with download).
:param is_retry: prevent additional retries with login endpoint.
:return: the response. :return: the response.
""" """
@ -465,7 +460,7 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
except APIThrottled: except APIThrottled:
if not raise_api_limit: if not raise_api_limit:
logger.info("API request limit hit, waiting and trying again once.") logger.info("API request limit hit, waiting and trying again once.")
time.sleep(2) time.sleep(15)
return self.checked(fn, raise_api_limit=True) return self.checked(fn, raise_api_limit=True)
raise raise
except (ConnectionError, Timeout, ReadTimeout): except (ConnectionError, Timeout, ReadTimeout):
@ -478,17 +473,16 @@ class OpenSubtitlesComProvider(ProviderRetryMixin, Provider):
except Exception: except Exception:
status_code = None status_code = None
else: else:
if status_code == 400: if status_code == 401:
raise ConfigurationError('Do not use email but username')
elif status_code == 401:
time.sleep(1)
log_request_response(response) log_request_response(response)
logged_in = self.reset_token() self.reset_token()
if logged_in: if is_retry:
return self.checked(fn, raise_api_limit=raise_api_limit, validate_json=validate_json,
json_key_name=json_key_name, validate_content=validate_content)
else:
raise AuthenticationError('Login failed') raise AuthenticationError('Login failed')
else:
time.sleep(1)
self.login(is_retry=True)
self.checked(fn, raise_api_limit=raise_api_limit, validate_json=validate_json,
json_key_name=json_key_name, validate_content=validate_content, is_retry=True)
elif status_code == 403: elif status_code == 403:
log_request_response(response) log_request_response(response)
raise ProviderError("Bazarr API key seems to be in problem") raise ProviderError("Bazarr API key seems to be in problem")