[hypem] Correctly handle cookies (Closes #7762)
[youtube-dl] / youtube_dl / extractor / hypem.py
1 from __future__ import unicode_literals
2
3 import json
4 import time
5
6 from .common import InfoExtractor
7 from ..compat import compat_urllib_parse
8 from ..utils import (
9     ExtractorError,
10     sanitized_Request,
11 )
12
13
14 class HypemIE(InfoExtractor):
15     _VALID_URL = r'http://(?:www\.)?hypem\.com/track/(?P<id>[^/]+)/'
16     _TEST = {
17         'url': 'http://hypem.com/track/1v6ga/BODYWORK+-+TAME',
18         'md5': 'b9cc91b5af8995e9f0c1cee04c575828',
19         'info_dict': {
20             'id': '1v6ga',
21             'ext': 'mp3',
22             'title': 'Tame',
23             'uploader': 'BODYWORK',
24         }
25     }
26
27     def _real_extract(self, url):
28         track_id = self._match_id(url)
29
30         data = {'ax': 1, 'ts': time.time()}
31         data_encoded = compat_urllib_parse.urlencode(data)
32         complete_url = url + "?" + data_encoded
33         request = sanitized_Request(complete_url)
34         response, urlh = self._download_webpage_handle(
35             request, track_id, 'Downloading webpage with the url')
36
37         html_tracks = self._html_search_regex(
38             r'(?ms)<script type="application/json" id="displayList-data">\s*(.*?)\s*</script>',
39             response, 'tracks')
40         try:
41             track_list = json.loads(html_tracks)
42             track = track_list['tracks'][0]
43         except ValueError:
44             raise ExtractorError('Hypemachine contained invalid JSON.')
45
46         key = track['key']
47         track_id = track['id']
48         artist = track['artist']
49         title = track['song']
50
51         serve_url = "http://hypem.com/serve/source/%s/%s" % (track_id, key)
52         request = sanitized_Request(
53             serve_url, '', {'Content-Type': 'application/json'})
54         song_data = self._download_json(request, track_id, 'Downloading metadata')
55         final_url = song_data["url"]
56
57         return {
58             'id': track_id,
59             'url': final_url,
60             'ext': 'mp3',
61             'title': title,
62             'uploader': artist,
63         }