artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-06-27 13:15:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions

Browse Source
This commit is contained in:
doe1080 2025-05-27 17:35:56 +09:00
parent 65f0bfd53f
commit bace78a42a
1 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
yt_dlp/extractor/startrek.py
View File

@ -1,12 +1,11 @@
from .common import InfoExtractor from .common import InfoExtractor
from ..utils import ( from ..utils import (
clean_html, clean_html,
determine_ext,
parse_iso8601, parse_iso8601,
update_url, update_url,
url_or_none, url_or_none,
) )
from ..utils.traversal import traverse_obj from ..utils.traversal import subs_list_to_dict, traverse_obj
class StarTrekIE(InfoExtractor): class StarTrekIE(InfoExtractor):
@ -50,35 +49,27 @@ class StarTrekIE(InfoExtractor):
video_id = self._match_id(url) video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id) webpage = self._download_webpage(url, video_id)
page_props = traverse_obj(self._search_nextjs_data( page_props = self._search_nextjs_data(webpage, video_id)['props']['pageProps']
webpage, video_id), ('props', 'pageProps', {dict}), default={}) video_data = page_props['video']['data']
video_data = traverse_obj(page_props, ('video', 'data', {dict}))
if youtube_id := video_data.get('youtube_video_id'): if youtube_id := video_data.get('youtube_video_id'):
return self.url_result(youtube_id, 'Youtube') return self.url_result(youtube_id, 'Youtube')
series_id = traverse_obj(video_data, ( series_id = traverse_obj(video_data, (
'series_and_movies', ..., 'series_or_movie', 'slug', {str}, any)) 'series_and_movies', ..., 'series_or_movie', 'slug', {str}, any))
subtitles = {}
if sub_url := traverse_obj(video_data, ('legacy_subtitle_file', {url_or_none})):
subtitles.setdefault('en', []).append({
'ext': determine_ext(sub_url),
'url': sub_url,
})
return { return {
'id': video_id, 'id': video_id,
'series': traverse_obj(page_props, ( 'series': traverse_obj(page_props, (
'queried', 'header', 'tab3', 'slices', ..., 'items', 'queried', 'header', 'tab3', 'slices', ..., 'items',
lambda _, v: v['link']['slug'] == series_id, 'link_copy', {str}, any)), lambda _, v: v['link']['slug'] == series_id, 'link_copy', {str}, any)),
'series_id': series_id, 'series_id': series_id,
'subtitles': subtitles,
**traverse_obj(video_data, { **traverse_obj(video_data, {
'title': ('title', ..., 'text', {clean_html}, any), 'title': ('title', ..., 'text', {clean_html}, any),
'alt_title': ('subhead', ..., 'text', {clean_html}, any), 'alt_title': ('subhead', ..., 'text', {clean_html}, any),
'categories': ('category', 'data', 'category_name', {str.upper}, all), 'categories': ('category', 'data', 'category_name', {str.upper}, filter, all),
'description': ('slices', ..., 'primary', 'content', ..., 'text', {clean_html}, any), 'description': ('slices', ..., 'primary', 'content', ..., 'text', {clean_html}, any),
'release_timestamp': ('published', {parse_iso8601}), 'release_timestamp': ('published', {parse_iso8601}),
'subtitles': ({'url': 'legacy_subtitle_file'}, all, {subs_list_to_dict(lang='en')}),
'thumbnail': ('poster_frame', 'url', {url_or_none}, {update_url(query=None)}), 'thumbnail': ('poster_frame', 'url', {url_or_none}, {update_url(query=None)}),
'url': ('legacy_video_url', {url_or_none}), 'url': ('legacy_video_url', {url_or_none}),
}), }),