]> jfr.im git - yt-dlp.git/blame - youtube_dl/extractor/soundcloud.py
[prosiebensat1] improve geo restriction handling(closes #23571)
[yt-dlp.git] / youtube_dl / extractor / soundcloud.py
CommitLineData
dcdb292f 1# coding: utf-8
fbcd7b5f
PH
2from __future__ import unicode_literals
3
92790f4e 4import itertools
73602bcd 5import re
aad0d6d5 6
2abf7cab 7from .common import (
8 InfoExtractor,
9 SearchInfoExtractor
10)
1cc79574 11from ..compat import (
aad0d6d5 12 compat_str,
668de34c 13 compat_urlparse,
1cc79574
PH
14)
15from ..utils import (
aad0d6d5 16 ExtractorError,
e09965d5 17 float_or_none,
548c3957 18 HEADRequest,
eb920777 19 int_or_none,
e09965d5 20 KNOWN_EXTENSIONS,
e09965d5
S
21 mimetype2ext,
22 str_or_none,
f516f440
S
23 try_get,
24 unified_timestamp,
d7c7100e 25 update_url_query,
7c5307f4 26 url_or_none,
d7c7100e 27)
aad0d6d5
PH
28
29
548c3957 30class SoundcloudEmbedIE(InfoExtractor):
cf80ff18
RA
31 _VALID_URL = r'https?://(?:w|player|p)\.soundcloud\.com/player/?.*?\burl=(?P<id>.+)'
32 _TEST = {
33 # from https://www.soundi.fi/uutiset/ennakkokuuntelussa-timo-kaukolammen-station-to-station-to-station-julkaisua-juhlitaan-tanaan-g-livelabissa/
34 'url': 'https://w.soundcloud.com/player/?visual=true&url=https%3A%2F%2Fapi.soundcloud.com%2Fplaylists%2F922213810&show_artwork=true&maxwidth=640&maxheight=960&dnt=1&secret_token=s-ziYey',
35 'only_matching': True,
36 }
548c3957
RA
37
38 @staticmethod
39 def _extract_urls(webpage):
40 return [m.group('url') for m in re.finditer(
41 r'<iframe[^>]+src=(["\'])(?P<url>(?:https?://)?(?:w\.)?soundcloud\.com/player.+?)\1',
42 webpage)]
43
44 def _real_extract(self, url):
cf80ff18
RA
45 query = compat_urlparse.parse_qs(
46 compat_urlparse.urlparse(url).query)
47 api_url = query['url'][0]
48 secret_token = query.get('secret_token')
49 if secret_token:
50 api_url = update_url_query(api_url, {'secret_token': secret_token[0]})
51 return self.url_result(api_url)
548c3957
RA
52
53
aad0d6d5
PH
54class SoundcloudIE(InfoExtractor):
55 """Information extractor for soundcloud.com
56 To access the media, the uid of the song and a stream token
57 must be extracted from the page source and the script must make
58 a request to media.soundcloud.com/crossdomain.xml. Then
59 the media can be grabbed by requesting from an url composed
60 of the stream token and uid
61 """
62
20991253 63 _VALID_URL = r'''(?x)^(?:https?://)?
71507a11 64 (?:(?:(?:www\.|m\.)?soundcloud\.com/
836ef264 65 (?!stations/track)
4ff50ef8 66 (?P<uploader>[\w\d-]+)/
3ef2da2d 67 (?!(?:tracks|albums|sets(?:/.+?)?|reposts|likes|spotlight)/?(?:$|[?#]))
22a6f150 68 (?P<title>[\w\d-]+)/?
de2dd4c5 69 (?P<token>[^?]+?)?(?:[?].*)?$)
548c3957 70 |(?:api(?:-v2)?\.soundcloud\.com/tracks/(?P<track_id>\d+)
0403b069 71 (?:/?\?secret_token=(?P<secret_token>[^&]+))?)
eb6a41ba
JMF
72 )
73 '''
fbcd7b5f 74 IE_NAME = 'soundcloud'
12c167c8
JMF
75 _TESTS = [
76 {
fbcd7b5f 77 'url': 'http://soundcloud.com/ethmusic/lostin-powers-she-so-heavy',
fbcd7b5f
PH
78 'md5': 'ebef0a451b909710ed1d7787dddbf0d7',
79 'info_dict': {
0eb9fb9f
JMF
80 'id': '62986583',
81 'ext': 'mp3',
f516f440 82 'title': 'Lostin Powers - She so Heavy (SneakPreview) Adrian Ackers Blueprint 1',
0eb9fb9f
JMF
83 'description': 'No Downloads untill we record the finished version this weekend, i was too pumped n i had to post it , earl is prolly gonna b hella p.o\'d',
84 'uploader': 'E.T. ExTerrestrial Music',
548c3957 85 'uploader_id': '1571244',
f516f440
S
86 'timestamp': 1349920598,
87 'upload_date': '20121011',
e09965d5 88 'duration': 143.216,
4bfd294e 89 'license': 'all-rights-reserved',
f516f440
S
90 'view_count': int,
91 'like_count': int,
92 'comment_count': int,
93 'repost_count': int,
12c167c8
JMF
94 }
95 },
96 # not streamable song
97 {
fbcd7b5f
PH
98 'url': 'https://soundcloud.com/the-concept-band/goldrushed-mastered?in=the-concept-band/sets/the-royal-concept-ep',
99 'info_dict': {
100 'id': '47127627',
101 'ext': 'mp3',
102 'title': 'Goldrushed',
63ad0315 103 'description': 'From Stockholm Sweden\r\nPovel / Magnus / Filip / David\r\nwww.theroyalconcept.com',
fbcd7b5f 104 'uploader': 'The Royal Concept',
548c3957 105 'uploader_id': '9615865',
f516f440 106 'timestamp': 1337635207,
fbcd7b5f 107 'upload_date': '20120521',
f516f440 108 'duration': 30,
4bfd294e 109 'license': 'all-rights-reserved',
f516f440
S
110 'view_count': int,
111 'like_count': int,
112 'comment_count': int,
113 'repost_count': int,
12c167c8 114 },
fbcd7b5f 115 'params': {
12c167c8 116 # rtmp
fbcd7b5f 117 'skip_download': True,
12c167c8 118 },
548c3957 119 'skip': 'Preview',
12c167c8 120 },
de2dd4c5
JMF
121 # private link
122 {
fbcd7b5f
PH
123 'url': 'https://soundcloud.com/jaimemf/youtube-dl-test-video-a-y-baw/s-8Pjrp',
124 'md5': 'aa0dd32bfea9b0c5ef4f02aacd080604',
125 'info_dict': {
126 'id': '123998367',
127 'ext': 'mp3',
128 'title': 'Youtube - Dl Test Video \'\' Ä↭',
fbcd7b5f 129 'description': 'test chars: \"\'/\\ä↭',
f516f440 130 'uploader': 'jaimeMF',
548c3957 131 'uploader_id': '69767071',
f516f440 132 'timestamp': 1386604920,
fbcd7b5f 133 'upload_date': '20131209',
e09965d5 134 'duration': 9.927,
4bfd294e 135 'license': 'all-rights-reserved',
f516f440
S
136 'view_count': int,
137 'like_count': int,
138 'comment_count': int,
139 'repost_count': int,
de2dd4c5
JMF
140 },
141 },
9296738f 142 # private link (alt format)
143 {
144 'url': 'https://api.soundcloud.com/tracks/123998367?secret_token=s-8Pjrp',
145 'md5': 'aa0dd32bfea9b0c5ef4f02aacd080604',
146 'info_dict': {
147 'id': '123998367',
148 'ext': 'mp3',
149 'title': 'Youtube - Dl Test Video \'\' Ä↭',
9296738f 150 'description': 'test chars: \"\'/\\ä↭',
f516f440 151 'uploader': 'jaimeMF',
548c3957 152 'uploader_id': '69767071',
f516f440 153 'timestamp': 1386604920,
9296738f 154 'upload_date': '20131209',
e09965d5 155 'duration': 9.927,
4bfd294e 156 'license': 'all-rights-reserved',
f516f440
S
157 'view_count': int,
158 'like_count': int,
159 'comment_count': int,
160 'repost_count': int,
9296738f 161 },
162 },
f67ca84d
JMF
163 # downloadable song
164 {
00a82ea8 165 'url': 'https://soundcloud.com/oddsamples/bus-brakes',
eae12e3f 166 'md5': '7624f2351f8a3b2e7cd51522496e7631',
fbcd7b5f 167 'info_dict': {
00a82ea8 168 'id': '128590877',
eae12e3f 169 'ext': 'mp3',
00a82ea8 170 'title': 'Bus Brakes',
0eb9fb9f 171 'description': 'md5:0053ca6396e8d2fd7b7e1595ef12ab66',
00a82ea8 172 'uploader': 'oddsamples',
548c3957 173 'uploader_id': '73680509',
f516f440 174 'timestamp': 1389232924,
00a82ea8 175 'upload_date': '20140109',
e09965d5 176 'duration': 17.346,
4bfd294e 177 'license': 'cc-by-sa',
f516f440
S
178 'view_count': int,
179 'like_count': int,
180 'comment_count': int,
181 'repost_count': int,
f67ca84d
JMF
182 },
183 },
d7c7100e
S
184 # private link, downloadable format
185 {
186 'url': 'https://soundcloud.com/oriuplift/uponly-238-no-talking-wav/s-AyZUd',
187 'md5': '64a60b16e617d41d0bef032b7f55441e',
188 'info_dict': {
189 'id': '340344461',
190 'ext': 'wav',
191 'title': 'Uplifting Only 238 [No Talking] (incl. Alex Feed Guestmix) (Aug 31, 2017) [wav]',
192 'description': 'md5:fa20ee0fca76a3d6df8c7e57f3715366',
193 'uploader': 'Ori Uplift Music',
548c3957 194 'uploader_id': '12563093',
f516f440 195 'timestamp': 1504206263,
d7c7100e 196 'upload_date': '20170831',
e09965d5 197 'duration': 7449.096,
d7c7100e 198 'license': 'all-rights-reserved',
f516f440
S
199 'view_count': int,
200 'like_count': int,
201 'comment_count': int,
202 'repost_count': int,
d7c7100e
S
203 },
204 },
0b0870f9
PV
205 # no album art, use avatar pic for thumbnail
206 {
207 'url': 'https://soundcloud.com/garyvee/sideways-prod-mad-real',
208 'md5': '59c7872bc44e5d99b7211891664760c2',
209 'info_dict': {
210 'id': '309699954',
211 'ext': 'mp3',
212 'title': 'Sideways (Prod. Mad Real)',
213 'description': 'md5:d41d8cd98f00b204e9800998ecf8427e',
214 'uploader': 'garyvee',
548c3957 215 'uploader_id': '2366352',
f516f440 216 'timestamp': 1488152409,
0b0870f9 217 'upload_date': '20170226',
e09965d5 218 'duration': 207.012,
0b0870f9
PV
219 'thumbnail': r're:https?://.*\.jpg',
220 'license': 'all-rights-reserved',
f516f440
S
221 'view_count': int,
222 'like_count': int,
223 'comment_count': int,
224 'repost_count': int,
0b0870f9
PV
225 },
226 'params': {
227 'skip_download': True,
228 },
229 },
843ad179 230 # not available via api.soundcloud.com/i1/tracks/id/streams
e09965d5
S
231 {
232 'url': 'https://soundcloud.com/giovannisarani/mezzo-valzer',
233 'md5': 'e22aecd2bc88e0e4e432d7dcc0a1abf7',
234 'info_dict': {
235 'id': '583011102',
236 'ext': 'mp3',
237 'title': 'Mezzo Valzer',
238 'description': 'md5:4138d582f81866a530317bae316e8b61',
239 'uploader': 'Giovanni Sarani',
548c3957 240 'uploader_id': '3352531',
e09965d5
S
241 'timestamp': 1551394171,
242 'upload_date': '20190228',
243 'duration': 180.157,
244 'thumbnail': r're:https?://.*\.jpg',
245 'license': 'all-rights-reserved',
246 'view_count': int,
247 'like_count': int,
248 'comment_count': int,
249 'repost_count': int,
250 },
251 'expected_warnings': ['Unable to download JSON metadata'],
252 }
12c167c8 253 ]
aad0d6d5 254
548c3957
RA
255 _API_BASE = 'https://api.soundcloud.com/'
256 _API_V2_BASE = 'https://api-v2.soundcloud.com/'
257 _BASE_URL = 'https://soundcloud.com/'
d1b27220 258 _CLIENT_ID = 'YUKXoArFcqrlQn9tfNHvvyfnDISj04zk'
548c3957
RA
259 _IMAGE_REPL_RE = r'-([0-9a-z]+)\.jpg'
260
261 _ARTWORK_MAP = {
262 'mini': 16,
263 'tiny': 20,
264 'small': 32,
265 'badge': 47,
266 't67x67': 67,
267 'large': 100,
268 't300x300': 300,
269 'crop': 400,
270 't500x500': 500,
271 'original': 0,
272 }
fbdf8d15 273
7d239269
JMF
274 @classmethod
275 def _resolv_url(cls, url):
548c3957 276 return SoundcloudIE._API_V2_BASE + 'resolve?url=' + url + '&client_id=' + cls._CLIENT_ID
7d239269 277
548c3957 278 def _extract_info_dict(self, info, full_title=None, secret_token=None, version=2):
12c167c8 279 track_id = compat_str(info['id'])
f516f440 280 title = info['title']
548c3957 281 track_base_url = self._API_BASE + 'tracks/%s' % track_id
e09965d5
S
282
283 format_urls = set()
5e114e4b 284 formats = []
73602bcd 285 query = {'client_id': self._CLIENT_ID}
548c3957 286 if secret_token:
73602bcd 287 query['secret_token'] = secret_token
548c3957 288
2e9ad59a 289 if info.get('downloadable') and info.get('has_downloads_left'):
73602bcd 290 format_url = update_url_query(
548c3957 291 info.get('download_url') or track_base_url + '/download', query)
e09965d5 292 format_urls.add(format_url)
548c3957
RA
293 if version == 2:
294 v1_info = self._download_json(
295 track_base_url, track_id, query=query, fatal=False) or {}
296 else:
297 v1_info = info
5e114e4b 298 formats.append({
2a15e706 299 'format_id': 'download',
548c3957
RA
300 'ext': v1_info.get('original_format') or 'mp3',
301 'filesize': int_or_none(v1_info.get('original_content_size')),
2a15e706 302 'url': format_url,
5e114e4b
PH
303 'preference': 10,
304 })
305
548c3957
RA
306 def invalid_url(url):
307 return not url or url in format_urls or re.search(r'/(?:preview|playlist)/0/30/', url)
e09965d5 308
548c3957
RA
309 def add_format(f, protocol):
310 mobj = re.search(r'\.(?P<abr>\d+)\.(?P<ext>[0-9a-z]{3,4})(?=[/?])', stream_url)
311 if mobj:
312 for k, v in mobj.groupdict().items():
313 if not f.get(k):
314 f[k] = v
315 format_id_list = []
316 if protocol:
317 format_id_list.append(protocol)
318 for k in ('ext', 'abr'):
319 v = f.get(k)
320 if v:
321 format_id_list.append(v)
322 abr = f.get('abr')
323 if abr:
324 f['abr'] = int(abr)
325 f.update({
326 'format_id': '_'.join(format_id_list),
327 'protocol': 'm3u8_native' if protocol == 'hls' else 'http',
328 })
329 formats.append(f)
e09965d5
S
330
331 # New API
332 transcodings = try_get(
333 info, lambda x: x['media']['transcodings'], list) or []
334 for t in transcodings:
335 if not isinstance(t, dict):
3a194cb4 336 continue
e09965d5 337 format_url = url_or_none(t.get('url'))
548c3957 338 if not format_url or t.get('snipped') or '/preview/' in format_url:
e09965d5
S
339 continue
340 stream = self._download_json(
548c3957 341 format_url, track_id, query=query, fatal=False)
e09965d5
S
342 if not isinstance(stream, dict):
343 continue
344 stream_url = url_or_none(stream.get('url'))
548c3957 345 if invalid_url(stream_url):
e09965d5
S
346 continue
347 format_urls.add(stream_url)
548c3957
RA
348 stream_format = t.get('format') or {}
349 protocol = stream_format.get('protocol')
e09965d5
S
350 if protocol != 'hls' and '/hls' in format_url:
351 protocol = 'hls'
352 ext = None
353 preset = str_or_none(t.get('preset'))
354 if preset:
355 ext = preset.split('_')[0]
548c3957
RA
356 if ext not in KNOWN_EXTENSIONS:
357 ext = mimetype2ext(stream_format.get('mime_type'))
358 add_format({
e09965d5 359 'url': stream_url,
e09965d5 360 'ext': ext,
548c3957
RA
361 }, 'http' if protocol == 'progressive' else protocol)
362
363 if not formats:
364 # Old API, does not work for some tracks (e.g.
365 # https://soundcloud.com/giovannisarani/mezzo-valzer)
366 # and might serve preview URLs (e.g.
367 # http://www.soundcloud.com/snbrn/ele)
368 format_dict = self._download_json(
369 track_base_url + '/streams', track_id,
370 'Downloading track url', query=query, fatal=False) or {}
371
372 for key, stream_url in format_dict.items():
373 if invalid_url(stream_url):
374 continue
375 format_urls.add(stream_url)
376 mobj = re.search(r'(http|hls)_([^_]+)_(\d+)_url', key)
377 if mobj:
378 protocol, ext, abr = mobj.groups()
379 add_format({
380 'abr': abr,
381 'ext': ext,
382 'url': stream_url,
383 }, protocol)
3a194cb4
S
384
385 if not formats:
386 # We fallback to the stream_url in the original info, this
387 # cannot be always used, sometimes it can give an HTTP 404 error
548c3957
RA
388 urlh = self._request_webpage(
389 HEADRequest(info.get('stream_url') or track_base_url + '/stream'),
390 track_id, query=query, fatal=False)
391 if urlh:
392 stream_url = urlh.geturl()
393 if not invalid_url(stream_url):
394 add_format({'url': stream_url}, 'http')
3a194cb4
S
395
396 for f in formats:
397 f['vcodec'] = 'none'
2a15e706 398
562ceab1 399 self._sort_formats(formats)
64bb5187 400
548c3957
RA
401 user = info.get('user') or {}
402
403 thumbnails = []
404 artwork_url = info.get('artwork_url')
405 thumbnail = artwork_url or user.get('avatar_url')
406 if isinstance(thumbnail, compat_str):
407 if re.search(self._IMAGE_REPL_RE, thumbnail):
408 for image_id, size in self._ARTWORK_MAP.items():
409 i = {
410 'id': image_id,
411 'url': re.sub(self._IMAGE_REPL_RE, '-%s.jpg' % image_id, thumbnail),
412 }
413 if image_id == 'tiny' and not artwork_url:
414 size = 18
415 elif image_id == 'original':
416 i['preference'] = 10
417 if size:
418 i.update({
419 'width': size,
420 'height': size,
421 })
422 thumbnails.append(i)
423 else:
424 thumbnails = [{'url': thumbnail}]
425
426 def extract_count(key):
427 return int_or_none(info.get('%s_count' % key))
428
429 return {
430 'id': track_id,
431 'uploader': user.get('username'),
432 'uploader_id': str_or_none(user.get('id')) or user.get('permalink'),
433 'uploader_url': user.get('permalink_url'),
434 'timestamp': unified_timestamp(info.get('created_at')),
435 'title': title,
436 'description': info.get('description'),
437 'thumbnails': thumbnails,
438 'duration': float_or_none(info.get('duration'), 1000),
439 'webpage_url': info.get('permalink_url'),
440 'license': info.get('license'),
441 'view_count': extract_count('playback'),
442 'like_count': extract_count('favoritings') or extract_count('likes'),
443 'comment_count': extract_count('comment'),
444 'repost_count': extract_count('reposts'),
445 'genre': info.get('genre'),
446 'formats': formats
447 }
7d239269 448
aad0d6d5 449 def _real_extract(self, url):
548c3957 450 mobj = re.match(self._VALID_URL, url)
aad0d6d5 451
eb6a41ba 452 track_id = mobj.group('track_id')
4bfd294e 453
548c3957
RA
454 query = {
455 'client_id': self._CLIENT_ID,
456 }
457 if track_id:
458 info_json_url = self._API_V2_BASE + 'tracks/' + track_id
eb6a41ba 459 full_title = track_id
9296738f 460 token = mobj.group('secret_token')
461 if token:
548c3957 462 query['secret_token'] = token
eb6a41ba 463 else:
548c3957 464 full_title = resolve_title = '%s/%s' % mobj.group('uploader', 'title')
de2dd4c5 465 token = mobj.group('token')
de2dd4c5
JMF
466 if token:
467 resolve_title += '/%s' % token
548c3957 468 info_json_url = self._resolv_url(self._BASE_URL + resolve_title)
5f6a1245 469
548c3957 470 version = 2
e09965d5 471 info = self._download_json(
548c3957
RA
472 info_json_url, full_title, 'Downloading info JSON', query=query, fatal=False)
473 if not info:
474 info = self._download_json(
475 info_json_url.replace(self._API_V2_BASE, self._API_BASE),
476 full_title, 'Downloading info JSON', query=query)
477 version = 1
e09965d5 478
548c3957 479 return self._extract_info_dict(info, full_title, token, version)
aad0d6d5 480
20991253 481
7518a61d 482class SoundcloudPlaylistBaseIE(SoundcloudIE):
548c3957
RA
483 def _extract_track_entries(self, tracks, token=None):
484 entries = []
485 for track in tracks:
486 track_id = str_or_none(track.get('id'))
487 url = track.get('permalink_url')
488 if not url:
489 if not track_id:
490 continue
491 url = self._API_V2_BASE + 'tracks/' + track_id
492 if token:
493 url += '?secret_token=' + token
494 entries.append(self.url_result(
495 url, SoundcloudIE.ie_key(), track_id))
496 return entries
8e45e1cc
S
497
498
7518a61d 499class SoundcloudSetIE(SoundcloudPlaylistBaseIE):
c808ef81 500 _VALID_URL = r'https?://(?:(?:www|m)\.)?soundcloud\.com/(?P<uploader>[\w\d-]+)/sets/(?P<slug_title>[\w\d-]+)(?:/(?P<token>[^?/]+))?'
fbcd7b5f 501 IE_NAME = 'soundcloud:set'
22a6f150
PH
502 _TESTS = [{
503 'url': 'https://soundcloud.com/the-concept-band/sets/the-royal-concept-ep',
504 'info_dict': {
a9551e90 505 'id': '2284613',
22a6f150
PH
506 'title': 'The Royal Concept EP',
507 },
bf2dc9cc 508 'playlist_mincount': 5,
f7043ef3
S
509 }, {
510 'url': 'https://soundcloud.com/the-concept-band/sets/the-royal-concept-ep/token',
511 'only_matching': True,
22a6f150 512 }]
aad0d6d5 513
aad0d6d5
PH
514 def _real_extract(self, url):
515 mobj = re.match(self._VALID_URL, url)
aad0d6d5 516
548c3957 517 full_title = '%s/sets/%s' % mobj.group('uploader', 'slug_title')
2f834e93 518 token = mobj.group('token')
519 if token:
520 full_title += '/' + token
aad0d6d5 521
548c3957
RA
522 info = self._download_json(self._resolv_url(
523 self._BASE_URL + full_title), full_title)
aad0d6d5 524
aad0d6d5 525 if 'errors' in info:
214e74bf
JMF
526 msgs = (compat_str(err['error_message']) for err in info['errors'])
527 raise ExtractorError('unable to download video webpage: %s' % ','.join(msgs))
aad0d6d5 528
548c3957 529 entries = self._extract_track_entries(info['tracks'], token)
b14fa8e6 530
548c3957
RA
531 return self.playlist_result(
532 entries, str_or_none(info.get('id')), info.get('title'))
92790f4e
JMF
533
534
836ef264 535class SoundcloudPagedPlaylistBaseIE(SoundcloudPlaylistBaseIE):
836ef264
S
536 def _extract_playlist(self, base_url, playlist_id, playlist_title):
537 COMMON_QUERY = {
548c3957 538 'limit': 2000000000,
836ef264
S
539 'client_id': self._CLIENT_ID,
540 'linked_partitioning': '1',
541 }
542
543 query = COMMON_QUERY.copy()
544 query['offset'] = 0
545
548c3957 546 next_href = base_url
836ef264
S
547
548 entries = []
549 for i in itertools.count():
550 response = self._download_json(
548c3957
RA
551 next_href, playlist_id,
552 'Downloading track page %s' % (i + 1), query=query)
836ef264
S
553
554 collection = response['collection']
3ef2da2d
S
555
556 if not isinstance(collection, list):
557 collection = []
558
559 # Empty collection may be returned, in this case we proceed
560 # straight to next_href
836ef264 561
7c5307f4
S
562 def resolve_entry(candidates):
563 for cand in candidates:
564 if not isinstance(cand, dict):
565 continue
566 permalink_url = url_or_none(cand.get('permalink_url'))
567 if not permalink_url:
568 continue
569 return self.url_result(
570 permalink_url,
548c3957
RA
571 SoundcloudIE.ie_key() if SoundcloudIE.suitable(permalink_url) else None,
572 str_or_none(cand.get('id')), cand.get('title'))
836ef264
S
573
574 for e in collection:
7c5307f4
S
575 entry = resolve_entry((e, e.get('track'), e.get('playlist')))
576 if entry:
577 entries.append(entry)
836ef264
S
578
579 next_href = response.get('next_href')
580 if not next_href:
581 break
582
548c3957
RA
583 next_href = response['next_href']
584 parsed_next_href = compat_urlparse.urlparse(next_href)
585 query = compat_urlparse.parse_qs(parsed_next_href.query)
586 query.update(COMMON_QUERY)
836ef264
S
587
588 return {
589 '_type': 'playlist',
590 'id': playlist_id,
591 'title': playlist_title,
592 'entries': entries,
593 }
594
595
596class SoundcloudUserIE(SoundcloudPagedPlaylistBaseIE):
16a08978
S
597 _VALID_URL = r'''(?x)
598 https?://
599 (?:(?:www|m)\.)?soundcloud\.com/
600 (?P<user>[^/]+)
601 (?:/
3ef2da2d 602 (?P<rsrc>tracks|albums|sets|reposts|likes|spotlight)
16a08978
S
603 )?
604 /?(?:[?#].*)?$
605 '''
fbcd7b5f 606 IE_NAME = 'soundcloud:user'
22a6f150 607 _TESTS = [{
b6423e6c 608 'url': 'https://soundcloud.com/soft-cell-official',
22a6f150 609 'info_dict': {
b6423e6c
S
610 'id': '207965082',
611 'title': 'Soft Cell (All)',
22a6f150 612 },
b6423e6c 613 'playlist_mincount': 28,
22a6f150 614 }, {
b6423e6c 615 'url': 'https://soundcloud.com/soft-cell-official/tracks',
22a6f150 616 'info_dict': {
b6423e6c
S
617 'id': '207965082',
618 'title': 'Soft Cell (Tracks)',
22a6f150 619 },
b6423e6c 620 'playlist_mincount': 27,
03b9c944 621 }, {
b6423e6c
S
622 'url': 'https://soundcloud.com/soft-cell-official/albums',
623 'info_dict': {
624 'id': '207965082',
625 'title': 'Soft Cell (Albums)',
626 },
627 'playlist_mincount': 1,
628 }, {
629 'url': 'https://soundcloud.com/jcv246/sets',
80fb6d4a 630 'info_dict': {
b6423e6c 631 'id': '12982173',
548c3957 632 'title': 'Jordi / cv (Sets)',
80fb6d4a 633 },
8e45e1cc 634 'playlist_mincount': 2,
80fb6d4a 635 }, {
b6423e6c 636 'url': 'https://soundcloud.com/jcv246/reposts',
80fb6d4a 637 'info_dict': {
b6423e6c
S
638 'id': '12982173',
639 'title': 'Jordi / cv (Reposts)',
80fb6d4a 640 },
b6423e6c 641 'playlist_mincount': 6,
80fb6d4a 642 }, {
b6423e6c 643 'url': 'https://soundcloud.com/clalberg/likes',
80fb6d4a 644 'info_dict': {
b6423e6c
S
645 'id': '11817582',
646 'title': 'clalberg (Likes)',
80fb6d4a 647 },
b6423e6c 648 'playlist_mincount': 5,
80fb6d4a
S
649 }, {
650 'url': 'https://soundcloud.com/grynpyret/spotlight',
651 'info_dict': {
652 'id': '7098329',
bf2dc9cc 653 'title': 'Grynpyret (Spotlight)',
80fb6d4a
S
654 },
655 'playlist_mincount': 1,
22a6f150 656 }]
92790f4e 657
80fb6d4a 658 _BASE_URL_MAP = {
548c3957
RA
659 'all': 'stream/users/%s',
660 'tracks': 'users/%s/tracks',
661 'albums': 'users/%s/albums',
662 'sets': 'users/%s/playlists',
663 'reposts': 'stream/users/%s/reposts',
664 'likes': 'users/%s/likes',
665 'spotlight': 'users/%s/spotlight',
80fb6d4a
S
666 }
667
92790f4e
JMF
668 def _real_extract(self, url):
669 mobj = re.match(self._VALID_URL, url)
670 uploader = mobj.group('user')
671
20991253 672 user = self._download_json(
548c3957
RA
673 self._resolv_url(self._BASE_URL + uploader),
674 uploader, 'Downloading user info')
80fb6d4a
S
675
676 resource = mobj.group('rsrc') or 'all'
80fb6d4a 677
836ef264 678 return self._extract_playlist(
548c3957
RA
679 self._API_V2_BASE + self._BASE_URL_MAP[resource] % user['id'],
680 str_or_none(user.get('id')),
681 '%s (%s)' % (user['username'], resource.capitalize()))
97afd99a 682
92790f4e 683
836ef264
S
684class SoundcloudTrackStationIE(SoundcloudPagedPlaylistBaseIE):
685 _VALID_URL = r'https?://(?:(?:www|m)\.)?soundcloud\.com/stations/track/[^/]+/(?P<id>[^/?#&]+)'
686 IE_NAME = 'soundcloud:trackstation'
687 _TESTS = [{
688 'url': 'https://soundcloud.com/stations/track/officialsundial/your-text',
689 'info_dict': {
690 'id': '286017854',
548c3957 691 'title': 'Track station: your text',
836ef264
S
692 },
693 'playlist_mincount': 47,
694 }]
80fb6d4a 695
836ef264
S
696 def _real_extract(self, url):
697 track_name = self._match_id(url)
80fb6d4a 698
548c3957 699 track = self._download_json(self._resolv_url(url), track_name)
836ef264 700 track_id = self._search_regex(
548c3957 701 r'soundcloud:track-stations:(\d+)', track['id'], 'track id')
92790f4e 702
836ef264 703 return self._extract_playlist(
548c3957
RA
704 self._API_V2_BASE + 'stations/%s/tracks' % track['id'],
705 track_id, 'Track station: %s' % track['title'])
20991253
PH
706
707
7518a61d 708class SoundcloudPlaylistIE(SoundcloudPlaylistBaseIE):
548c3957 709 _VALID_URL = r'https?://api(?:-v2)?\.soundcloud\.com/playlists/(?P<id>[0-9]+)(?:/?\?secret_token=(?P<token>[^&]+?))?$'
20991253 710 IE_NAME = 'soundcloud:playlist'
46f74bcf 711 _TESTS = [{
f1c05100 712 'url': 'https://api.soundcloud.com/playlists/4110309',
46f74bcf
PH
713 'info_dict': {
714 'id': '4110309',
715 'title': 'TILT Brass - Bowery Poetry Club, August \'03 [Non-Site SCR 02]',
716 'description': 're:.*?TILT Brass - Bowery Poetry Club',
717 },
718 'playlist_count': 6,
719 }]
20991253
PH
720
721 def _real_extract(self, url):
722 mobj = re.match(self._VALID_URL, url)
723 playlist_id = mobj.group('id')
20991253 724
548c3957 725 query = {
20991253 726 'client_id': self._CLIENT_ID,
2f834e93 727 }
728 token = mobj.group('token')
2f834e93 729 if token:
548c3957 730 query['secret_token'] = token
2f834e93 731
20991253 732 data = self._download_json(
548c3957
RA
733 self._API_V2_BASE + 'playlists/' + playlist_id,
734 playlist_id, 'Downloading playlist', query=query)
20991253 735
548c3957 736 entries = self._extract_track_entries(data['tracks'], token)
20991253 737
548c3957
RA
738 return self.playlist_result(
739 entries, playlist_id, data.get('title'), data.get('description'))
2abf7cab 740
741
742class SoundcloudSearchIE(SearchInfoExtractor, SoundcloudIE):
743 IE_NAME = 'soundcloud:search'
744 IE_DESC = 'Soundcloud search'
328a22e1 745 _MAX_RESULTS = float('inf')
2abf7cab 746 _TESTS = [{
747 'url': 'scsearch15:post-avant jazzcore',
748 'info_dict': {
749 'title': 'post-avant jazzcore',
750 },
751 'playlist_count': 15,
752 }]
753
754 _SEARCH_KEY = 'scsearch'
328a22e1 755 _MAX_RESULTS_PER_PAGE = 200
756 _DEFAULT_RESULTS_PER_PAGE = 50
2abf7cab 757
758 def _get_collection(self, endpoint, collection_id, **query):
a3372437 759 limit = min(
328a22e1 760 query.get('limit', self._DEFAULT_RESULTS_PER_PAGE),
761 self._MAX_RESULTS_PER_PAGE)
548c3957
RA
762 query.update({
763 'limit': limit,
764 'client_id': self._CLIENT_ID,
765 'linked_partitioning': 1,
766 'offset': 0,
767 })
768 next_url = update_url_query(self._API_V2_BASE + endpoint, query)
2abf7cab 769
2abf7cab 770 collected_results = 0
771
f6c903e7 772 for i in itertools.count(1):
7e347275 773 response = self._download_json(
f6c903e7 774 next_url, collection_id, 'Downloading page {0}'.format(i),
7e347275 775 'Unable to download API page')
2abf7cab 776
f6c903e7
S
777 collection = response.get('collection', [])
778 if not collection:
779 break
2abf7cab 780
f6c903e7 781 collection = list(filter(bool, collection))
2abf7cab 782 collected_results += len(collection)
783
f6c903e7
S
784 for item in collection:
785 yield self.url_result(item['uri'], SoundcloudIE.ie_key())
2abf7cab 786
f6c903e7 787 if not collection or collected_results >= limit:
2abf7cab 788 break
789
7e347275 790 next_url = response.get('next_href')
f6c903e7
S
791 if not next_url:
792 break
2abf7cab 793
794 def _get_n_results(self, query, n):
548c3957 795 tracks = self._get_collection('search/tracks', query, limit=n, q=query)
f6c903e7 796 return self.playlist_result(tracks, playlist_title=query)