Merge remote-tracking branch 'Dineshs91/f4m-2.0'
[youtube-dl] / youtube_dl / extractor / grooveshark.py
index 77c5a91721ecdde64e6b6815ad71c32532ea8e72..fff74a70a891fc163ff488408e4df348564b8a29 100644 (file)
@@ -3,18 +3,18 @@ from __future__ import unicode_literals
 
 import time
 import math
+import os.path
 import re
 
-from urllib import quote, urlencode
-from os.path import basename
 
 from .common import InfoExtractor
-from ..utils import ExtractorError, compat_urllib_request, compat_html_parser
-
-from ..utils import compat_urlparse
-urlparse = compat_urlparse.urlparse
-urlunparse = compat_urlparse.urlunparse
-urldefrag = compat_urlparse.urldefrag
+from ..compat import (
+    compat_html_parser,
+    compat_urllib_parse,
+    compat_urllib_request,
+    compat_urlparse,
+)
+from ..utils import ExtractorError
 
 
 class GroovesharkHtmlParser(compat_html_parser.HTMLParser):
@@ -47,12 +47,12 @@ class GroovesharkIE(InfoExtractor):
     _VALID_URL = r'https?://(www\.)?grooveshark\.com/#!/s/([^/]+)/([^/]+)'
     _TEST = {
         'url': 'http://grooveshark.com/#!/s/Jolene+Tenth+Key+Remix+Ft+Will+Sessions/6SS1DW?src=5',
-        'md5': 'bbccc50b19daca23b8f961152c1dc95b',
+        'md5': '7ecf8aefa59d6b2098517e1baa530023',
         'info_dict': {
             'id': '6SS1DW',
             'title': 'Jolene (Tenth Key Remix ft. Will Sessions)',
             'ext': 'mp3',
-            'duration': 227
+            'duration': 227,
         }
     }
 
@@ -60,27 +60,27 @@ class GroovesharkIE(InfoExtractor):
     do_bootstrap_request = True
 
     def _parse_target(self, target):
-        uri = urlparse(target)
+        uri = compat_urlparse.urlparse(target)
         hash = uri.fragment[1:].split('?')[0]
-        token = basename(hash.rstrip('/'))
+        token = os.path.basename(hash.rstrip('/'))
         return (uri, hash, token)
 
     def _build_bootstrap_url(self, target):
         (uri, hash, token) = self._parse_target(target)
-        query = 'getCommunicationToken=1&hash=%s&%d' % (quote(hash, safe=''), self.ts)
-        return (urlunparse((uri.scheme, uri.netloc, '/preload.php', None, query, None)), token)
+        query = 'getCommunicationToken=1&hash=%s&%d' % (compat_urllib_parse.quote(hash, safe=''), self.ts)
+        return (compat_urlparse.urlunparse((uri.scheme, uri.netloc, '/preload.php', None, query, None)), token)
 
     def _build_meta_url(self, target):
         (uri, hash, token) = self._parse_target(target)
-        query = 'hash=%s&%d' % (quote(hash, safe=''), self.ts)
-        return (urlunparse((uri.scheme, uri.netloc, '/preload.php', None, query, None)), token)
+        query = 'hash=%s&%d' % (compat_urllib_parse.quote(hash, safe=''), self.ts)
+        return (compat_urlparse.urlunparse((uri.scheme, uri.netloc, '/preload.php', None, query, None)), token)
 
     def _build_stream_url(self, meta):
-        return urlunparse(('http', meta['streamKey']['ip'], '/stream.php', None, None, None))
+        return compat_urlparse.urlunparse(('http', meta['streamKey']['ip'], '/stream.php', None, None, None))
 
     def _build_swf_referer(self, target, obj):
         (uri, _, _) = self._parse_target(target)
-        return urlunparse((uri.scheme, uri.netloc, obj['attrs']['data'], None, None, None))
+        return compat_urlparse.urlunparse((uri.scheme, uri.netloc, obj['attrs']['data'], None, None, None))
 
     def _transform_bootstrap(self, js):
         return re.split('(?m)^\s*try\s*{', js)[0] \
@@ -93,7 +93,7 @@ class GroovesharkIE(InfoExtractor):
         (meta_url, token) = self._build_meta_url(target)
         self.to_screen('Metadata URL: %s' % meta_url)
 
-        headers = {'Referer': urldefrag(target)[0]}
+        headers = {'Referer': compat_urlparse.urldefrag(target)[0]}
         req = compat_urllib_request.Request(meta_url, headers=headers)
         res = self._download_json(req, token,
                                   transform_source=self._transform_meta)
@@ -112,7 +112,7 @@ class GroovesharkIE(InfoExtractor):
     def _get_bootstrap(self, target):
         (bootstrap_url, token) = self._build_bootstrap_url(target)
 
-        headers = {'Referer': urldefrag(target)[0]}
+        headers = {'Referer': compat_urlparse.urldefrag(target)[0]}
         req = compat_urllib_request.Request(bootstrap_url, headers=headers)
         res = self._download_json(req, token, fatal=False,
                                   note='Downloading player bootstrap data',
@@ -123,22 +123,35 @@ class GroovesharkIE(InfoExtractor):
     def _get_playerpage(self, target):
         (_, _, token) = self._parse_target(target)
 
-        res = self._download_webpage(
+        webpage = self._download_webpage(
             target, token,
             note='Downloading player page',
             errnote='Unable to download player page',
             fatal=False)
 
-        if res is not None:
-            o = GroovesharkHtmlParser.extract_object_tags(res)
-            return (res, [x for x in o if x['attrs']['id'] == 'jsPlayerEmbed'])
+        if webpage is not None:
+            # Search (for example German) error message
+            error_msg = self._html_search_regex(
+                r'<div id="content">\s*<h2>(.*?)</h2>', webpage,
+                'error message', default=None)
+            if error_msg is not None:
+                error_msg = error_msg.replace('\n', ' ')
+                raise ExtractorError('Grooveshark said: %s' % error_msg)
+
+        if webpage is not None:
+            o = GroovesharkHtmlParser.extract_object_tags(webpage)
+            return (webpage, [x for x in o if x['attrs']['id'] == 'jsPlayerEmbed'])
 
-        return (res, None)
+        return (webpage, None)
+
+    def _real_initialize(self):
+        self.ts = int(time.time() * 1000)  # timestamp in millis
 
     def _real_extract(self, url):
         (target_uri, _, token) = self._parse_target(url)
 
         # 1. Fill cookiejar by making a request to the player page
+        swf_referer = None
         if self.do_playerpage_request:
             (_, player_objs) = self._get_playerpage(url)
             if player_objs is not None:
@@ -157,16 +170,15 @@ class GroovesharkIE(InfoExtractor):
         stream_url = self._build_stream_url(meta)
         duration = int(math.ceil(float(meta['streamKey']['uSecs']) / 1000000))
         post_dict = {'streamKey': meta['streamKey']['streamKey']}
-        post_data = urlencode(post_dict).encode('utf-8')
+        post_data = compat_urllib_parse.urlencode(post_dict).encode('utf-8')
         headers = {
             'Content-Length': len(post_data),
             'Content-Type': 'application/x-www-form-urlencoded'
         }
-
-        if 'swf_referer' in locals():
+        if swf_referer is not None:
             headers['Referer'] = swf_referer
 
-        info_dict = {
+        return {
             'id': token,
             'title': meta['song']['Name'],
             'http_method': 'POST',
@@ -174,32 +186,6 @@ class GroovesharkIE(InfoExtractor):
             'ext': 'mp3',
             'format': 'mp3 audio',
             'duration': duration,
-
-            # various ways of supporting the download request.
-            # remove keys unnecessary to the eventual post implementation
-            'post_data': post_data,
-            'post_dict': post_dict,
-            'headers': headers
+            'http_post_data': post_data,
+            'http_headers': headers,
         }
-
-        if 'swf_referer' in locals():
-            info_dict['http_referer'] = swf_referer
-
-        return info_dict
-
-    def _real_initialize(self):
-        self.ts = int(time.time() * 1000)  # timestamp in millis
-
-    def _download_json(self, url_or_request, video_id,
-                       note=u'Downloading JSON metadata',
-                       errnote=u'Unable to download JSON metadata',
-                       fatal=True,
-                       transform_source=None):
-        try:
-            out = super(GroovesharkIE, self)._download_json(
-                url_or_request, video_id, note, errnote, transform_source)
-            return out
-        except ExtractorError as ee:
-            if fatal:
-                raise ee
-        return None