b6021d4732
* Create a get_collection method similar to get_* methods for objects. * Fix title initialization of a collection * Remove the hack were both id and title were allowed when CDing. That hack only worked with the canalplus module, and failed with others like redmine (but they still showed in suggestions). Moreover, the canalplus module now has friendlier IDs so this is not really needed anymore. * Allow backends to tell if a path is valid or not. For instance, it now allows to cd in a Redmine project with no issues in it. It also won't display "404" for invalid project IDs. By default, we still use the unreliable method of checking there is at least one result in iter_resources(). * Fix cd completion to work with unicode strings (all strings after an unicode string were ignored!) * Do not suggest '..' when completing cd in the root refs #774
104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2010-2012 Nicolas Duhamel, Laurent Bachelier
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob 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 Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
import urllib
|
|
|
|
import lxml.etree
|
|
|
|
from weboob.tools.browser import BaseBrowser
|
|
from weboob.tools.browser.decorators import id2url
|
|
|
|
from .pages import InitPage, VideoPage
|
|
from .video import CanalplusVideo
|
|
|
|
from weboob.capabilities.collection import CollectionNotFound
|
|
|
|
__all__ = ['CanalplusBrowser']
|
|
|
|
|
|
class XMLParser(object):
|
|
def parse(self, data, encoding=None):
|
|
if encoding is None:
|
|
parser = None
|
|
else:
|
|
parser = lxml.etree.XMLParser(encoding=encoding, strip_cdata=False)
|
|
return lxml.etree.XML(data.get_data(), parser)
|
|
|
|
|
|
class CanalplusBrowser(BaseBrowser):
|
|
DOMAIN = u'service.canal-plus.com'
|
|
ENCODING = 'utf-8'
|
|
PAGES = {
|
|
r'http://service.canal-plus.com/video/rest/initPlayer/cplus/': InitPage,
|
|
r'http://service.canal-plus.com/video/rest/search/cplus/.*': VideoPage,
|
|
r'http://service.canal-plus.com/video/rest/getVideosLiees/cplus/(?P<id>.+)': VideoPage,
|
|
r'http://service.canal-plus.com/video/rest/getMEAs/cplus/.*': VideoPage,
|
|
}
|
|
|
|
#We need lxml.etree.XMLParser for read CDATA
|
|
PARSER = XMLParser()
|
|
FORMATS = {
|
|
'sd': 'BAS_DEBIT',
|
|
'hd': 'HD',
|
|
}
|
|
|
|
def __init__(self, quality, *args, **kwargs):
|
|
BaseBrowser.__init__(self, parser=self.PARSER, *args, **kwargs)
|
|
self.quality = self.FORMATS.get(quality, self.FORMATS['hd'])
|
|
|
|
def home(self):
|
|
self.location('http://service.canal-plus.com/video/rest/initPlayer/cplus/')
|
|
|
|
def search_videos(self, pattern):
|
|
self.location('http://service.canal-plus.com/video/rest/search/cplus/' + urllib.quote_plus(pattern.replace('/', '').encode('utf-8')))
|
|
return self.page.iter_results()
|
|
|
|
@id2url(CanalplusVideo.id2url)
|
|
def get_video(self, url, video=None):
|
|
self.location(url)
|
|
return self.page.get_video(video, self.quality)
|
|
|
|
def iter_resources(self, split_path):
|
|
if not self.is_on_page(InitPage):
|
|
self.home()
|
|
channels = self.page.get_channels()
|
|
|
|
if len(split_path) == 0:
|
|
for channel in channels:
|
|
if len(channel.split_path) == 1:
|
|
yield channel
|
|
elif len(split_path) == 1:
|
|
for channel in channels:
|
|
if len(channel.split_path) == 2 and split_path[0] == channel.split_path[0]:
|
|
yield channel
|
|
elif len(split_path) == 2:
|
|
subchannels = self.iter_resources(split_path[0:1])
|
|
try:
|
|
channel = [subchannel for subchannel in subchannels
|
|
if split_path == subchannel.split_path][0]
|
|
self.location("http://service.canal-plus.com/video/rest/getMEAs/cplus/%s" % channel._link_id)
|
|
assert self.is_on_page(VideoPage)
|
|
for video in self.page.iter_channel():
|
|
yield video
|
|
except IndexError:
|
|
raise CollectionNotFound(split_path)
|
|
|
|
else:
|
|
raise CollectionNotFound(split_path)
|