mirror of https://github.com/morpheus65535/bazarr
159 lines
5.6 KiB
Python
159 lines
5.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
import logging
|
|
import os
|
|
|
|
from requests import Session
|
|
from subzero.language import Language
|
|
|
|
from subliminal import Movie, Episode, ProviderError, __short_version__
|
|
from subliminal_patch.subtitle import Subtitle, guess_matches
|
|
from subliminal.providers import ParserBeautifulSoup
|
|
from subliminal.subtitle import fix_line_ending, SUBTITLE_EXTENSIONS
|
|
from subliminal_patch.providers import Provider
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SubtitulamosTVSubtitle(Subtitle):
|
|
provider_name = 'subtitulamostv'
|
|
hash_verifiable = False
|
|
|
|
def __init__(self, language, page_link, download_link, title, release_info):
|
|
super(SubtitulamosTVSubtitle, self).__init__(language, hearing_impaired=False, page_link=page_link)
|
|
self.download_link = download_link
|
|
self.title = title
|
|
self.release_info = release_info
|
|
|
|
@property
|
|
def id(self):
|
|
return self.download_link
|
|
|
|
def get_matches(self, video):
|
|
matches = {'series', 'season', 'episode', 'year'}
|
|
|
|
title_lower = self.title.lower()
|
|
release_info_lower = self.release_info.lower()
|
|
|
|
if video.title and video.title.lower() in title_lower:
|
|
matches.add('title')
|
|
|
|
if video.release_group and video.release_group.lower() in release_info_lower:
|
|
matches.add('release_group')
|
|
|
|
if video.resolution and video.resolution.lower() in release_info_lower:
|
|
matches.add('resolution')
|
|
|
|
if video.source:
|
|
formats = [video.source.lower()]
|
|
if formats[0] == "web":
|
|
formats.append("webdl")
|
|
formats.append("web-dl")
|
|
formats.append("webrip")
|
|
for frmt in formats:
|
|
if frmt in release_info_lower:
|
|
matches.add('source')
|
|
break
|
|
|
|
if video.video_codec:
|
|
video_codecs = [video.video_codec.lower()]
|
|
if video_codecs[0] == "h.264":
|
|
video_codecs.append("h264")
|
|
video_codecs.append("x264")
|
|
elif video_codecs[0] == "h.265":
|
|
video_codecs.append("h265")
|
|
video_codecs.append("x265")
|
|
for vc in video_codecs:
|
|
if vc in release_info_lower:
|
|
matches.add('video_codec')
|
|
break
|
|
|
|
return matches
|
|
|
|
|
|
class SubtitulamosTVProvider(Provider):
|
|
"""Subtitulamostv Provider"""
|
|
languages = {Language.fromietf(lang) for lang in ['en', 'es']}
|
|
video_types = (Episode,)
|
|
|
|
server_url = 'https://subtitulamos.tv'
|
|
|
|
def __init__(self):
|
|
self.session = None
|
|
|
|
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):
|
|
subtitle_name = "%s %dx%02d" % (video.series, video.season, video.episode)
|
|
logger.debug('Searching subtitles "%s"' % subtitle_name)
|
|
|
|
response = self.session.get(self.server_url + '/search/query', params={'q': video.series}, timeout=10)
|
|
response.raise_for_status()
|
|
result = response.json()
|
|
|
|
subtitles = []
|
|
for serie in result:
|
|
# skip non-matching series
|
|
if video.series.lower() != serie['name'].lower():
|
|
continue
|
|
|
|
response = self.session.get(self.server_url + "/shows/%d/season/%d" % (serie['id'], video.season),
|
|
timeout=10)
|
|
response.raise_for_status()
|
|
soup = ParserBeautifulSoup(response.text, ['lxml', 'html.parser'])
|
|
|
|
for episode in soup.select('div.episode'):
|
|
episode_soup = episode.find('a')
|
|
episode_name = episode_soup.text
|
|
episode_url = episode_soup['href']
|
|
|
|
# skip non-matching episodes
|
|
if subtitle_name.lower() not in episode_name.lower():
|
|
continue
|
|
|
|
for lang in episode.select("div.subtitle-language"):
|
|
if "English" in lang.text:
|
|
language = "en"
|
|
elif "Español" in lang.text:
|
|
language = "es"
|
|
else:
|
|
continue # not supported yet
|
|
logger.debug('Found subtitles in "%s" language.', language)
|
|
|
|
for release in lang.find_next_sibling("div").select("div.sub"):
|
|
release_name = release.select('div.version-name')[0].text
|
|
release_url = release.select('a[href*="/download"]')[0]['href']
|
|
|
|
subtitles.append(
|
|
SubtitulamosTVSubtitle(
|
|
Language.fromietf(language),
|
|
self.server_url + episode_url,
|
|
self.server_url + release_url,
|
|
episode_name,
|
|
release_name
|
|
)
|
|
)
|
|
|
|
return subtitles
|
|
|
|
def list_subtitles(self, video, languages):
|
|
return self.query(languages, video)
|
|
|
|
def download_subtitle(self, subtitle):
|
|
logger.info('Downloading subtitle %r', subtitle)
|
|
r = self.session.get(subtitle.download_link, timeout=10)
|
|
r.raise_for_status()
|
|
|
|
r.encoding = "ISO-8859-1"
|
|
subtitle_content = r.content
|
|
|
|
if subtitle_content:
|
|
subtitle.content = fix_line_ending(subtitle_content)
|
|
else:
|
|
logger.debug('Could not download subtitle from %s', subtitle.download_link)
|