You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
4.5 KiB
107 lines
4.5 KiB
13 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
# Copyright 2011-2012 Antoine Bertin <diaoulael@gmail.com>
|
||
|
|
#
|
||
|
|
# This file is part of subliminal.
|
||
|
|
#
|
||
|
|
# subliminal is free software; you can redistribute it and/or modify it under
|
||
|
|
# the terms of the GNU Lesser General Public License as published by
|
||
|
|
# the Free Software Foundation; either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# subliminal is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU Lesser General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU Lesser General Public License
|
||
|
|
# along with subliminal. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
from . import ServiceBase
|
||
|
|
from ..exceptions import ServiceError, DownloadFailedError
|
||
|
|
from ..subtitles import get_subtitle_path, ResultSubtitle
|
||
|
|
from ..utils import to_unicode
|
||
|
|
from ..videos import Episode, Movie
|
||
|
|
from guessit.language import lang_set
|
||
|
|
from hashlib import md5, sha256
|
||
|
|
import guessit
|
||
|
|
import logging
|
||
|
|
import xmlrpclib
|
||
|
|
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class Podnapisi(ServiceBase):
|
||
|
|
server_url = 'http://ssp.podnapisi.net:8000'
|
||
|
|
api_based = True
|
||
|
|
languages = lang_set(['sl', 'en', 'nn', 'ko', 'de', 'is', 'cs', 'fr', 'it', 'bs', 'jp', 'ar', 'ro',
|
||
|
|
'hu', 'gr', 'zh', 'lt', 'et', 'lv', 'he', 'nl', 'da', 'sv', 'pl', 'ru', 'es',
|
||
|
|
'sq', 'tr', 'fi', 'pt', 'bg', 'mk', 'sr', 'sk', 'hr', 'hi', 'th', 'ca', 'uk',
|
||
|
|
'pb', 'ga', 'be', 'vi', 'fa', 'ca', 'id', 'ms'])
|
||
|
|
#FIXME: ag and cyr not recognized by guessit
|
||
|
|
videos = [Episode, Movie]
|
||
|
|
require_video = True
|
||
|
|
|
||
|
|
def __init__(self, config=None):
|
||
|
|
super(Podnapisi, self).__init__(config)
|
||
|
|
self.server = xmlrpclib.ServerProxy(self.server_url)
|
||
|
|
self.token = None
|
||
|
|
|
||
|
|
def init(self):
|
||
|
|
super(Podnapisi, self).init()
|
||
|
|
result = self.server.initiate(self.user_agent)
|
||
|
|
if result['status'] != 200:
|
||
|
|
raise ServiceError('Initiate failed')
|
||
|
|
username = 'python_subliminal'
|
||
|
|
password = sha256(md5('XWFXQ6gE5Oe12rv4qxXX').hexdigest() + result['nonce']).hexdigest()
|
||
|
|
self.token = result['session']
|
||
|
|
result = self.server.authenticate(self.token, username, password)
|
||
|
|
if result['status'] != 200:
|
||
|
|
raise ServiceError('Authenticate failed')
|
||
|
|
|
||
|
|
def terminate(self):
|
||
|
|
super(Podnapisi, self).terminate()
|
||
|
|
|
||
|
|
def query(self, filepath, languages, moviehash):
|
||
|
|
results = self.server.search(self.token, [moviehash])
|
||
|
|
if results['status'] != 200:
|
||
|
|
logger.error('Search failed with error code %d' % results['status'])
|
||
|
|
return []
|
||
|
|
if not results['results'] or not results['results'][moviehash]['subtitles']:
|
||
|
|
logger.debug(u'Could not find subtitles for %r with token %s' % (moviehash, self.token))
|
||
|
|
return []
|
||
|
|
subtitles = []
|
||
|
|
for result in results['results'][moviehash]['subtitles']:
|
||
|
|
language = guessit.Language(result['lang'])
|
||
|
|
if language == guessit.language.UNDETERMINED or language not in languages:
|
||
|
|
continue
|
||
|
|
path = get_subtitle_path(filepath, language, self.config.multi)
|
||
|
|
subtitle = ResultSubtitle(path, language, service=self.__class__.__name__.lower(), link=result['id'],
|
||
|
|
release=to_unicode(result['release']), confidence=result['weight'])
|
||
|
|
subtitles.append(subtitle)
|
||
|
|
if not subtitles:
|
||
|
|
return []
|
||
|
|
# Convert weight to confidence
|
||
|
|
max_weight = float(max([s.confidence for s in subtitles]))
|
||
|
|
min_weight = float(min([s.confidence for s in subtitles]))
|
||
|
|
for subtitle in subtitles:
|
||
|
|
if max_weight == 0 and min_weight == 0:
|
||
|
|
subtitle.confidence = 1.0
|
||
|
|
else:
|
||
|
|
subtitle.confidence = (subtitle.confidence - min_weight) / (max_weight - min_weight)
|
||
|
|
return subtitles
|
||
|
|
|
||
|
|
def list_checked(self, video, languages):
|
||
|
|
results = self.query(video.path, languages, video.hashes['OpenSubtitles'])
|
||
|
|
return results
|
||
|
|
|
||
|
|
def download(self, subtitle):
|
||
|
|
results = self.server.download(self.token, [subtitle.link])
|
||
|
|
if results['status'] != 200:
|
||
|
|
raise DownloadFailedError()
|
||
|
|
subtitle.link = 'http://www.podnapisi.net/static/podnapisi/' + results['names'][0]['filename']
|
||
|
|
self.download_file(subtitle.link, subtitle.path)
|
||
|
|
return subtitle
|
||
|
|
|
||
|
|
|
||
|
|
Service = Podnapisi
|