mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-16 06:15:09 -05:00
100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Copyright 2013 Julien Goret <jgoret@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
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
import logging
|
||
|
|
import urllib
|
||
|
|
|
||
|
|
logger = logging.getLogger("subliminal")
|
||
|
|
|
||
|
|
class Usub(ServiceBase):
|
||
|
|
server_url = 'http://www.u-sub.net/sous-titres'
|
||
|
|
site_url = 'http://www.u-sub.net/'
|
||
|
|
api_based = False
|
||
|
|
languages = language_set(['fr'])
|
||
|
|
videos = [Episode]
|
||
|
|
require_video = False
|
||
|
|
#required_features = ['permissive']
|
||
|
|
|
||
|
|
def list_checked(self, video, languages):
|
||
|
|
return self.query(video.path or video.release, languages, get_keywords(video.guess), series=video.series, season=video.season, episode=video.episode)
|
||
|
|
|
||
|
|
def query(self, filepath, languages, keywords=None, series=None, season=None, episode=None):
|
||
|
|
|
||
|
|
## Check if we really got informations about our episode
|
||
|
|
if series and season and episode:
|
||
|
|
request_series = series.lower().replace(' ', '-')
|
||
|
|
if isinstance(request_series, unicode):
|
||
|
|
request_series = request_series.encode('utf-8')
|
||
|
|
logger.debug(u'Getting subtitles for %s season %d episode %d with language %r' % (series, season, episode, languages))
|
||
|
|
r = self.session.get('%s/%s/saison_%s' % (self.server_url, urllib.quote(request_series),season))
|
||
|
|
if r.status_code == 404:
|
||
|
|
print "Error 404"
|
||
|
|
logger.debug(u'Could not find subtitles for %s' % (series))
|
||
|
|
return []
|
||
|
|
else:
|
||
|
|
print "One or more parameter missing"
|
||
|
|
raise ServiceError('One or more parameter missing')
|
||
|
|
|
||
|
|
## Check if we didn't got an big and nasty http error
|
||
|
|
if r.status_code != 200:
|
||
|
|
print u'Request %s returned status code %d' % (r.url, r.status_code)
|
||
|
|
logger.error(u'Request %s returned status code %d' % (r.url, r.status_code))
|
||
|
|
return []
|
||
|
|
|
||
|
|
## Editing episode informations to be able to use it with our search
|
||
|
|
if episode < 10 :
|
||
|
|
episode_num='0'+str(episode)
|
||
|
|
else :
|
||
|
|
episode_num=str(episode)
|
||
|
|
season_num = str(season)
|
||
|
|
series_name = series.lower().replace(' ', '.')
|
||
|
|
possible_episode_naming = [season_num+'x'+episode_num,season_num+episode_num]
|
||
|
|
|
||
|
|
|
||
|
|
## Actually parsing the page for the good subtitles
|
||
|
|
soup = BeautifulSoup(r.content, self.required_features)
|
||
|
|
subtitles = []
|
||
|
|
subtitles_list = soup.find('table', {'id' : 'subtitles_list'})
|
||
|
|
link_list = subtitles_list.findAll('a', {'class' : 'dl_link'})
|
||
|
|
|
||
|
|
for link in link_list :
|
||
|
|
link_url = link.get('href')
|
||
|
|
splited_link = link_url.split('/')
|
||
|
|
filename = splited_link[len(splited_link)-1]
|
||
|
|
for episode_naming in possible_episode_naming :
|
||
|
|
if episode_naming in filename :
|
||
|
|
for language in languages:
|
||
|
|
path = get_subtitle_path(filepath, language, self.config.multi)
|
||
|
|
subtitle = ResultSubtitle(path, language, self.__class__.__name__.lower(), '%s' % (link_url))
|
||
|
|
subtitles.append(subtitle)
|
||
|
|
return subtitles
|
||
|
|
|
||
|
|
def download(self, subtitle):
|
||
|
|
## All downloaded files are zip files
|
||
|
|
self.download_zip_file(subtitle.link, subtitle.path)
|
||
|
|
return subtitle
|
||
|
|
|
||
|
|
|
||
|
|
Service = Usub
|