diff options
author | Amish Bhadeshia <[email protected]> | 2015-03-23 13:02:44 +0000 |
---|---|---|
committer | Sergey M․ <[email protected]> | 2015-03-25 21:10:54 +0600 |
commit | 5c19d18cbfb6a0271a9d0b7ff516e1e2e0eeccfb (patch) | |
tree | 594bb63e185dbcda8f07f7e1669b46075c7d5f7d | |
parent | 838b93405bc31654b07e120795115aec6e883855 (diff) | |
download | youtube-dl-5c19d18cbfb6a0271a9d0b7ff516e1e2e0eeccfb.tar.gz youtube-dl-5c19d18cbfb6a0271a9d0b7ff516e1e2e0eeccfb.zip |
[22Tracks] Add new extractor
Conflicts:
youtube_dl/extractor/__init__.py
-rw-r--r-- | youtube_dl/extractor/twentytwotracks.py | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/youtube_dl/extractor/twentytwotracks.py b/youtube_dl/extractor/twentytwotracks.py new file mode 100644 index 000000000..b655a7503 --- /dev/null +++ b/youtube_dl/extractor/twentytwotracks.py @@ -0,0 +1,111 @@ +from __future__ import unicode_literals + +import re + +from .common import InfoExtractor + +# 22Tracks regularly replace the audio tracks that can be streamed on their +# site. The tracks usually expire after 1 months, so we can't add tests. + + +class TwentyTwoTracksIE(InfoExtractor): + _VALID_URL = r'http://22tracks\.com/([a-z]+)/([a-z]+[2]*)/(\d+)' + IE_NAME = 'TwentyTwoTracks:Tracks' + + def _extract_info(self, city, genre, track=''): + self._base_url = "http://22tracks.com/api/" + + if track == '': + itemid = genre + else: + itemid = track + + cities = self._download_json( + self._base_url + 'cities', itemid, + 'Downloading city info', 'Cannot download city info') + city_id = [x['id'] for x in cities if x['slug'] == city] + + genres = self._download_json( + self._base_url + 'genres/' + str(city_id[0]), itemid, + 'Downloading genre info', 'Cannot download genre info') + genre_id = [x['id'] for x in genres if x['slug'] == genre] + + tracks = self._download_json( + self._base_url + 'tracks/' + str(genre_id[0]), + itemid, 'Downloading track info', 'Cannot download track info') + + if track == '': + return [[x['title'] for x in genres if x['slug'] == genre][0], + tracks] + else: + return [x for x in tracks if x['id'] == itemid][0] + + def _get_token(self, filename, track_id): + token = self._download_json( + 'http://22tracks.com/token.php?desktop=true&u=%2F128%2f{0}'.format( + filename), track_id, 'Finding download link...') + + down_url = 'http://audio.22tracks.com{0}?st={1}&e={2}'.format( + token['filename'], + token['st'], + token['e']) + + return down_url + + def _real_extract(self, url): + mobj = re.match(self._VALID_URL, url) + + city_id = mobj.group(1) + genre_id = mobj.group(2) + track_id = mobj.group(3) + + self.to_screen(':: Track ID found! - Downloading single track') + + track_info = self._extract_info(city_id, genre_id, track_id) + + download_url = self._get_token(track_info['filename'], track_id) + title = '{0}-{1}'.format( + track_info['artist'].strip(), track_info['title'].strip()) + + return { + 'id': track_id, + 'url': download_url, + 'ext': 'mp3', + 'title': title, + 'duration': track_info['duration'] + } + + +class TwentyTwoTracksGenreIE(TwentyTwoTracksIE): + _VALID_URL = r'http://22tracks\.com/([a-z]+)/([a-z]+[2]*)/?' + IE_NAME = 'TwentyTwoTracks:Genre' + + def _real_extract(self, url): + mobj = re.match(self._VALID_URL, url) + + city_id = mobj.group(1) + genre_id = mobj.group(2) + + self.to_screen(':: Track ID not found! - Downloading entire genre') + + playlist_info = self._extract_info(city_id, genre_id) + + entries = [] + for track in playlist_info[1]: + title = '{0}-{1}'.format( + track['artist'].strip(), track['title'].strip()) + entries.append({ + 'id': track['id'], + 'url': self._get_token(track['filename'], track['id']), + 'ext': 'mp3', + 'title': title + }) + + self.to_screen(':: Links found - Downloading Playlist') + + return { + '_type': 'playlist', + 'id': genre_id, + 'title': playlist_info[0], + 'entries': entries + } |