[vlive] add support for playlists #7 (blackjack4494#223)

Authored by kyuyeunk
This commit is contained in:
pukkandan 2021-01-07 20:44:00 +05:30 committed by pukkandan
commit 7dbce2c532

View File

@ -72,6 +72,13 @@ class VLiveIE(VLiveBaseIE):
# works only with gcc=KR # works only with gcc=KR
'url': 'https://www.vlive.tv/video/225019', 'url': 'https://www.vlive.tv/video/225019',
'only_matching': True, 'only_matching': True,
}, {
'url': 'https://www.vlive.tv/video/223906',
'info_dict': {
'id': '58',
'title': 'RUN BTS!'
},
'playlist_mincount': 120
}] }]
def _real_initialize(self): def _real_initialize(self):
@ -105,10 +112,12 @@ def is_logged_in():
if not is_logged_in(): if not is_logged_in():
raise ExtractorError('Unable to log in', expected=True) raise ExtractorError('Unable to log in', expected=True)
def _call_api(self, path_template, video_id, fields=None): def _call_api(self, path_template, video_id, fields=None, limit=None):
query = {'appId': self._APP_ID, 'gcc': 'KR'} query = {'appId': self._APP_ID, 'gcc': 'KR'}
if fields: if fields:
query['fields'] = fields query['fields'] = fields
if limit:
query['limit'] = limit
try: try:
return self._download_json( return self._download_json(
'https://www.vlive.tv/globalv-web/vam-web/' + path_template % video_id, video_id, 'https://www.vlive.tv/globalv-web/vam-web/' + path_template % video_id, video_id,
@ -124,10 +133,34 @@ def _real_extract(self, url):
post = self._call_api( post = self._call_api(
'post/v1.0/officialVideoPost-%s', video_id, 'post/v1.0/officialVideoPost-%s', video_id,
'author{nickname},channel{channelCode,channelName},officialVideo{commentCount,exposeStatus,likeCount,playCount,playTime,status,title,type,vodId}') 'author{nickname},channel{channelCode,channelName},officialVideo{commentCount,exposeStatus,likeCount,playCount,playTime,status,title,type,vodId},playlist{playlistSeq,totalCount,name}')
video = post['officialVideo'] playlist = post.get('playlist')
if not playlist or self._downloader.params.get('noplaylist'):
if playlist:
self.to_screen(
'Downloading just video %s because of --no-playlist'
% video_id)
video = post['officialVideo']
return self._get_vlive_info(post, video, video_id)
else:
playlist_name = playlist.get('name')
playlist_id = str_or_none(playlist.get('playlistSeq'))
playlist_count = str_or_none(playlist.get('totalCount'))
playlist = self._call_api(
'playlist/v1.0/playlist-%s/posts', playlist_id, 'data', limit=playlist_count)
entries = []
for video_data in playlist['data']:
video = video_data.get('officialVideo')
video_id = str_or_none(video.get('videoSeq'))
entries.append(self._get_vlive_info(video_data, video, video_id))
return self.playlist_result(entries, playlist_id, playlist_name)
def _get_vlive_info(self, post, video, video_id):
def get_common_fields(): def get_common_fields():
channel = post.get('channel') or {} channel = post.get('channel') or {}
return { return {
@ -323,22 +356,17 @@ def _real_extract(self, url):
video_id = compat_str(video_id) video_id = compat_str(video_id)
if video_type in ('PLAYLIST'): if video_type in ('PLAYLIST'):
playlist_videos = try_get( first_video_id = try_get(
video, video,
lambda x: x['videoPlaylist']['videoList'], list) lambda x: x['videoPlaylist']['videoList'][0]['videoSeq'], int)
if not playlist_videos:
if not first_video_id:
continue continue
for playlist_video in playlist_videos: entries.append(
playlist_video_id = playlist_video.get('videoSeq') self.url_result(
if not playlist_video_id: 'http://www.vlive.tv/video/%s' % first_video_id,
continue ie=VLiveIE.ie_key(), video_id=first_video_id))
playlist_video_id = compat_str(playlist_video_id)
entries.append(
self.url_result(
'http://www.vlive.tv/video/%s' % playlist_video_id,
ie=VLiveIE.ie_key(), video_id=playlist_video_id))
else: else:
entries.append( entries.append(
self.url_result( self.url_result(