mirror
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Releases Wiki Activity

no log: update subf2m provider

This commit is contained in:
Vitiko 2022-04-18 16:58:11 -04:00
parent 5e0461eee5
commit d95d6b9d35
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libs/subliminal_patch/providers/subf2m.py
View File

@ -1,11 +1,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import io
import logging import logging
from zipfile import ZipFile, is_zipfile
from rarfile import RarFile, is_rarfile
from guessit import guessit from guessit import guessit
from requests import Session from requests import Session
from bs4 import BeautifulSoup as bso from bs4 import BeautifulSoup as bso
@ -16,7 +12,8 @@ from subliminal_patch.core import Movie
from subliminal_patch.providers import Provider from subliminal_patch.providers import Provider
from subliminal_patch.subtitle import Subtitle from subliminal_patch.subtitle import Subtitle
from subliminal_patch.subtitle import guess_matches from subliminal_patch.subtitle import guess_matches
from subliminal_patch.providers.mixins import ProviderSubtitleArchiveMixin from subliminal_patch.providers.utils import get_archive_from_bytes
from subliminal_patch.providers.utils import get_subtitle_from_archive
from subzero.language import Language from subzero.language import Language
@ -88,7 +85,7 @@ _LANGUAGE_MAP = {
} }
class Subf2mProvider(Provider, ProviderSubtitleArchiveMixin): class Subf2mProvider(Provider):
provider_name = "subf2m" provider_name = "subf2m"
_supported_languages = {} _supported_languages = {}
@ -240,18 +237,12 @@ class Subf2mProvider(Provider, ProviderSubtitleArchiveMixin):
downloaded = self._session.get(download_url, allow_redirects=True) downloaded = self._session.get(download_url, allow_redirects=True)
archive_stream = io.BytesIO(downloaded.content) archive = get_archive_from_bytes(downloaded.content)
if is_zipfile(archive_stream): if archive is None:
logger.debug("Identified zip archive")
archive = ZipFile(archive_stream)
elif is_rarfile(archive_stream):
logger.debug("Identified rar archive")
archive = RarFile(archive_stream)
else:
raise APIThrottled(f"Invalid archive: {subtitle.page_link}") raise APIThrottled(f"Invalid archive: {subtitle.page_link}")
subtitle.content = self.get_subtitle_from_archive(subtitle, archive) subtitle.content = get_subtitle_from_archive(archive, get_first_subtitle=True)
def _get_subtitle_from_item(item, language): def _get_subtitle_from_item(item, language):