mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-07-05 21:08:33 +00:00
[Substack] allow multiple embeds regardless of post type
This commit is contained in:
jhwgh1968
parent
d49565da20
commit
e89132c678
@ -133,7 +133,9 @@ def _real_extract(self, url):
|
||||
canonical_url = urllib.parse.urlparse(url)._replace(netloc=domain).geturl()
|
||||
|
||||
post_type = webpage_info['post']['type']
|
||||
post_title = traverse_obj(webpage_info, ('post', 'title'))
|
||||
formats, subtitles = [], {}
|
||||
all_results = []
|
||||
result_info = {
|
||||
'id': str(webpage_info['post']['id']),
|
||||
'subtitles': subtitles,
|
||||
@ -144,7 +146,9 @@ def _real_extract(self, url):
|
||||
'uploader_id': str_or_none(traverse_obj(webpage_info, ('post', 'publication_id'))),
|
||||
'webpage_url': canonical_url,
|
||||
}
|
||||
# handle specific post types which are based on media
|
||||
if post_type == 'podcast':
|
||||
podcast_result = dict(result_info)
|
||||
fmt = {'url': webpage_info['post']['podcast_url']}
|
||||
if not determine_ext(fmt['url'], default_ext=None):
|
||||
# The redirected format URL expires but the original URL doesn't,
|
||||
@ -153,89 +157,87 @@ def _real_extract(self, url):
|
||||
HEADRequest(fmt['url']), display_id,
|
||||
'Resolving podcast file extension',
|
||||
'Podcast URL is invalid').url)
|
||||
result_info['formats'] = [fmt]
|
||||
return result_info
|
||||
elif post_type == 'video':
|
||||
podcast_result['formats'] = [fmt]
|
||||
all_results.append(podcast_result)
|
||||
if post_type == 'video':
|
||||
video_result = dict(result_info)
|
||||
formats, subtitles = self._extract_video_formats(
|
||||
webpage_info['post']['videoUpload']['id'], canonical_url)
|
||||
result_info.update({
|
||||
video_result.update({
|
||||
'formats': formats,
|
||||
'subtitles': subtitles,
|
||||
})
|
||||
return result_info
|
||||
elif post_type == 'newsletter':
|
||||
results = []
|
||||
found_items = []
|
||||
post_id = str(webpage_info['post']['id'])
|
||||
post_title = traverse_obj(webpage_info, ('post', 'title'))
|
||||
assert result_info['uploader_id'] is not None, 'newsletter posted without user_id'
|
||||
all_results.append(video_result)
|
||||
|
||||
video_players = re.finditer(
|
||||
r'<div[^>]*data-component-name="VideoEmbedPlayer"[^>]*>',
|
||||
webpage)
|
||||
for vp in video_players:
|
||||
video_id = self._search_regex(r'id="([^"]+)"',
|
||||
vp.group(0),
|
||||
'video id', group=1).replace('media-', '')
|
||||
video_metadata_url = urllib.parse.urljoin(url, f'/api/v1/video/upload/{video_id}')
|
||||
json_vid_data = self._download_json(video_metadata_url, video_id)
|
||||
assert video_id == json_vid_data['id'], 'unexpected json metadata retrieved'
|
||||
json_vid_data['_type'] = 'video'
|
||||
found_items.append(json_vid_data)
|
||||
# search for embedded players on the page
|
||||
found_items = []
|
||||
post_id = str(webpage_info['post']['id'])
|
||||
assert result_info['uploader_id'] is not None, 'newsletter posted without user_id'
|
||||
|
||||
audio_players = re.finditer(
|
||||
r'<div[^>]*data-component-name="AudioEmbedPlayer"[^>]*>',
|
||||
webpage)
|
||||
for ap in audio_players:
|
||||
video_uri = self._search_regex(r'src="(/api/v1/audio/upload/[^"]+)"',
|
||||
webpage[ap.start():],
|
||||
'video uri', group=1)
|
||||
video_metadata_url = urllib.parse.urljoin(url, video_uri.replace('/src', ''))
|
||||
video_id = self._search_regex(r'upload/([^/]+)/src',
|
||||
video_uri,
|
||||
'video id', group=1)
|
||||
json_vid_data = self._download_json(video_metadata_url, video_id)
|
||||
assert video_id == json_vid_data['id'], 'unexpected json metadata retrieved'
|
||||
json_vid_data['_type'] = 'audio'
|
||||
json_vid_data['_uri'] = video_uri
|
||||
found_items.append(json_vid_data)
|
||||
video_players = re.finditer(
|
||||
r'<div[^>]*data-component-name="VideoEmbedPlayer"[^>]*>',
|
||||
webpage)
|
||||
for vp in video_players:
|
||||
video_id = self._search_regex(r'id="([^"]+)"',
|
||||
vp.group(0),
|
||||
'video id', group=1).replace('media-', '')
|
||||
video_metadata_url = urllib.parse.urljoin(url, f'/api/v1/video/upload/{video_id}')
|
||||
json_vid_data = self._download_json(video_metadata_url, video_id)
|
||||
assert video_id == json_vid_data['id'], 'unexpected json metadata retrieved'
|
||||
json_vid_data['_type'] = 'video'
|
||||
found_items.append(json_vid_data)
|
||||
|
||||
for json_vid_data in found_items:
|
||||
video_id = json_vid_data['id']
|
||||
if 'video' in json_vid_data['_type']:
|
||||
formats, subtitles = self._extract_video_formats(
|
||||
video_id, canonical_url,
|
||||
query_params={'override_publication_id': result_info['uploader_id']})
|
||||
else:
|
||||
fmt = {'url': urllib.parse.urljoin(url, json_vid_data['_uri'])}
|
||||
if not determine_ext(fmt['url'], default_ext=None):
|
||||
# The redirected format URL expires but the original URL doesn't,
|
||||
# so we only want to extract the extension from this request
|
||||
fmt['ext'] = determine_ext(self._request_webpage(
|
||||
HEADRequest(fmt['url']), video_id,
|
||||
'Resolving audio file extension',
|
||||
'Embedded audio URL is invalid').url)
|
||||
formats.append(fmt)
|
||||
audio_players = re.finditer(
|
||||
r'<div[^>]*data-component-name="AudioEmbedPlayer"[^>]*>',
|
||||
webpage)
|
||||
for ap in audio_players:
|
||||
video_uri = self._search_regex(r'src="(/api/v1/audio/upload/[^"]+)"',
|
||||
webpage[ap.start():],
|
||||
'video uri', group=1)
|
||||
video_metadata_url = urllib.parse.urljoin(url, video_uri.replace('/src', ''))
|
||||
video_id = self._search_regex(r'upload/([^/]+)/src',
|
||||
video_uri,
|
||||
'video id', group=1)
|
||||
json_vid_data = self._download_json(video_metadata_url, video_id)
|
||||
assert video_id == json_vid_data['id'], 'unexpected json metadata retrieved'
|
||||
json_vid_data['_type'] = 'audio'
|
||||
json_vid_data['_uri'] = video_uri
|
||||
found_items.append(json_vid_data)
|
||||
|
||||
new_result = dict(result_info)
|
||||
new_result.update({
|
||||
'formats': formats,
|
||||
'subtitles': subtitles,
|
||||
'id': video_id,
|
||||
'title': json_vid_data.get('name', None),
|
||||
'description': None,
|
||||
'thumbnail': json_vid_data.get('thumbnail_url', None),
|
||||
'duration': float_or_none(json_vid_data.get('duration', None)),
|
||||
# videos can be cross-embedded, so the publication id should be of the uploader
|
||||
'uploader_id': str_or_none(json_vid_data.get('user_id', None)),
|
||||
'uploader': None, # video uploader username is not included
|
||||
})
|
||||
results.append(new_result)
|
||||
|
||||
if len(results) > 0:
|
||||
playlist_title = f'Videos for {post_title}'
|
||||
return self.playlist_result(results, post_id, playlist_title)
|
||||
for json_vid_data in found_items:
|
||||
video_id = json_vid_data['id']
|
||||
if 'video' in json_vid_data['_type']:
|
||||
formats, subtitles = self._extract_video_formats(
|
||||
video_id, canonical_url,
|
||||
query_params={'override_publication_id': result_info['uploader_id']})
|
||||
else:
|
||||
self.raise_no_formats(f'Page type "{post_type}" contains no supported embeds')
|
||||
fmt = {'url': urllib.parse.urljoin(url, json_vid_data['_uri'])}
|
||||
if not determine_ext(fmt['url'], default_ext=None):
|
||||
# The redirected format URL expires but the original URL doesn't,
|
||||
# so we only want to extract the extension from this request
|
||||
fmt['ext'] = determine_ext(self._request_webpage(
|
||||
HEADRequest(fmt['url']), video_id,
|
||||
'Resolving audio file extension',
|
||||
'Embedded audio URL is invalid').url)
|
||||
formats.append(fmt)
|
||||
|
||||
self.raise_no_formats(f'Page type "{post_type}" is not supported')
|
||||
new_result = dict(result_info)
|
||||
new_result.update({
|
||||
'formats': formats,
|
||||
'subtitles': subtitles,
|
||||
'id': video_id,
|
||||
'title': json_vid_data.get('name', None),
|
||||
'description': None,
|
||||
'thumbnail': json_vid_data.get('thumbnail_url', None),
|
||||
'duration': float_or_none(json_vid_data.get('duration', None)),
|
||||
# videos can be cross-embedded, so the publication id should be of the uploader
|
||||
'uploader_id': str_or_none(json_vid_data.get('user_id', None)),
|
||||
'uploader': None, # video uploader username is not included
|
||||
})
|
||||
all_results.append(new_result)
|
||||
|
||||
if len(all_results) > 0:
|
||||
playlist_title = f'Videos for {post_title}'
|
||||
return self.playlist_result(all_results, post_id, playlist_title)
|
||||
else:
|
||||
self.raise_no_formats(f'Page type "{post_type}" contains no supported embeds')
|
||||
|
||||
Loading…
Reference in New Issue
Block a user