diff options
Diffstat (limited to 'pyload/plugins/crypter/DailymotionBatch.py')
| -rw-r--r-- | pyload/plugins/crypter/DailymotionBatch.py | 98 | 
1 files changed, 98 insertions, 0 deletions
| diff --git a/pyload/plugins/crypter/DailymotionBatch.py b/pyload/plugins/crypter/DailymotionBatch.py new file mode 100644 index 000000000..2c818d8bc --- /dev/null +++ b/pyload/plugins/crypter/DailymotionBatch.py @@ -0,0 +1,98 @@ +# -*- coding: utf-8 -*- + +import re + +from urlparse import urljoin + +from pyload.utils import json_loads +from pyload.plugins.Crypter import Crypter +from pyload.utils import safe_join + + +class DailymotionBatch(Crypter): +    __name__ = "DailymotionBatch" +    __type__ = "crypter" +    __version__ = "0.01" + +    __pattern__ = r'https?://(?:www\.)?dailymotion\.com/((playlists/)?(?P<TYPE>playlist|user)/)?(?P<ID>[\w^_]+)(?(TYPE)|#)' + +    __description__ = """Dailymotion.com channel & playlist decrypter""" +    __author_name__ = "Walter Purcaro" +    __author_mail__ = "vuolter@gmail.com" + + +    def api_response(self, ref, req=None): +        url = urljoin("https://api.dailymotion.com/", ref) +        page = self.load(url, get=req) +        return json_loads(page) + +    def getPlaylistInfo(self, id): +        ref = "playlist/" + id +        req = {"fields": "name,owner.screenname"} +        playlist = self.api_response(ref, req) + +        if "error" in playlist: +            return + +        name = playlist['name'] +        owner = playlist['owner.screenname'] +        return name, owner + +    def _getPlaylists(self, user_id, page=1): +        ref = "user/%s/playlists" % user_id +        req = {"fields": "id", "page": page, "limit": 100} +        user = self.api_response(ref, req) + +        if "error" in user: +            return + +        for playlist in user['list']: +            yield playlist['id'] + +        if user['has_more']: +            for item in self._getPlaylists(user_id, page + 1): +                yield item + +    def getPlaylists(self, user_id): +        return [(id,) + self.getPlaylistInfo(id) for id in self._getPlaylists(user_id)] + +    def _getVideos(self, id, page=1): +        ref = "playlist/%s/videos" % id +        req = {"fields": "url", "page": page, "limit": 100} +        playlist = self.api_response(ref, req) + +        if "error" in playlist: +            return + +        for video in playlist['list']: +            yield video['url'] + +        if playlist['has_more']: +            for item in self._getVideos(id, page + 1): +                yield item + +    def getVideos(self, playlist_id): +        return list(self._getVideos(playlist_id))[::-1] + +    def decrypt(self, pyfile): +        m = re.match(self.__pattern__, pyfile.url) +        m_id = m.group("ID") +        m_type = m.group("TYPE") + +        if m_type == "playlist": +            self.logDebug("Url recognized as Playlist") +            p_info = self.getPlaylistInfo(m_id) +            playlists = [(m_id,) + p_info] if p_info else None +        else: +            self.logDebug("Url recognized as Channel") +            playlists = self.getPlaylists(m_id) +            self.logDebug("%s playlist\s found on channel \"%s\"" % (len(playlists), m_id)) + +        if not playlists: +            self.fail("No playlist available") + +        for p_id, p_name, p_owner in playlists: +            p_videos = self.getVideos(p_id) +            p_folder = safe_join(self.config['general']['download_folder'], p_owner, p_name) +            self.logDebug("%s video\s found on playlist \"%s\"" % (len(p_videos), p_name)) +            self.packages.append((p_name, p_videos, p_folder))  #: folder is NOT recognized by pyload 0.4.9! | 
