diff options
| author | Keith Beckman <kbeckman@becknet.com> | 2014-05-20 02:55:21 -0400 | 
|---|---|---|
| committer | Keith Beckman <kbeckman@becknet.com> | 2014-05-20 02:55:21 -0400 | 
| commit | 7ed806d24165a1ce909e2a1de340703392b216cd (patch) | |
| tree | e33d1a9cb99d13f4498c29b8c8287b0801edfa76 | |
| parent | dd06c95e43ebbef35e2ef8b589bd25a646166553 (diff) | |
Fixed pyflakes and pep8 warnings
| -rw-r--r-- | youtube_dl/extractor/grooveshark.py | 45 | 
1 files changed, 24 insertions, 21 deletions
| diff --git a/youtube_dl/extractor/grooveshark.py b/youtube_dl/extractor/grooveshark.py index 6798addce..165cccf69 100644 --- a/youtube_dl/extractor/grooveshark.py +++ b/youtube_dl/extractor/grooveshark.py @@ -11,6 +11,7 @@ from os.path import basename  from .common import InfoExtractor  from ..utils import ExtractorError, compat_urllib_request, compat_html_parser +  class GroovesharkHtmlParser(compat_html_parser.HTMLParser):      def __init__(self):          self._current_object = None @@ -20,10 +21,10 @@ class GroovesharkHtmlParser(compat_html_parser.HTMLParser):      def handle_starttag(self, tag, attrs):          attrs = dict((k, v) for k, v in attrs)          if tag == 'object': -            self._current_object = { 'attrs': attrs, 'params': [] } +            self._current_object = {'attrs': attrs, 'params': []}          elif tag == 'param':              self._current_object['params'].append(attrs) -     +      def handle_endtag(self, tag):          if tag == 'object':              self.objects.append(self._current_object) @@ -36,6 +37,7 @@ class GroovesharkHtmlParser(compat_html_parser.HTMLParser):          p.close()          return p.objects +  class GroovesharkIE(InfoExtractor):      _VALID_URL = r'https?://(www\.)?grooveshark\.com/#!/s/([^/]+)/([^/]+)'      _TEST = { @@ -104,7 +106,7 @@ class GroovesharkIE(InfoExtractor):      def _get_bootstrap(self, target):          (bootstrap_url, token) = self._build_bootstrap_url(target) -         +          headers = {'Referer': urldefrag(target)[0]}          req = compat_urllib_request.Request(bootstrap_url, headers=headers)          res = self._download_json(req, token, fatal=False, @@ -112,25 +114,25 @@ class GroovesharkIE(InfoExtractor):                                    errnote='Unable to download player bootstrap data',                                    transform_source=self._transform_bootstrap)          return res -     +      def _get_playerpage(self, target):          (_, _, token) = self._parse_target(target) -         +          res = 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']) -         +          return (res, None) -     +      def _real_extract(self, url):          (target_uri, _, token) = self._parse_target(url) -     +          # 1. Fill cookiejar by making a request to the player page          if self.do_playerpage_request:              (_, player_objs) = self._get_playerpage(url) @@ -142,10 +144,10 @@ class GroovesharkIE(InfoExtractor):          if self.do_bootstrap_request:              bootstrap = self._get_bootstrap(url)              self.to_screen('CommunicationToken: %s' % bootstrap['getCommunicationToken']) -         +          # 3. Ask preload.php for track metadata.          meta = self._get_meta(url) -         +          # 4. Construct stream request for track.          stream_url = self._build_stream_url(meta)          duration = int(math.ceil(float(meta['streamKey']['uSecs']) / 1000000)) @@ -154,13 +156,13 @@ class GroovesharkIE(InfoExtractor):          headers = {              'Content-Length': len(post_data),              'Content-Type': 'application/x-www-form-urlencoded' -            } +        }          if 'swf_referer' in locals(): -            headers['Referer'] = swf_referer         +            headers['Referer'] = swf_referer + +        req = compat_urllib_request.Request(stream_url, post_data, headers) -        req = compat_urllib_request.Request(streamurl, post_data, headers) -                  info_dict = {              'id': token,              'title': meta['song']['Name'], @@ -169,16 +171,18 @@ 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,              'request': req -            } -         +        } +          if 'swf_referer' in locals(): -            info_dict['http_referer'] = swf_referer         -         +            info_dict['http_referer'] = swf_referer +          return info_dict      def _real_initialize(self): @@ -197,4 +201,3 @@ class GroovesharkIE(InfoExtractor):              if fatal:                  raise ee          return None - | 
