mirror of https://github.com/morpheus65535/bazarr
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
import logging
|
|
|
|
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('thesubdb = subliminal.converters.thesubdb:TheSubDBConverter')
|
|
|
|
|
|
class TheSubDBSubtitle(Subtitle):
|
|
"""TheSubDB Subtitle."""
|
|
provider_name = 'thesubdb'
|
|
|
|
def __init__(self, language, hash):
|
|
super(TheSubDBSubtitle, self).__init__(language)
|
|
self.hash = hash
|
|
|
|
@property
|
|
def id(self):
|
|
return self.hash + '-' + str(self.language)
|
|
|
|
def get_matches(self, video):
|
|
matches = set()
|
|
|
|
# hash
|
|
if 'thesubdb' in video.hashes and video.hashes['thesubdb'] == self.hash:
|
|
matches.add('hash')
|
|
|
|
return matches
|
|
|
|
|
|
class TheSubDBProvider(Provider):
|
|
"""TheSubDB Provider."""
|
|
languages = {Language.fromthesubdb(l) for l in language_converters['thesubdb'].codes}
|
|
required_hash = 'thesubdb'
|
|
server_url = 'http://api.thesubdb.com/'
|
|
subtitle_class = TheSubDBSubtitle
|
|
|
|
def __init__(self):
|
|
self.session = None
|
|
|
|
def initialize(self):
|
|
self.session = Session()
|
|
self.session.headers['User-Agent'] = ('SubDB/1.0 (subliminal/%s; https://github.com/Diaoul/subliminal)' %
|
|
__short_version__)
|
|
|
|
def terminate(self):
|
|
self.session.close()
|
|
|
|
def query(self, hash):
|
|
# make the query
|
|
params = {'action': 'search', 'hash': hash}
|
|
logger.info('Searching subtitles %r', params)
|
|
r = self.session.get(self.server_url, params=params, timeout=10)
|
|
|
|
# handle subtitles not found and errors
|
|
if r.status_code == 404:
|
|
logger.debug('No subtitles found')
|
|
return []
|
|
r.raise_for_status()
|
|
|
|
# loop over languages
|
|
subtitles = []
|
|
for language_code in r.text.split(','):
|
|
language = Language.fromthesubdb(language_code)
|
|
|
|
subtitle = self.subtitle_class(language, hash)
|
|
logger.debug('Found subtitle %r', subtitle)
|
|
subtitles.append(subtitle)
|
|
|
|
return subtitles
|
|
|
|
def list_subtitles(self, video, languages):
|
|
return [s for s in self.query(video.hashes['thesubdb']) if s.language in languages]
|
|
|
|
def download_subtitle(self, subtitle):
|
|
logger.info('Downloading subtitle %r', subtitle)
|
|
params = {'action': 'download', 'hash': subtitle.hash, 'language': subtitle.language.alpha2}
|
|
r = self.session.get(self.server_url, params=params, timeout=10)
|
|
r.raise_for_status()
|
|
|
|
subtitle.content = fix_line_ending(r.content)
|