[youtube] Fix extraction.
[youtube-dl] / youtube_dl / extractor / mixcloud.py
1 from __future__ import unicode_literals
2
3 import itertools
4 import re
5
6 from .common import InfoExtractor
7 from ..compat import (
8     compat_b64decode,
9     compat_chr,
10     compat_ord,
11     compat_str,
12     compat_urllib_parse_unquote,
13     compat_zip
14 )
15 from ..utils import (
16     int_or_none,
17     parse_iso8601,
18     strip_or_none,
19     try_get,
20 )
21
22
23 class MixcloudBaseIE(InfoExtractor):
24     def _call_api(self, object_type, object_fields, display_id, username, slug=None):
25         lookup_key = object_type + 'Lookup'
26         return self._download_json(
27             'https://www.mixcloud.com/graphql', display_id, query={
28                 'query': '''{
29   %s(lookup: {username: "%s"%s}) {
30     %s
31   }
32 }''' % (lookup_key, username, ', slug: "%s"' % slug if slug else '', object_fields)
33             })['data'][lookup_key]
34
35
36 class MixcloudIE(MixcloudBaseIE):
37     _VALID_URL = r'https?://(?:(?:www|beta|m)\.)?mixcloud\.com/([^/]+)/(?!stream|uploads|favorites|listens|playlists)([^/]+)'
38     IE_NAME = 'mixcloud'
39
40     _TESTS = [{
41         'url': 'http://www.mixcloud.com/dholbach/cryptkeeper/',
42         'info_dict': {
43             'id': 'dholbach_cryptkeeper',
44             'ext': 'm4a',
45             'title': 'Cryptkeeper',
46             'description': 'After quite a long silence from myself, finally another Drum\'n\'Bass mix with my favourite current dance floor bangers.',
47             'uploader': 'Daniel Holbach',
48             'uploader_id': 'dholbach',
49             'thumbnail': r're:https?://.*\.jpg',
50             'view_count': int,
51             'timestamp': 1321359578,
52             'upload_date': '20111115',
53         },
54     }, {
55         'url': 'http://www.mixcloud.com/gillespeterson/caribou-7-inch-vinyl-mix-chat/',
56         'info_dict': {
57             'id': 'gillespeterson_caribou-7-inch-vinyl-mix-chat',
58             'ext': 'mp3',
59             'title': 'Caribou 7 inch Vinyl Mix & Chat',
60             'description': 'md5:2b8aec6adce69f9d41724647c65875e8',
61             'uploader': 'Gilles Peterson Worldwide',
62             'uploader_id': 'gillespeterson',
63             'thumbnail': 're:https?://.*',
64             'view_count': int,
65             'timestamp': 1422987057,
66             'upload_date': '20150203',
67         },
68     }, {
69         'url': 'https://beta.mixcloud.com/RedLightRadio/nosedrip-15-red-light-radio-01-18-2016/',
70         'only_matching': True,
71     }]
72     _DECRYPTION_KEY = 'IFYOUWANTTHEARTISTSTOGETPAIDDONOTDOWNLOADFROMMIXCLOUD'
73
74     @staticmethod
75     def _decrypt_xor_cipher(key, ciphertext):
76         """Encrypt/Decrypt XOR cipher. Both ways are possible because it's XOR."""
77         return ''.join([
78             compat_chr(compat_ord(ch) ^ compat_ord(k))
79             for ch, k in compat_zip(ciphertext, itertools.cycle(key))])
80
81     def _real_extract(self, url):
82         username, slug = re.match(self._VALID_URL, url).groups()
83         username, slug = compat_urllib_parse_unquote(username), compat_urllib_parse_unquote(slug)
84         track_id = '%s_%s' % (username, slug)
85
86         cloudcast = self._call_api('cloudcast', '''audioLength
87     comments(first: 100) {
88       edges {
89         node {
90           comment
91           created
92           user {
93             displayName
94             username
95           }
96         }
97       }
98       totalCount
99     }
100     description
101     favorites {
102       totalCount
103     }
104     featuringArtistList
105     isExclusive
106     name
107     owner {
108       displayName
109       url
110       username
111     }
112     picture(width: 1024, height: 1024) {
113         url
114     }
115     plays
116     publishDate
117     reposts {
118       totalCount
119     }
120     streamInfo {
121       dashUrl
122       hlsUrl
123       url
124     }
125     tags {
126       tag {
127         name
128       }
129     }''', track_id, username, slug)
130
131         title = cloudcast['name']
132
133         stream_info = cloudcast['streamInfo']
134         formats = []
135
136         for url_key in ('url', 'hlsUrl', 'dashUrl'):
137             format_url = stream_info.get(url_key)
138             if not format_url:
139                 continue
140             decrypted = self._decrypt_xor_cipher(
141                 self._DECRYPTION_KEY, compat_b64decode(format_url))
142             if url_key == 'hlsUrl':
143                 formats.extend(self._extract_m3u8_formats(
144                     decrypted, track_id, 'mp4', entry_protocol='m3u8_native',
145                     m3u8_id='hls', fatal=False))
146             elif url_key == 'dashUrl':
147                 formats.extend(self._extract_mpd_formats(
148                     decrypted, track_id, mpd_id='dash', fatal=False))
149             else:
150                 formats.append({
151                     'format_id': 'http',
152                     'url': decrypted,
153                     'downloader_options': {
154                         # Mixcloud starts throttling at >~5M
155                         'http_chunk_size': 5242880,
156                     },
157                 })
158
159         if not formats and cloudcast.get('isExclusive'):
160             self.raise_login_required()
161
162         self._sort_formats(formats)
163
164         comments = []
165         for edge in (try_get(cloudcast, lambda x: x['comments']['edges']) or []):
166             node = edge.get('node') or {}
167             text = strip_or_none(node.get('comment'))
168             if not text:
169                 continue
170             user = node.get('user') or {}
171             comments.append({
172                 'author': user.get('displayName'),
173                 'author_id': user.get('username'),
174                 'text': text,
175                 'timestamp': parse_iso8601(node.get('created')),
176             })
177
178         tags = []
179         for t in cloudcast.get('tags'):
180             tag = try_get(t, lambda x: x['tag']['name'], compat_str)
181             if not tag:
182                 tags.append(tag)
183
184         get_count = lambda x: int_or_none(try_get(cloudcast, lambda y: y[x]['totalCount']))
185
186         owner = cloudcast.get('owner') or {}
187
188         return {
189             'id': track_id,
190             'title': title,
191             'formats': formats,
192             'description': cloudcast.get('description'),
193             'thumbnail': try_get(cloudcast, lambda x: x['picture']['url'], compat_str),
194             'uploader': owner.get('displayName'),
195             'timestamp': parse_iso8601(cloudcast.get('publishDate')),
196             'uploader_id': owner.get('username'),
197             'uploader_url': owner.get('url'),
198             'duration': int_or_none(cloudcast.get('audioLength')),
199             'view_count': int_or_none(cloudcast.get('plays')),
200             'like_count': get_count('favorites'),
201             'repost_count': get_count('reposts'),
202             'comment_count': get_count('comments'),
203             'comments': comments,
204             'tags': tags,
205             'artist': ', '.join(cloudcast.get('featuringArtistList') or []) or None,
206         }
207
208
209 class MixcloudPlaylistBaseIE(MixcloudBaseIE):
210     def _get_cloudcast(self, node):
211         return node
212
213     def _get_playlist_title(self, title, slug):
214         return title
215
216     def _real_extract(self, url):
217         username, slug = re.match(self._VALID_URL, url).groups()
218         username = compat_urllib_parse_unquote(username)
219         if not slug:
220             slug = 'uploads'
221         else:
222             slug = compat_urllib_parse_unquote(slug)
223         playlist_id = '%s_%s' % (username, slug)
224
225         is_playlist_type = self._ROOT_TYPE == 'playlist'
226         playlist_type = 'items' if is_playlist_type else slug
227         list_filter = ''
228
229         has_next_page = True
230         entries = []
231         while has_next_page:
232             playlist = self._call_api(
233                 self._ROOT_TYPE, '''%s
234     %s
235     %s(first: 100%s) {
236       edges {
237         node {
238           %s
239         }
240       }
241       pageInfo {
242         endCursor
243         hasNextPage
244       }
245     }''' % (self._TITLE_KEY, self._DESCRIPTION_KEY, playlist_type, list_filter, self._NODE_TEMPLATE),
246                 playlist_id, username, slug if is_playlist_type else None)
247
248             items = playlist.get(playlist_type) or {}
249             for edge in items.get('edges', []):
250                 cloudcast = self._get_cloudcast(edge.get('node') or {})
251                 cloudcast_url = cloudcast.get('url')
252                 if not cloudcast_url:
253                     continue
254                 entries.append(self.url_result(
255                     cloudcast_url, MixcloudIE.ie_key(), cloudcast.get('slug')))
256
257             page_info = items['pageInfo']
258             has_next_page = page_info['hasNextPage']
259             list_filter = ', after: "%s"' % page_info['endCursor']
260
261         return self.playlist_result(
262             entries, playlist_id,
263             self._get_playlist_title(playlist[self._TITLE_KEY], slug),
264             playlist.get(self._DESCRIPTION_KEY))
265
266
267 class MixcloudUserIE(MixcloudPlaylistBaseIE):
268     _VALID_URL = r'https?://(?:www\.)?mixcloud\.com/(?P<id>[^/]+)/(?P<type>uploads|favorites|listens|stream)?/?$'
269     IE_NAME = 'mixcloud:user'
270
271     _TESTS = [{
272         'url': 'http://www.mixcloud.com/dholbach/',
273         'info_dict': {
274             'id': 'dholbach_uploads',
275             'title': 'Daniel Holbach (uploads)',
276             'description': 'md5:b60d776f0bab534c5dabe0a34e47a789',
277         },
278         'playlist_mincount': 36,
279     }, {
280         'url': 'http://www.mixcloud.com/dholbach/uploads/',
281         'info_dict': {
282             'id': 'dholbach_uploads',
283             'title': 'Daniel Holbach (uploads)',
284             'description': 'md5:b60d776f0bab534c5dabe0a34e47a789',
285         },
286         'playlist_mincount': 36,
287     }, {
288         'url': 'http://www.mixcloud.com/dholbach/favorites/',
289         'info_dict': {
290             'id': 'dholbach_favorites',
291             'title': 'Daniel Holbach (favorites)',
292             'description': 'md5:b60d776f0bab534c5dabe0a34e47a789',
293         },
294         # 'params': {
295         #     'playlist_items': '1-100',
296         # },
297         'playlist_mincount': 396,
298     }, {
299         'url': 'http://www.mixcloud.com/dholbach/listens/',
300         'info_dict': {
301             'id': 'dholbach_listens',
302             'title': 'Daniel Holbach (listens)',
303             'description': 'md5:b60d776f0bab534c5dabe0a34e47a789',
304         },
305         # 'params': {
306         #     'playlist_items': '1-100',
307         # },
308         'playlist_mincount': 1623,
309         'skip': 'Large list',
310     }, {
311         'url': 'https://www.mixcloud.com/FirstEar/stream/',
312         'info_dict': {
313             'id': 'FirstEar_stream',
314             'title': 'First Ear (stream)',
315             'description': 'Curators of good music\r\n\r\nfirstearmusic.com',
316         },
317         'playlist_mincount': 271,
318     }]
319
320     _TITLE_KEY = 'displayName'
321     _DESCRIPTION_KEY = 'biog'
322     _ROOT_TYPE = 'user'
323     _NODE_TEMPLATE = '''slug
324           url'''
325
326     def _get_playlist_title(self, title, slug):
327         return '%s (%s)' % (title, slug)
328
329
330 class MixcloudPlaylistIE(MixcloudPlaylistBaseIE):
331     _VALID_URL = r'https?://(?:www\.)?mixcloud\.com/(?P<user>[^/]+)/playlists/(?P<playlist>[^/]+)/?$'
332     IE_NAME = 'mixcloud:playlist'
333
334     _TESTS = [{
335         'url': 'https://www.mixcloud.com/maxvibes/playlists/jazzcat-on-ness-radio/',
336         'info_dict': {
337             'id': 'maxvibes_jazzcat-on-ness-radio',
338             'title': 'Ness Radio sessions',
339         },
340         'playlist_mincount': 59,
341     }]
342     _TITLE_KEY = 'name'
343     _DESCRIPTION_KEY = 'description'
344     _ROOT_TYPE = 'playlist'
345     _NODE_TEMPLATE = '''cloudcast {
346             slug
347             url
348           }'''
349
350     def _get_cloudcast(self, node):
351         return node.get('cloudcast') or {}