mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-06-27 17:08:32 +00:00
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import base64
|
|
import json
|
|
|
|
from .common import InfoExtractor
|
|
from ..utils import (
|
|
extract_attributes,
|
|
float_or_none,
|
|
parse_resolution,
|
|
qualities,
|
|
url_or_none,
|
|
urljoin,
|
|
)
|
|
from ..utils.traversal import (
|
|
find_element,
|
|
require,
|
|
traverse_obj,
|
|
)
|
|
|
|
|
|
class AppleConnectIE(InfoExtractor):
|
|
IE_NAME = 'apple:music:connect'
|
|
IE_DESC = 'Apple Music Connect'
|
|
|
|
_BASE_URL = 'https://music.apple.com'
|
|
_QUALITIES = {
|
|
'provisionalUploadVideo': (None, None),
|
|
'sdVideo': (640, 480),
|
|
'sdVideoWithPlusAudio': (640, 480),
|
|
'sd480pVideo': (720, 480),
|
|
'720pHdVideo': (1280, 720),
|
|
'1080pHdVideo': (1440, 1080),
|
|
}
|
|
_VALID_URL = r'https?://music\.apple\.com/\w{0,2}/post/(?P<id>\d+)'
|
|
_TESTS = [{
|
|
'url': 'https://music.apple.com/us/post/1018290019',
|
|
'info_dict': {
|
|
'id': '1018290019',
|
|
'ext': 'm4v',
|
|
'title': 'Energy',
|
|
'duration': 177.911,
|
|
'thumbnail': r're:https?://.+\.png',
|
|
'upload_date': '20150710',
|
|
'uploader': 'Drake',
|
|
},
|
|
}, {
|
|
'url': 'https://music.apple.com/us/post/1016746627',
|
|
'info_dict': {
|
|
'id': '1016746627',
|
|
'ext': 'm4v',
|
|
'title': 'Body Shop (Madonna) - Chellous Lima (Acoustic Cover)',
|
|
'duration': 210.278,
|
|
'thumbnail': r're:https?://.+\.png',
|
|
'upload_date': '20150706',
|
|
'uploader': 'Chellous Lima',
|
|
},
|
|
}]
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, video_id)
|
|
|
|
js_url = traverse_obj(webpage, (
|
|
{find_element(tag='script', attr='crossorigin', value='', html=True)},
|
|
{extract_attributes}, 'src', {urljoin(self._BASE_URL)}, {require('JS URL')}))
|
|
js = self._download_webpage(js_url, video_id)
|
|
|
|
header = base64.urlsafe_b64encode(
|
|
json.dumps({
|
|
'alg': 'ES256',
|
|
'typ': 'JWT',
|
|
'kid': 'WebPlayKid',
|
|
}, separators=(',', ':')).encode(),
|
|
).decode().rstrip('=')
|
|
jwt = self._search_regex(
|
|
fr'(["\'])(?P<jwt>{header}(?:\.[\w-]+){{2}})\1', js, 'JSON Web Token', group='jwt')
|
|
|
|
videos = self._download_json(
|
|
'https://amp-api.music.apple.com/v1/catalog/us/uploaded-videos',
|
|
video_id, headers={
|
|
'Authorization': f'Bearer {jwt}',
|
|
'Origin': self._BASE_URL,
|
|
}, query={'ids': video_id, 'l': 'en-US'})
|
|
attributes = traverse_obj(videos, (
|
|
'data', ..., 'attributes', any, {require('video information')}))
|
|
|
|
formats = []
|
|
quality = qualities(list(self._QUALITIES.keys()))
|
|
for format_id, src_url in traverse_obj(attributes, (
|
|
'assetTokens', {dict.items}, lambda _, v: url_or_none(v[1]),
|
|
)):
|
|
formats.append({
|
|
'ext': 'm4v',
|
|
'format_id': format_id,
|
|
'quality': quality(format_id),
|
|
'url': src_url,
|
|
**parse_resolution(src_url),
|
|
**traverse_obj(self._QUALITIES, (format_id, {
|
|
'height': 1,
|
|
'width': 0,
|
|
})),
|
|
})
|
|
|
|
return {
|
|
'id': video_id,
|
|
'formats': formats,
|
|
'thumbnail': self._html_search_meta(
|
|
('og:image', 'og:image:secure_url', 'twitter:image'), webpage),
|
|
**traverse_obj(attributes, {
|
|
'title': ('name', {str}),
|
|
'duration': ('durationInMilliseconds', {float_or_none(scale=1000)}),
|
|
'upload_date': ('uploadDate', {str}, {lambda x: x.replace('-', '')}),
|
|
'uploader': (('artistName', 'uploadingArtistName'), {str}, any),
|
|
'webpage_url': ('postUrl', {url_or_none}),
|
|
}),
|
|
}
|