Merge pull request #916 from orikad/development

bugfix language id in wizdom provider
pull/922/head
morpheus65535 4 years ago committed by GitHub
commit 5eb62d6f1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -4,7 +4,7 @@ import logging
import os
import zipfile
from babelfish import Language
from subzero.language import Language
from guessit import guessit
from requests import Session
@ -75,7 +75,7 @@ class WizdomSubtitle(Subtitle):
class WizdomProvider(Provider):
"""Wizdom Provider."""
languages = {Language.fromalpha2(l) for l in ['he']}
languages = {Language(l) for l in ['heb']}
server_url = 'wizdom.xyz'
_tmdb_api_key = 'a51ee051bcd762543373903de296e0a3'
@ -156,7 +156,7 @@ class WizdomProvider(Provider):
# loop over results
subtitles = {}
for result in results:
language = Language.fromalpha2('he')
language = Language('heb')
hearing_impaired = False
subtitle_id = result['id']
release = result['version']
@ -197,6 +197,9 @@ class WizdomProvider(Provider):
r = self.session.get(url, headers={'Referer': subtitle.page_link}, timeout=10)
r.raise_for_status()
if len(r.content) == 0:
return
# open the zip
with zipfile.ZipFile(io.BytesIO(r.content)) as zf:
# remove some filenames from the namelist
@ -204,4 +207,4 @@ class WizdomProvider(Provider):
if len(namelist) > 1:
raise ProviderError('More than one file to unzip')
subtitle.content = fix_line_ending(zf.read(namelist[0]))
subtitle.content = fix_line_ending(zf.read(namelist[0]))

Loading…
Cancel
Save