Adding new Sucha provider

README.md

@@ -66,6 +66,7 @@ If you need something that is not already part of Bazarr, feel free to create a
 * Subsunacs.net
 * subtitri.id.lv
 * Subtitulamos.tv
+* Sucha
 * Supersubtitles
 * Titlovi
 * Titrari.ro

libs/subliminal_patch/providers/sucha.py

@@ -0,0 +1,196 @@
+# -*- coding: utf-8 -*-
+import logging
+import os
+import io
+import rarfile
+import zipfile
+
+from requests import Session
+from subzero.language import Language
+
+from subliminal import Movie, Episode
+from subliminal_patch.exceptions import APIThrottled
+from subliminal.exceptions import ServiceUnavailable
+from subliminal_patch.subtitle import Subtitle
+from subliminal.subtitle import fix_line_ending, SUBTITLE_EXTENSIONS
+from subliminal_patch.providers import Provider
+
+logger = logging.getLogger(__name__)
+
+server_url = "http://sapi.caretas.club/"
+page_url = "https://sucha.caretas.club/"
+
+
+class SuchaSubtitle(Subtitle):
+    provider_name = "sucha"
+    hash_verifiable = False
+
+    def __init__(
+        self, language, page_link, filename, download_link, hearing_impaired, matches
+    ):
+        super(SuchaSubtitle, self).__init__(
+            language, hearing_impaired=hearing_impaired, page_link=page_url
+        )
+        self.download_link = download_link
+        self.referer = page_link
+        self.language = language
+        self.release_info = filename
+        self.filename = filename
+        self.found_matches = matches
+
+    @property
+    def id(self):
+        return self.download_link
+
+    def get_matches(self, video):
+        return self.found_matches
+
+
+class SuchaProvider(Provider):
+    """Sucha Provider"""
+
+    languages = {Language.fromalpha2(l) for l in ["es"]}
+    language_list = list(languages)
+    logger.debug(languages)
+    video_types = (Episode, Movie)
+
+    def initialize(self):
+        self.session = Session()
+        self.session.headers = {
+            "User-Agent": os.environ.get("SZ_USER_AGENT", "Sub-Zero/2")
+        }
+
+    def terminate(self):
+        self.session.close()
+
+    def query(self, languages, video):
+        movie_year = video.year if video.year else None
+        is_episode = True if isinstance(video, Episode) else False
+        imdb_id = video.imdb_id if video.imdb_id else None
+        language = self.language_list[0]
+        if is_episode:
+            q = {
+                "query": "{} S{:02}E{:02}".format(
+                    video.series, video.season, video.episode
+                )
+            }
+        else:
+            if imdb_id:
+                q = {"query": imdb_id}
+            else:
+                q = {"query": video.title, "year": movie_year}
+
+        logger.debug("Searching subtitles: {}".format(q["query"]))
+
+        res = self.session.get(server_url + "search", params=q, timeout=10)
+        res.raise_for_status()
+        result = res.json()
+
+        try:
+            subtitles = []
+            for i in result["results"]:
+                matches = set()
+                logger.debug(i["title"])
+                if video.title.lower() in i["title"].lower():
+                    matches.add("title")
+                if is_episode:
+                    if q["query"].lower() == i["title"].lower():
+                        matches.add("title")
+                        matches.add("series")
+                        matches.add("season")
+                        matches.add("episode")
+                        matches.add("year")
+                if i["year"] == video.year:
+                    matches.add("year")
+                if imdb_id:
+                    matches.add("imdb_id")
+                if (
+                    video.resolution
+                    and video.resolution.lower() in i["pseudo_file"].lower()
+                ):
+                    matches.add("resolution")
+
+                if video.source and video.source.lower() in i["pseudo_file"].lower():
+                    matches.add("source")
+
+                if video.video_codec:
+                    if (
+                        video.video_codec == "H.264"
+                        and "x264" in i["pseudo_file"].lower()
+                    ):
+                        matches.add("video_codec")
+                    elif (
+                        video.video_codec == "H.265"
+                        and "x265" in i["pseudo_file"].lower()
+                    ):
+                        matches.add("video_codec")
+                    elif video.video_codec.lower() in i["pseudo_file"].lower():
+                        matches.add("video_codec")
+
+                if video.audio_codec:
+                    if (
+                        video.audio_codec.lower().replace(" ", ".")
+                        in i["pseudo_file"].lower()
+                    ):
+                        matches.add("audio_codec")
+
+                subtitles.append(
+                    SuchaSubtitle(
+                        language,
+                        i["referer"],
+                        i["pseudo_file"],
+                        i["download_url"],
+                        i["hearing_impaired"],
+                        matches,
+                    )
+                )
+            return subtitles
+        except KeyError:
+            logger.debug("No subtitles found")
+            return []
+
+    def list_subtitles(self, video, languages):
+        return self.query(languages, video)
+
+    def _check_response(self, response):
+        if response.status_code != 200:
+            raise ServiceUnavailable("Bad status code: " + str(response.status_code))
+
+    def _get_archive(self, content):
+        archive_stream = io.BytesIO(content)
+        if rarfile.is_rarfile(archive_stream):
+            logger.debug("Identified rar archive")
+            archive = rarfile.RarFile(archive_stream)
+        elif zipfile.is_zipfile(archive_stream):
+            logger.debug("Identified zip archive")
+            archive = zipfile.ZipFile(archive_stream)
+        else:
+            raise APIThrottled("Unsupported compressed format")
+        return archive
+
+    def get_file(self, archive):
+        for name in archive.namelist():
+            if os.path.split(name)[-1].startswith("."):
+                continue
+            if not name.lower().endswith(SUBTITLE_EXTENSIONS):
+                continue
+            if (
+                "[eng]" in name.lower()
+                or ".en." in name.lower()
+                or ".eng." in name.lower()
+            ):
+                continue
+            logger.debug("Returning from archive: {}".format(name))
+            return archive.read(name)
+        raise APIThrottled("Can not find the subtitle in the compressed file")
+
+    def download_subtitle(self, subtitle):
+        logger.info("Downloading subtitle %r", subtitle)
+        response = self.session.get(
+            subtitle.download_link, headers={"Referer": subtitle.page_link}, timeout=10
+        )
+        response.raise_for_status()
+        self._check_response(response)
+        archive = self._get_archive(response.content)
+        subtitle_file = self.get_file(archive)
+        subtitle.content = fix_line_ending(subtitle_file)

views/settingsproviders.html

@@ -521,6 +521,18 @@
                 </div>
             </div>
 
+            <div class="row">
+                <div class="col-sm-3 text-right">
+                    <b>Sucha</b>
+                </div>
+                <div class="form-group col-sm-8">
+                    <label class="custom-control custom-checkbox">
+                        <input type="checkbox" class="custom-control-input provider" id="sucha">
+                        <span class="custom-control-label">Spanish subtitles provider.</span>
+                    </label>
+                </div>
+            </div>
+
             <div class="row">
                 <div class="col-sm-3 text-right">
                     <b>Supersubtitles</b>