mirror/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr synced 2024-12-23 08:13:14 +00:00
Code Issues Releases Wiki Activity

Fixed Supersubtitles provider #1300

Browse source
This commit is contained in:
morpheus65535 2021-02-28 09:23:55 -05:00
parent 5008d09230
commit 1b987d3470
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
libs/subliminal_patch/providers/supersubtitles.py
View file

@ -9,6 +9,7 @@ from babelfish import language_converters
from subzero.language import Language from subzero.language import Language
from requests import Session from requests import Session
import urllib.parse import urllib.parse
from random import randint
from subliminal.subtitle import fix_line_ending from subliminal.subtitle import fix_line_ending
from subliminal_patch.providers import Provider from subliminal_patch.providers import Provider
@ -23,6 +24,7 @@ from zipfile import ZipFile
from rarfile import RarFile, is_rarfile from rarfile import RarFile, is_rarfile
from subliminal_patch.utils import sanitize, fix_inconsistent_naming from subliminal_patch.utils import sanitize, fix_inconsistent_naming
from guessit import guessit from guessit import guessit
from .utils import FIRST_THOUSAND_OR_SO_USER_AGENTS as AGENT_LIST
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -175,7 +177,7 @@ class SuperSubtitlesProvider(Provider, ProviderSubtitleArchiveMixin):
def initialize(self): def initialize(self):
self.session = Session() self.session = Session()
self.session.headers = {'User-Agent': os.environ.get("SZ_USER_AGENT", "Sub-Zero/2")} self.session.headers = {'User-Agent': AGENT_LIST[randint(0, len(AGENT_LIST) - 1)]}
def terminate(self): def terminate(self):
self.session.close() self.session.close()
@ -251,18 +253,18 @@ class SuperSubtitlesProvider(Provider, ProviderSubtitleArchiveMixin):
except IndexError: except IndexError:
continue continue
result_title = fix_tv_naming(result_title).strip().replace("<EFBFBD>", "").replace(" ", ".") result_title = fix_tv_naming(result_title).strip().replace("<EFBFBD>", "").replace("& ", "").replace(" ", ".")
if not result_title: if not result_title:
continue continue
guessable = result_title.strip() + ".s01e01." + result_year guessable = result_title.strip() + ".s01e01." + result_year
guess = guessit(guessable, {'type': "episode"}) guess = guessit(guessable, {'type': "episode"})
if sanitize(original_title) == sanitize(guess['title']) and year and guess['year'] and \ if sanitize(original_title.replace('& ', '')) == sanitize(guess['title']) and year and guess['year'] and \
year == guess['year']: year == guess['year']:
# Return the founded id # Return the founded id
return result_id return result_id
elif sanitize(original_title) == sanitize(guess['title']) and not year: elif sanitize(original_title.replace('& ', '')) == sanitize(guess['title']) and not year:
# Return the founded id # Return the founded id
return result_id return result_id
@ -364,6 +366,7 @@ class SuperSubtitlesProvider(Provider, ProviderSubtitleArchiveMixin):
sub_english_name = sub_english_name.group() if sub_english_name else '' sub_english_name = sub_english_name.group() if sub_english_name else ''
sub_english_name = sub_english_name.split(' (')[0] sub_english_name = sub_english_name.split(' (')[0]
sub_english_name = sub_english_name.replace('&amp;', '&')
sub_version = 'n/a' sub_version = 'n/a'
if len(str(sub_english).split('(')) > 1: if len(str(sub_english).split('(')) > 1:
sub_version = (str(sub_english).split('(')[len(str(sub_english).split('(')) - 1]).split(')')[0] sub_version = (str(sub_english).split('(')[len(str(sub_english).split('(')) - 1]).split(')')[0]