From 63eded5aa38f2241fabbe99516064b941cf0d16d Mon Sep 17 00:00:00 2001 From: Vitiko Date: Sun, 17 Apr 2022 22:39:37 -0400 Subject: [PATCH] Add subf2m.co Provider --- frontend/src/pages/Settings/Providers/list.ts | 5 + libs/subliminal_patch/providers/subf2m.py | 268 ++++++++++++++++++ tests/subliminal_patch/test_subf2m.py | 98 +++++++ 3 files changed, 371 insertions(+) create mode 100644 libs/subliminal_patch/providers/subf2m.py create mode 100644 tests/subliminal_patch/test_subf2m.py diff --git a/frontend/src/pages/Settings/Providers/list.ts b/frontend/src/pages/Settings/Providers/list.ts index 158328cce..94724c435 100644 --- a/frontend/src/pages/Settings/Providers/list.ts +++ b/frontend/src/pages/Settings/Providers/list.ts @@ -197,6 +197,11 @@ export const ProviderList: Readonly = [ name: "Subs.sab.bz", description: "Bulgarian Subtitles Provider", }, + { + key: "subf2m", + name: "subf2m.co", + description: "Subscene Alternative Provider", + }, { key: "subs4free", name: "Subs4Free", diff --git a/libs/subliminal_patch/providers/subf2m.py b/libs/subliminal_patch/providers/subf2m.py new file mode 100644 index 000000000..061976837 --- /dev/null +++ b/libs/subliminal_patch/providers/subf2m.py @@ -0,0 +1,268 @@ +# -*- coding: utf-8 -*- + +import io +import logging + +from zipfile import ZipFile, is_zipfile +from rarfile import RarFile, is_rarfile + +from guessit import guessit +from requests import Session +from bs4 import BeautifulSoup as bso + +from subliminal_patch.exceptions import APIThrottled +from subliminal_patch.core import Episode +from subliminal_patch.core import Movie +from subliminal_patch.providers import Provider +from subliminal_patch.subtitle import Subtitle +from subliminal_patch.subtitle import guess_matches +from subliminal_patch.providers.mixins import ProviderSubtitleArchiveMixin + +from subzero.language import Language + +logger = logging.getLogger(__name__) + + +class Subf2mSubtitle(Subtitle): + provider_name = "subf2m" + hash_verifiable = False + + def __init__(self, language, page_link, release_info): + super().__init__(language, page_link=page_link) + + self.release_info = release_info + self._matches = set() + + def get_matches(self, video): + type_ = "episode" if isinstance(video, Episode) else "movie" + + for release in self.release_info.split("\n"): + self._matches |= guess_matches( + video, guessit(release.strip(), {"type": type_}) + ) + + return self._matches + + @property + def id(self): + return self.page_link + + +_BASE_URL = "https://subf2m.co" + +# TODO: add more seasons and languages + +_SEASONS = ( + "First", + "Second", + "Third", + "Fourth", + "Fifth", + "Sixth", + "Seventh", + "Eighth", + "Ninth", + "Tenth", + "Eleventh", + "Twelfth", + "Thirdteenth", + "Fourthteenth", + "Fifteenth", + "Sixteenth", + "Seventeenth", + "Eightheenth", + "Nineteenth", + "Tweentieth", +) + +_LANGUAGE_MAP = { + "english": "eng", + "farsi_persian": "per", + "arabic": "ara", + "spanish": "spa", + "portuguese": "por", + "italian": "ita", + "dutch": "dut", + "hebrew": "heb", + "indonesian": "ind", +} + + +class Subf2mProvider(Provider, ProviderSubtitleArchiveMixin): + provider_name = "subf2m" + + _supported_languages = {} + _supported_languages["brazillian-portuguese"] = Language("por", "BR") + + for key, val in _LANGUAGE_MAP.items(): + _supported_languages[key] = Language.fromalpha3b(val) + + _supported_languages_reversed = { + val: key for key, val in _supported_languages.items() + } + + languages = set(_supported_languages.values()) + + video_types = (Episode, Movie) + subtitle_class = Subf2mSubtitle + _session = None + + def initialize(self): + self._session = Session() + self._session.headers.update({"user-agent": "Bazarr"}) + + def terminate(self): + self._session.close() + + def _gen_results(self, query): + req = self._session.get( + f"{_BASE_URL}/subtitles/searchbytitle?query={query.replace(' ', '+')}&l=", + stream=True, + ) + text = "\n".join(line for line in req.iter_lines(decode_unicode=True) if line) + soup = bso(text, "html.parser") + + for title in soup.select("li div[class='title'] a"): + yield title + + def _search_movie(self, title, year): + title = title.lower() + year = f"({year})" + + found_movie = None + + for result in self._gen_results(title): + text = result.text.lower() + if title.lower() in text and year in text: + found_movie = result.get("href") + logger.debug("Movie found: %s", found_movie) + break + + return found_movie + + def _search_tv_show_season(self, title, season): + try: + season_str = f"{_SEASONS[season - 1]} Season" + except IndexError: + logger.debug("Season number not supported: %s", season) + return None + + expected_result = f"{title} - {season_str}".lower() + + found_tv_show_season = None + + for result in self._gen_results(title): + if expected_result in result.text.lower(): + found_tv_show_season = result.get("href") + logger.debug("TV Show season found: %s", found_tv_show_season) + break + + return found_tv_show_season + + def _find_movie_subtitles(self, path, language): + soup = self._get_subtitle_page_soup(path, language) + subtitles = [] + + for item in soup.select("li.item"): + subtitle = _get_subtitle_from_item(item, language) + if subtitle is None: + continue + + logger.debug("Found subtitle: %s", subtitle) + subtitles.append(subtitle) + + return subtitles + + def _find_episode_subtitles(self, path, season, episode, language): + # TODO: add season packs support? + + soup = self._get_subtitle_page_soup(path, language) + expected_substring = f"s{season:02}e{episode:02}".lower() + subtitles = [] + + for item in soup.select("li.item"): + if expected_substring in item.text.lower(): + subtitle = _get_subtitle_from_item(item, language) + if subtitle is None: + continue + + logger.debug("Found subtitle: %s", subtitle) + subtitles.append(subtitle) + + return subtitles + + def _get_subtitle_page_soup(self, path, language): + language_path = self._supported_languages_reversed[language] + + req = self._session.get(f"{_BASE_URL}{path}/{language_path}", stream=True) + text = "\n".join(line for line in req.iter_lines(decode_unicode=True) if line) + + return bso(text, "html.parser") + + def list_subtitles(self, video, languages): + is_episode = isinstance(video, Episode) + + if is_episode: + result = self._search_tv_show_season(video.series, video.season) + else: + result = self._search_movie(video.title, video.year) + + if result is None: + logger.debug("No results") + return [] + + subtitles = [] + + for language in languages: + if is_episode: + subtitles.extend( + self._find_episode_subtitles( + result, video.season, video.episode, language + ) + ) + else: + subtitles.extend(self._find_movie_subtitles(result, language)) + + return subtitles + + def download_subtitle(self, subtitle): + # TODO: add MustGetBlacklisted support + + req = self._session.get(subtitle.page_link, stream=True) + text = "\n".join(line for line in req.iter_lines(decode_unicode=True) if line) + soup = bso(text, "html.parser") + try: + download_url = _BASE_URL + str( + soup.select_one("a[id='downloadButton']")["href"] # type: ignore + ) + except (AttributeError, KeyError): + raise APIThrottled(f"Couldn't get download url from {subtitle.page_link}") + + downloaded = self._session.get(download_url, allow_redirects=True) + + archive_stream = io.BytesIO(downloaded.content) + + if is_zipfile(archive_stream): + 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}") + + subtitle.content = self.get_subtitle_from_archive(subtitle, archive) + + +def _get_subtitle_from_item(item, language): + release_info = "\n".join( + release.text for release in item.find("ul", {"class": "scrolllist"}) + ).strip() + + try: + path = item.find("a", {"class": "download icon-download"})["href"] # type: ignore + except (AttributeError, KeyError): + logger.debug("Couldn't get path: %s", item) + return None + + return Subf2mSubtitle(language, _BASE_URL + path, release_info) diff --git a/tests/subliminal_patch/test_subf2m.py b/tests/subliminal_patch/test_subf2m.py new file mode 100644 index 000000000..bdc956914 --- /dev/null +++ b/tests/subliminal_patch/test_subf2m.py @@ -0,0 +1,98 @@ +import pytest + +from subliminal_patch.providers.subf2m import Subf2mProvider +from subliminal_patch.providers.subf2m import Subf2mSubtitle +from subzero.language import Language + + +def test_search_movie(movies): + movie = movies["dune"] + + with Subf2mProvider() as provider: + result = provider._search_movie(movie.title, movie.year) + assert result == "/subtitles/dune-2021" + + +def test_search_tv_show_season(episodes): + episode = episodes["breaking_bad_s01e01"] + + with Subf2mProvider() as provider: + result = provider._search_tv_show_season(episode.series, episode.season) + assert result == "/subtitles/breaking-bad-first-season" + + +@pytest.mark.parametrize("language", [Language.fromalpha2("en"), Language("por", "BR")]) +def test_find_movie_subtitles(language): + path = "/subtitles/dune-2021" + with Subf2mProvider() as provider: + for sub in provider._find_movie_subtitles(path, language): + assert sub.language == language + + +@pytest.mark.parametrize("language", [Language.fromalpha2("en"), Language("por", "BR")]) +def test_find_episode_subtitles(language): + path = "/subtitles/breaking-bad-first-season" + with Subf2mProvider() as provider: + for sub in provider._find_episode_subtitles(path, 1, 1, language): + assert sub.language == language + + +@pytest.fixture +def subtitle(): + release_info = """Dune-2021.All.WEBDLL + Dune.2021.WEBRip.XviD.MP3-XVID + Dune.2021.WEBRip.XviD.MP3-SHITBOX + Dune.2021.WEBRip.x264-SHITBOX + Dune.2021.WEBRip.x264-ION10 + Dune.2021.HDRip.XviD-EVO[TGx] + Dune.2021.HDRip.XviD-EVO + Dune.2021.720p.HDRip.900MB.x264-GalaxyRG + Dune.2021.1080p.HDRip.X264-EVO + Dune.2021.1080p.HDRip.1400MB.x264-GalaxyRG""" + + return Subf2mSubtitle( + Language.fromalpha3b("per"), + "https://subf2m.co/subtitles/dune-2021/farsi_persian/2604701", + release_info, + ) + + +@pytest.fixture +def subtitle_episode(): + return Subf2mSubtitle( + Language.fromalpha2("en"), + "https://subf2m.co/subtitles/breaking-bad-first-season/english/161227", + "Breaking.Bad.S01E01-7.DSR-HDTV.eng", + ) + + +def test_subtitle_get_matches(subtitle, movies): + assert subtitle.get_matches(movies["dune"]) + + +def test_subtitle_get_matches_episode(subtitle_episode, episodes): + assert subtitle_episode.get_matches(episodes["breaking_bad_s01e01"]) + + +def test_list_subtitles_movie(movies): + with Subf2mProvider() as provider: + assert provider.list_subtitles(movies["dune"], {Language.fromalpha2("en")}) + + +def test_list_subtitles_episode(episodes): + with Subf2mProvider() as provider: + assert provider.list_subtitles( + episodes["breaking_bad_s01e01"], {Language.fromalpha2("en")} + ) + + +def test_download_subtitle(subtitle): + with Subf2mProvider() as provider: + provider.download_subtitle(subtitle) + assert subtitle.is_valid() + + +def test_download_subtitle_episode(subtitle_episode): + with Subf2mProvider() as provider: + provider.download_subtitle(subtitle_episode) + assert subtitle_episode.is_valid()