mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-08-15 00:48:28 +00:00
Use traverse_obj for get json info data
This commit is contained in:
parent
4f21363359
commit
89eac10fa8
@ -2,6 +2,7 @@
|
||||
|
||||
from .common import InfoExtractor
|
||||
from ..utils import ExtractorError, int_or_none
|
||||
from ..utils.traversal import traverse_obj
|
||||
|
||||
|
||||
class SmotrimIE(InfoExtractor):
|
||||
@ -126,24 +127,19 @@ def _real_extract(self, url):
|
||||
raise ExtractorError(str(e), expected=True)
|
||||
if json_info.get('status') != 200:
|
||||
raise ExtractorError('Json download error. Status code: %s' % str(json_info.get('status')), expected=True)
|
||||
try:
|
||||
media_info = json_info.get('data').get('playlist').get('medialist')[0]
|
||||
except (KeyError, AttributeError, TypeError) as e:
|
||||
raise ExtractorError('media_info get error: %s' % str(e), expected=True)
|
||||
try:
|
||||
media_info = traverse_obj(json_info, ('data', 'playlist', 'medialist', 0))
|
||||
if not media_info:
|
||||
raise ExtractorError('Unable to get media_info', expected=True)
|
||||
formats, subtitles = self._extract_m3u8_formats_and_subtitles(
|
||||
media_info.get('sources').get('m3u8').get('auto'), video_id, 'mp4', m3u8_id='hls',
|
||||
traverse_obj(media_info, ('sources', 'm3u8', 'auto')), video_id, 'mp4', m3u8_id='hls',
|
||||
)
|
||||
res = {
|
||||
|
||||
return {
|
||||
'id': video_id,
|
||||
'title': media_info.get('title'),
|
||||
'thumbnail': media_info.get('pictures').get('16:9'),
|
||||
'title': traverse_obj(media_info, ('title')),
|
||||
'thumbnail': traverse_obj(media_info, ('pictures', '16:9')),
|
||||
'formats': formats,
|
||||
'subtitles': subtitles,
|
||||
'is_live': json_info.get('data').get('playlist').get('type') == 'live',
|
||||
'duration': int_or_none(media_info.get('duration')),
|
||||
'is_live': traverse_obj(json_info, ('data', 'playlist', 'type')) == 'live',
|
||||
'duration': int_or_none(traverse_obj(media_info, ('duration'))),
|
||||
}
|
||||
except (KeyError, AttributeError, TypeError) as e:
|
||||
raise ExtractorError('Result error: %s' % str(e), expected=True)
|
||||
|
||||
return res
|
||||
|
Loading…
Reference in New Issue
Block a user