mirror/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr synced 2025-02-22 14:00:52 +00:00
Code Issues Releases Wiki Activity

Fixed zimuku provider to prevent unexpected exception. #1459

Browse source
This commit is contained in:
morpheus65535 2021-07-14 19:16:12 -04:00
parent 8429087d8a
commit 8cdd362caa
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
libs/subliminal_patch/providers/zimuku.py
View file

@ -18,8 +18,8 @@ from subzero.language import Language
from guessit import guessit
from requests import Session
from six import text_type
from random import randint
from subliminal import __short_version__
from subliminal.providers import ParserBeautifulSoup
from subliminal_patch.providers import Provider
from subliminal.subtitle import (
@ -30,6 +30,7 @@ from subliminal_patch.subtitle import (
Subtitle,
guess_matches
)
from .utils import FIRST_THOUSAND_OR_SO_USER_AGENTS as AGENT_LIST
from subliminal.video import Episode, Movie
logger = logging.getLogger(__name__)
@ -89,8 +90,6 @@ class ZimukuProvider(Provider):
search_url = "/search?q={}"
download_url = "http://zimuku.org/"
UserAgent = "Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; Trident/6.0)"
subtitle_class = ZimukuSubtitle
def __init__(self):
@ -98,7 +97,7 @@ class ZimukuProvider(Provider):
def initialize(self):
self.session = Session()
self.session.headers["User-Agent"] = "Subliminal/{}".format(__short_version__)
self.session.headers["User-Agent"] = AGENT_LIST[randint(0, len(AGENT_LIST) - 1)]
def terminate(self):
self.session.close()
@ -258,9 +257,13 @@ class ZimukuProvider(Provider):
logger.info("Downloading subtitle %r", subtitle)
self.session = subtitle.session
download_link = _get_archive_dowload_link(self.session, subtitle.page_link)
r = self.session.get(download_link, timeout=30)
r = self.session.get(download_link, headers={'Referer': subtitle.page_link}, timeout=30)
r.raise_for_status()
filename = r.headers["Content-Disposition"]
try:
filename = r.headers["Content-Disposition"]
except KeyError:
logger.debug("Unable to parse subtitles filename. Dropping this subtitles.")
return
if not r.content:
logger.debug("Unable to download subtitle. No data returned from provider")