Madeorsk/youtube-dl
1
0
Fork 0
mirror of https://codeberg.org/polarisfm/youtube-dl synced 2024-11-29 19:47:54 +01:00
Code Issues Releases Wiki Activity

Changed playlist result urls to provide video id.

Browse Source 
Changed playlist result urls to provide video id which is needed to keep the information which cover got used to show for the playlist.
With the video id, cover image urls for the playlist result can get created.
This commit is contained in:
Crypto90 2020-07-21 15:47:19 +02:00 committed by GitHub
parent 73670d03df
commit 45419a7863
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
youtube_dl/extractor/youtube.py
View File

@ -326,20 +326,21 @@ class YoutubeEntryListBaseInfoExtractor(YoutubeBaseInfoExtractor):
class YoutubePlaylistBaseInfoExtractor(YoutubeEntryListBaseInfoExtractor): class YoutubePlaylistBaseInfoExtractor(YoutubeEntryListBaseInfoExtractor):
def _process_page(self, content): def _process_page(self, content):
for video_id, video_title, video_duration in self.extract_videos_from_page(content): for video_id, video_title, video_duration, playlist_video_id in self.extract_videos_from_page(content):
if len(video_id) == 11: if len(video_id) == 11:
# Youtube video id found # Youtube video id found
yield self.url_result(video_id, 'Youtube', video_id, video_title, video_duration) yield self.url_result(video_id, 'Youtube', video_id, video_title, video_duration)
elif len(video_id) > 11: elif len(video_id) > 11:
# Youtube playlist id found # Youtube playlist id found
yield self.url_result('https://www.youtube.com/playlist?list=%s' % video_id, 'YoutubePlaylist', video_id, video_title, video_duration) yield self.url_result('https://www.youtube.com/watch?v=%s&list=%s' % (playlist_video_id, video_id), 'YoutubePlaylist', video_id, video_title, video_duration)
def extract_videos_from_page_impl(self, video_re, page, ids_in_page, titles_in_page, durations_in_page): def extract_videos_from_page_impl(self, video_re, page, ids_in_page, titles_in_page, durations_in_page, playlist_video_id_in_page):
for mobj in re.finditer(video_re, page): for mobj in re.finditer(video_re, page):
# The link with index 0 is not the first video of the playlist (not sure if still actual) # The link with index 0 is not the first video of the playlist (not sure if still actual)
if 'index' in mobj.groupdict() and mobj.group('id') == '0': if 'index' in mobj.groupdict() and mobj.group('id') == '0':
continue continue
video_id = mobj.group('id') video_id_original = mobj.group('id')
video_id = video_id_original
playlist_id = mobj.group('plid') if 'plid' in mobj.groupdict() else None playlist_id = mobj.group('plid') if 'plid' in mobj.groupdict() else None
if playlist_id is not None: if playlist_id is not None:
video_id = playlist_id video_id = playlist_id
@ -360,17 +361,21 @@ class YoutubePlaylistBaseInfoExtractor(YoutubeEntryListBaseInfoExtractor):
titles_in_page[idx] = video_title titles_in_page[idx] = video_title
if video_duration and not durations_in_page[idx]: if video_duration and not durations_in_page[idx]:
durations_in_page[idx] = video_duration durations_in_page[idx] = video_duration
if playlist_id is not None and not playlist_video_id_in_page[idx]:
playlist_video_id_in_page[idx] = video_id_original
except ValueError: except ValueError:
ids_in_page.append(video_id) ids_in_page.append(video_id)
titles_in_page.append(video_title) titles_in_page.append(video_title)
durations_in_page.append(video_duration) durations_in_page.append(video_duration)
playlist_video_id_in_page.append(video_id_original)
def extract_videos_from_page(self, page): def extract_videos_from_page(self, page):
ids_in_page = [] ids_in_page = []
playlist_video_id_in_page = []
titles_in_page = [] titles_in_page = []
durations_in_page = [] durations_in_page = []
self.extract_videos_from_page_impl(self._VIDEO_RE, page, ids_in_page, titles_in_page, durations_in_page) self.extract_videos_from_page_impl(self._VIDEO_RE, page, ids_in_page, titles_in_page, durations_in_page, playlist_video_id_in_page)
return zip(ids_in_page, titles_in_page, durations_in_page) return zip(ids_in_page, titles_in_page, durations_in_page, playlist_video_id_in_page)
class YoutubePlaylistsBaseInfoExtractor(YoutubeEntryListBaseInfoExtractor): class YoutubePlaylistsBaseInfoExtractor(YoutubeEntryListBaseInfoExtractor):
@ -2772,7 +2777,7 @@ class YoutubePlaylistIE(YoutubePlaylistBaseInfoExtractor):
video_title = video_title.strip() video_title = video_title.strip()
ids_in_page.append(video_id) ids_in_page.append(video_id)
titles_in_page.append(video_title) titles_in_page.append(video_title)
# TODO: ADD VIDEO DURATION HERE TOO! # TODO: ADD VIDEO DURATION HERE TOO?
durations_in_page.append(None) durations_in_page.append(None)
# Fallback with old _VIDEO_RE # Fallback with old _VIDEO_RE