discogs.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435
  1. from .common import InfoExtractor
  2. from .youtube import YoutubeIE
  3. from ..utils import traverse_obj
  4. class DiscogsReleasePlaylistIE(InfoExtractor):
  5. _VALID_URL = r'https?://(?:www\.)?discogs\.com/(?P<type>release|master)/(?P<id>\d+)'
  6. _TESTS = [{
  7. 'url': 'https://www.discogs.com/release/1-The-Persuader-Stockholm',
  8. 'info_dict': {
  9. 'id': 'release1',
  10. 'title': 'Stockholm',
  11. },
  12. 'playlist_mincount': 7,
  13. }, {
  14. 'url': 'https://www.discogs.com/master/113-Vince-Watson-Moments-In-Time',
  15. 'info_dict': {
  16. 'id': 'master113',
  17. 'title': 'Moments In Time',
  18. },
  19. 'playlist_mincount': 53,
  20. }]
  21. def _real_extract(self, url):
  22. playlist_id, playlist_type = self._match_valid_url(url).group('id', 'type')
  23. display_id = f'{playlist_type}{playlist_id}'
  24. response = self._download_json(
  25. f'https://api.discogs.com/{playlist_type}s/{playlist_id}', display_id)
  26. entries = [
  27. self.url_result(video['uri'], YoutubeIE, video_title=video.get('title'))
  28. for video in traverse_obj(response, ('videos', lambda _, v: YoutubeIE.suitable(v['uri'])))]
  29. return self.playlist_result(entries, display_id, response.get('title'))