147 lines
4.9 KiB
Python
147 lines
4.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import re
|
|
|
|
from .common import InfoExtractor
|
|
from ..utils import (
|
|
ExtractorError,
|
|
int_or_none,
|
|
orderedSet,
|
|
)
|
|
|
|
from hashlib import md5
|
|
from Crypto.Cipher import AES
|
|
from binascii import b2a_hex
|
|
from ..compat import compat_ord, compat_chr
|
|
|
|
|
|
def md5hex(data):
|
|
""" return hex string of md5 of the given string """
|
|
return md5(data).hexdigest().encode('utf-8')
|
|
|
|
|
|
def hexaescrypt(data, key):
|
|
""" returns hex string of aes encrypted data """
|
|
c = AES.new(key, AES.MODE_ECB)
|
|
return b2a_hex(c.encrypt(data))
|
|
|
|
|
|
def calcurlkey(songid, md5origin, mediaver=4, fmt=1):
|
|
""" Calculate the deezer download url given the songid, origin and media+format """
|
|
data = b'\xa4'.join(_.encode("utf-8") for _ in [md5origin, str(fmt), str(songid), str(mediaver)])
|
|
data = b'\xa4'.join([md5hex(data), data]) + b'\xa4'
|
|
if len(data) % 16:
|
|
data += b'\x00' * (16 - len(data) % 16)
|
|
return hexaescrypt(data, "jo6aey6haid2Teih").decode('utf-8')
|
|
|
|
|
|
def calcblowfishkey(songid):
|
|
""" Calculate the Blowfish decrypt key for a given songid """
|
|
h = md5hex(b"%d" % songid)
|
|
key = b"g4el58wc0zvf9na1"
|
|
return "".join(compat_chr(compat_ord(h[i]) ^ compat_ord(h[i + 16]) ^ compat_ord(key[i])) for i in range(16))
|
|
|
|
|
|
def getformat(song):
|
|
""" return format id for a song """
|
|
if song["FILESIZE_MP3_320"]:
|
|
return 3
|
|
if song["FILESIZE_MP3_256"]:
|
|
return 5
|
|
return 1
|
|
|
|
|
|
class DeezerPlaylistIE(InfoExtractor):
|
|
_VALID_URL = r'https?://(?:www\.)?deezer\.com/\w+/(?P<id>[0-9]+)'
|
|
_TEST = {
|
|
'url': 'http://www.deezer.com/playlist/176747451',
|
|
'info_dict': {
|
|
'id': '176747451',
|
|
'title': 'Best!',
|
|
'uploader': 'Anonymous',
|
|
'thumbnail': r're:^https?://cdn-images.deezer.com/images/cover/.*\.jpg$',
|
|
},
|
|
'playlist_count': 30,
|
|
'skip': 'Only available in .de',
|
|
}
|
|
|
|
def _real_extract(self, url):
|
|
if 'test' not in self._downloader.params:
|
|
self._downloader.report_warning('For now, this extractor only supports the 30 second previews. Patches welcome!')
|
|
|
|
mobj = re.match(self._VALID_URL, url)
|
|
playlist_id = mobj.group('id')
|
|
|
|
webpage = self._download_webpage(url, playlist_id)
|
|
geoblocking_msg = self._html_search_regex(
|
|
r'<p class="soon-txt">(.*?)</p>', webpage, 'geoblocking message',
|
|
default=None)
|
|
if geoblocking_msg is not None:
|
|
raise ExtractorError(
|
|
'Deezer said: %s' % geoblocking_msg, expected=True)
|
|
|
|
host_stream_cdn = self._search_regex(
|
|
r'var HOST_STREAM_CDN = \'(.*?)\'', webpage, 'host stream cdn')
|
|
|
|
data_json = self._search_regex(
|
|
(r'__DZR_APP_STATE__\s*=\s*({.+?})\s*</script>',
|
|
r'naboo\.display\(\'[^\']+\',\s*(.*?)\);\n'),
|
|
webpage, 'data JSON')
|
|
data = json.loads(data_json)
|
|
|
|
playlist_title = data.get('DATA', {}).get('TITLE')
|
|
playlist_uploader = data.get('DATA', {}).get('PARENT_USERNAME')
|
|
# playlist_thumbnail = self._search_regex( r'<img id="naboo_playlist_image".*?src="([^"]+)"', webpage, 'playlist thumbnail')
|
|
|
|
entries = []
|
|
|
|
if 'SONGS' in data:
|
|
# playlist, album
|
|
songlist = data['SONGS']['data']
|
|
elif 'MD5_ORIGIN' in data['DATA']:
|
|
# track
|
|
songlist = [data['DATA']]
|
|
elif "ALBUMS" in data:
|
|
songlist = []
|
|
for album in data["ALBUMS"]["data"]:
|
|
songlist.extend(album['SONGS']['data'])
|
|
else:
|
|
raise ExtractorError('Could not find songs')
|
|
|
|
# album, playlist
|
|
for s in songlist:
|
|
|
|
urlkey = calcurlkey(int(s["SNG_ID"]), str(s["MD5_ORIGIN"]), int(s["MEDIA_VERSION"]), getformat(s))
|
|
url = host_stream_cdn.replace('{0}', str(s["MD5_ORIGIN"])[0]) + "/" + urlkey
|
|
|
|
formats = [{
|
|
'format_id': 'preview',
|
|
'key': calcblowfishkey(int(s["SNG_ID"])),
|
|
'url': url,
|
|
'preference': -100, # Only the first 30 seconds
|
|
'ext': 'mp3',
|
|
'protocol': 'deezer',
|
|
}]
|
|
self._sort_formats(formats)
|
|
artists = ', '.join(
|
|
orderedSet(a['ART_NAME'] for a in s['ARTISTS']))
|
|
entries.append({
|
|
'id': s['SNG_ID'],
|
|
'duration': int_or_none(s.get('DURATION')),
|
|
'title': '%s - %s' % (artists, s['SNG_TITLE']),
|
|
'uploader': s['ART_NAME'],
|
|
'uploader_id': s['ART_ID'],
|
|
'age_limit': 16 if s.get('EXPLICIT_LYRICS') == '1' else 0,
|
|
'formats': formats,
|
|
})
|
|
|
|
return {
|
|
'_type': 'playlist',
|
|
'id': playlist_id,
|
|
'title': playlist_title,
|
|
'uploader': playlist_uploader,
|
|
# 'thumbnail': playlist_thumbnail,
|
|
'entries': entries,
|
|
}
|