mirror of https://github.com/morpheus65535/bazarr
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import
|
|
import json
|
|
import logging
|
|
import os
|
|
|
|
from babelfish import Language, language_converters
|
|
from requests import Session
|
|
|
|
from . import Provider
|
|
from .. import __short_version__
|
|
from ..subtitle import Subtitle, fix_line_ending
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
language_converters.register('shooter = subliminal.converters.shooter:ShooterConverter')
|
|
|
|
|
|
class ShooterSubtitle(Subtitle):
|
|
"""Shooter Subtitle."""
|
|
provider_name = 'shooter'
|
|
|
|
def __init__(self, language, hash, download_link):
|
|
super(ShooterSubtitle, self).__init__(language)
|
|
self.hash = hash
|
|
self.download_link = download_link
|
|
|
|
@property
|
|
def id(self):
|
|
return self.download_link
|
|
|
|
def get_matches(self, video):
|
|
matches = set()
|
|
|
|
# hash
|
|
if 'shooter' in video.hashes and video.hashes['shooter'] == self.hash:
|
|
matches.add('hash')
|
|
|
|
return matches
|
|
|
|
|
|
class ShooterProvider(Provider):
|
|
"""Shooter Provider."""
|
|
languages = {Language(l) for l in ['eng', 'zho']}
|
|
server_url = 'https://www.shooter.cn/api/subapi.php'
|
|
subtitle_class = ShooterSubtitle
|
|
|
|
def __init__(self):
|
|
self.session = None
|
|
|
|
def initialize(self):
|
|
self.session = Session()
|
|
self.session.headers['User-Agent'] = 'Subliminal/%s' % __short_version__
|
|
|
|
def terminate(self):
|
|
self.session.close()
|
|
|
|
def query(self, language, filename, hash=None):
|
|
# query the server
|
|
params = {'filehash': hash, 'pathinfo': os.path.realpath(filename), 'format': 'json', 'lang': language.shooter}
|
|
logger.debug('Searching subtitles %r', params)
|
|
r = self.session.post(self.server_url, params=params, timeout=10)
|
|
r.raise_for_status()
|
|
|
|
# handle subtitles not found
|
|
if r.content == b'\xff':
|
|
logger.debug('No subtitles found')
|
|
return []
|
|
|
|
# parse the subtitles
|
|
results = json.loads(r.text)
|
|
subtitles = [self.subtitle_class(language, hash, t['Link']) for s in results for t in s['Files']]
|
|
|
|
return subtitles
|
|
|
|
def list_subtitles(self, video, languages):
|
|
return [s for l in languages for s in self.query(l, video.name, video.hashes.get('shooter'))]
|
|
|
|
def download_subtitle(self, subtitle):
|
|
logger.info('Downloading subtitle %r', subtitle)
|
|
r = self.session.get(subtitle.download_link, timeout=10)
|
|
r.raise_for_status()
|
|
|
|
subtitle.content = fix_line_ending(r.content)
|