Adding new Sucha provider

pull/1127/head
vitiko98 4 years ago
parent bd7854326f
commit 2e797e100b

@ -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

@ -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)

@ -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>

Loading…
Cancel
Save