morpheus65535
1 year ago
parent
906c2e9cb9
commit
aa0af3f601
@ -1,177 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import io
|
||||
import logging
|
||||
import os
|
||||
import zipfile
|
||||
|
||||
import rarfile
|
||||
from requests import Session
|
||||
from guessit import guessit
|
||||
from subliminal import Episode, Movie
|
||||
from subliminal.exceptions import ServiceUnavailable
|
||||
from subliminal.subtitle import SUBTITLE_EXTENSIONS, fix_line_ending
|
||||
from subliminal_patch.exceptions import APIThrottled
|
||||
from subliminal_patch.providers import Provider
|
||||
from subliminal_patch.subtitle import Subtitle, guess_matches
|
||||
from subzero.language import Language
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
SERVER_URL = "http://sapidb.caretas.club"
|
||||
PAGE_URL = "https://sucha.caretas.club"
|
||||
UNDESIRED_FILES = ("[eng]", ".en.", ".eng.", ".fr.", ".pt.")
|
||||
|
||||
|
||||
class SuchaSubtitle(Subtitle):
|
||||
provider_name = "sucha"
|
||||
hash_verifiable = False
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
language,
|
||||
release_info,
|
||||
filename,
|
||||
download_id,
|
||||
download_type,
|
||||
matches,
|
||||
):
|
||||
super(SuchaSubtitle, self).__init__(
|
||||
language, hearing_impaired=False, page_link=PAGE_URL
|
||||
)
|
||||
self.download_id = download_id
|
||||
self.download_type = download_type
|
||||
self.language = language
|
||||
self.guessed_release_info = release_info
|
||||
self.filename = filename
|
||||
self.release_info = (
|
||||
release_info if len(release_info) > len(filename) else filename
|
||||
)
|
||||
self.found_matches = matches
|
||||
|
||||
@property
|
||||
def id(self):
|
||||
return self.download_id
|
||||
|
||||
def get_matches(self, video):
|
||||
type_ = "episode" if isinstance(video, Episode) else "movie"
|
||||
self.found_matches |= guess_matches(
|
||||
video,
|
||||
guessit(self.filename, {"type": type_}),
|
||||
)
|
||||
self.found_matches |= guess_matches(
|
||||
video,
|
||||
guessit(self.guessed_release_info, {"type": type_}),
|
||||
)
|
||||
return self.found_matches
|
||||
|
||||
|
||||
class SuchaProvider(Provider):
|
||||
"""Sucha Provider"""
|
||||
|
||||
# This is temporary. Castilian spanish subtitles may exist, but are rare
|
||||
# and currently impossible to guess from the API.
|
||||
languages = {Language("spa", "MX")}
|
||||
language_list = list(languages)
|
||||
video_types = (Episode, Movie)
|
||||
|
||||
def initialize(self):
|
||||
self.session = Session()
|
||||
self.session.headers.update(
|
||||
{"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 or "0"
|
||||
is_episode = isinstance(video, Episode)
|
||||
type_str = "episode" if is_episode else "movie"
|
||||
language = self.language_list[0]
|
||||
|
||||
if is_episode:
|
||||
q = {"query": f"{video.series} S{video.season:02}E{video.episode:02}"}
|
||||
else:
|
||||
q = {"query": video.title, "year": movie_year}
|
||||
|
||||
logger.debug(f"Searching subtitles: {q}")
|
||||
result = self.session.get(f"{SERVER_URL}/{type_str}", params=q, timeout=10)
|
||||
result.raise_for_status()
|
||||
|
||||
results = result.json()
|
||||
if isinstance(results, dict):
|
||||
logger.debug("No subtitles found")
|
||||
return []
|
||||
|
||||
subtitles = []
|
||||
for item in results:
|
||||
matches = set()
|
||||
title = item.get("title", "").lower()
|
||||
alt_title = item.get("alt_title", title).lower()
|
||||
|
||||
if any(video.title.lower() in item for item in (title, alt_title)):
|
||||
matches.add("title")
|
||||
|
||||
if str(item["year"]) == video.year:
|
||||
matches.add("year")
|
||||
|
||||
if is_episode and any(
|
||||
q["query"].lower() in item for item in (title, alt_title)
|
||||
):
|
||||
matches.update(("title", "series", "season", "episode", "year"))
|
||||
|
||||
subtitles.append(
|
||||
SuchaSubtitle(
|
||||
language,
|
||||
item["release"],
|
||||
item["filename"],
|
||||
str(item["id"]),
|
||||
type_str,
|
||||
matches,
|
||||
)
|
||||
)
|
||||
return subtitles
|
||||
|
||||
def list_subtitles(self, video, languages):
|
||||
return self.query(languages, video)
|
||||
|
||||
def _get_archive(self, content):
|
||||
archive_stream = io.BytesIO(content)
|
||||
|
||||
if rarfile.is_rarfile(archive_stream):
|
||||
logger.debug("Identified rar archive")
|
||||
return rarfile.RarFile(archive_stream)
|
||||
|
||||
if zipfile.is_zipfile(archive_stream):
|
||||
logger.debug("Identified zip archive")
|
||||
return zipfile.ZipFile(archive_stream)
|
||||
|
||||
raise APIThrottled("Unsupported compressed format")
|
||||
|
||||
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 any(undesired in name.lower() for undesired in UNDESIRED_FILES):
|
||||
continue
|
||||
|
||||
logger.debug(f"Returning from archive: {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(
|
||||
f"{SERVER_URL}/download",
|
||||
params={"id": subtitle.download_id, "type": subtitle.download_type},
|
||||
timeout=10,
|
||||
)
|
||||
response.raise_for_status()
|
||||
archive = self._get_archive(response.content)
|
||||
subtitle_file = self.get_file(archive)
|
||||
subtitle.content = fix_line_ending(subtitle_file)
|
||||
Loading…
Reference in new issue