mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-25 14:25:05 +00:00
105 lines
5.3 KiB
Python
105 lines
5.3 KiB
Python
# -*- 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
|
|
from ..language import language_set, Language
|
|
from ..subtitles import get_subtitle_path, ResultSubtitle
|
|
from ..utils import get_keywords, split_keyword
|
|
from ..videos import Episode, Movie
|
|
from bs4 import BeautifulSoup
|
|
import logging
|
|
|
|
from _23 import quote
|
|
from six import PY2, text_type
|
|
|
|
|
|
logger = logging.getLogger("subliminal")
|
|
|
|
|
|
class SubsWiki(ServiceBase):
|
|
server_url = 'http://www.subswiki.com'
|
|
site_url = 'http://www.subswiki.com'
|
|
api_based = False
|
|
languages = language_set(['eng-US', 'eng-GB', 'eng', 'fre', 'pob', 'por', 'spa-ES', u'spa', u'ita', u'cat'])
|
|
language_map = {u'Español': Language('spa'), u'Español (España)': Language('spa'), u'Español (Latinoamérica)': Language('spa'),
|
|
u'Català': Language('cat'), u'Brazilian': Language('pob'), u'English (US)': Language('eng-US'),
|
|
u'English (UK)': Language('eng-GB')}
|
|
language_code = 'name'
|
|
videos = [Episode, Movie]
|
|
require_video = False
|
|
required_features = ['permissive']
|
|
|
|
def list_checked(self, video, languages):
|
|
results = []
|
|
if isinstance(video, Episode):
|
|
results = self.query(video.path or video.release, languages, get_keywords(video.guess), series=video.series,
|
|
season=video.season, episode=video.episode)
|
|
elif isinstance(video, Movie) and video.year:
|
|
results = self.query(video.path or video.release, languages, get_keywords(video.guess), movie=video.title,
|
|
year=video.year)
|
|
return results
|
|
|
|
def query(self, filepath, languages, keywords=None, series=None, season=None, episode=None, movie=None, year=None):
|
|
if series and season and episode:
|
|
request_series = series.lower().replace(' ', '_')
|
|
if PY2 and isinstance(request_series, text_type):
|
|
request_series = request_series.encode('utf-8')
|
|
logger.debug(u'Getting subtitles for %s season %d episode %d with languages %r' % (series, season, episode, languages))
|
|
r = self.session.get('%s/serie/%s/%s/%s/' % (self.server_url, quote(request_series), season, episode))
|
|
if r.status_code == 404:
|
|
logger.debug(u'Could not find subtitles for %s season %d episode %d with languages %r' % (series, season, episode, languages))
|
|
return []
|
|
elif movie and year:
|
|
request_movie = movie.title().replace(' ', '_')
|
|
if PY2 and isinstance(request_movie, text_type):
|
|
request_movie = request_movie.encode('utf-8')
|
|
logger.debug(u'Getting subtitles for %s (%d) with languages %r' % (movie, year, languages))
|
|
r = self.session.get('%s/film/%s_(%d)' % (self.server_url, quote(request_movie), year))
|
|
if r.status_code == 404:
|
|
logger.debug(u'Could not find subtitles for %s (%d) with languages %r' % (movie, year, languages))
|
|
return []
|
|
else:
|
|
raise ServiceError('One or more parameter missing')
|
|
if r.status_code != 200:
|
|
logger.error(u'Request %s returned status code %d' % (r.url, r.status_code))
|
|
return []
|
|
soup = BeautifulSoup(r.content, self.required_features)
|
|
subtitles = []
|
|
for sub in soup('td', {'class': 'NewsTitle'}):
|
|
sub_keywords = split_keyword(sub.b.string.lower())
|
|
if keywords and not keywords & sub_keywords:
|
|
logger.debug(u'None of subtitle keywords %r in %r' % (sub_keywords, keywords))
|
|
continue
|
|
for html_language in sub.parent.parent.find_all('td', {'class': 'language'}):
|
|
language = self.get_language(html_language.string.strip())
|
|
if language not in languages:
|
|
logger.debug(u'Language %r not in wanted languages %r' % (language, languages))
|
|
continue
|
|
html_status = html_language.find_next_sibling('td')
|
|
status = html_status.strong.string.strip()
|
|
if status != 'Completado':
|
|
logger.debug(u'Wrong subtitle status %s' % status)
|
|
continue
|
|
path = get_subtitle_path(filepath, language, self.config.multi)
|
|
subtitle = ResultSubtitle(path, language, self.__class__.__name__.lower(), '%s%s' %
|
|
(self.server_url, html_status.find_next('td').find('a')['href']))
|
|
subtitles.append(subtitle)
|
|
return subtitles
|
|
|
|
|
|
Service = SubsWiki
|