mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-26 01:01:25 +01:00
Compare commits
1 Commits
1339a600e5
...
07f7c09eef
Author | SHA1 | Date | |
---|---|---|---|
|
07f7c09eef |
|
@ -481,7 +481,7 @@ class TestTraversalHelpers:
|
||||||
'id': 'name',
|
'id': 'name',
|
||||||
'data': 'content',
|
'data': 'content',
|
||||||
'url': 'url',
|
'url': 'url',
|
||||||
}, all, {subs_list_to_dict(lang=None)}]) == {
|
}, all, {subs_list_to_dict}]) == {
|
||||||
'de': [{'url': 'https://example.com/subs/de.ass'}],
|
'de': [{'url': 'https://example.com/subs/de.ass'}],
|
||||||
'en': [{'data': 'content'}],
|
'en': [{'data': 'content'}],
|
||||||
}, 'subs with mandatory items missing should be filtered'
|
}, 'subs with mandatory items missing should be filtered'
|
||||||
|
@ -507,54 +507,6 @@ class TestTraversalHelpers:
|
||||||
{'url': 'https://example.com/subs/en1', 'ext': 'ext'},
|
{'url': 'https://example.com/subs/en1', 'ext': 'ext'},
|
||||||
{'url': 'https://example.com/subs/en2', 'ext': 'ext'},
|
{'url': 'https://example.com/subs/en2', 'ext': 'ext'},
|
||||||
]}, '`quality` key should sort subtitle list accordingly'
|
]}, '`quality` key should sort subtitle list accordingly'
|
||||||
assert traverse_obj([
|
|
||||||
{'name': 'de', 'url': 'https://example.com/subs/de.ass'},
|
|
||||||
{'name': 'de'},
|
|
||||||
{'name': 'en', 'content': 'content'},
|
|
||||||
{'url': 'https://example.com/subs/en'},
|
|
||||||
], [..., {
|
|
||||||
'id': 'name',
|
|
||||||
'url': 'url',
|
|
||||||
'data': 'content',
|
|
||||||
}, all, {subs_list_to_dict(lang='en')}]) == {
|
|
||||||
'de': [{'url': 'https://example.com/subs/de.ass'}],
|
|
||||||
'en': [
|
|
||||||
{'data': 'content'},
|
|
||||||
{'url': 'https://example.com/subs/en'},
|
|
||||||
],
|
|
||||||
}, 'optionally provided lang should be used if no id available'
|
|
||||||
assert traverse_obj([
|
|
||||||
{'name': 1, 'url': 'https://example.com/subs/de1'},
|
|
||||||
{'name': {}, 'url': 'https://example.com/subs/de2'},
|
|
||||||
{'name': 'de', 'ext': 1, 'url': 'https://example.com/subs/de3'},
|
|
||||||
{'name': 'de', 'ext': {}, 'url': 'https://example.com/subs/de4'},
|
|
||||||
], [..., {
|
|
||||||
'id': 'name',
|
|
||||||
'url': 'url',
|
|
||||||
'ext': 'ext',
|
|
||||||
}, all, {subs_list_to_dict(lang=None)}]) == {
|
|
||||||
'de': [
|
|
||||||
{'url': 'https://example.com/subs/de3'},
|
|
||||||
{'url': 'https://example.com/subs/de4'},
|
|
||||||
],
|
|
||||||
}, 'non str types should be ignored for id and ext'
|
|
||||||
assert traverse_obj([
|
|
||||||
{'name': 1, 'url': 'https://example.com/subs/de1'},
|
|
||||||
{'name': {}, 'url': 'https://example.com/subs/de2'},
|
|
||||||
{'name': 'de', 'ext': 1, 'url': 'https://example.com/subs/de3'},
|
|
||||||
{'name': 'de', 'ext': {}, 'url': 'https://example.com/subs/de4'},
|
|
||||||
], [..., {
|
|
||||||
'id': 'name',
|
|
||||||
'url': 'url',
|
|
||||||
'ext': 'ext',
|
|
||||||
}, all, {subs_list_to_dict(lang='de')}]) == {
|
|
||||||
'de': [
|
|
||||||
{'url': 'https://example.com/subs/de1'},
|
|
||||||
{'url': 'https://example.com/subs/de2'},
|
|
||||||
{'url': 'https://example.com/subs/de3'},
|
|
||||||
{'url': 'https://example.com/subs/de4'},
|
|
||||||
],
|
|
||||||
}, 'non str types should be replaced by default id'
|
|
||||||
|
|
||||||
def test_trim_str(self):
|
def test_trim_str(self):
|
||||||
with pytest.raises(TypeError):
|
with pytest.raises(TypeError):
|
||||||
|
@ -573,7 +525,7 @@ class TestTraversalHelpers:
|
||||||
def test_unpack(self):
|
def test_unpack(self):
|
||||||
assert unpack(lambda *x: ''.join(map(str, x)))([1, 2, 3]) == '123'
|
assert unpack(lambda *x: ''.join(map(str, x)))([1, 2, 3]) == '123'
|
||||||
assert unpack(join_nonempty)([1, 2, 3]) == '1-2-3'
|
assert unpack(join_nonempty)([1, 2, 3]) == '1-2-3'
|
||||||
assert unpack(join_nonempty, delim=' ')([1, 2, 3]) == '1 2 3'
|
assert unpack(join_nonempty(delim=' '))([1, 2, 3]) == '1 2 3'
|
||||||
with pytest.raises(TypeError):
|
with pytest.raises(TypeError):
|
||||||
unpack(join_nonempty)()
|
unpack(join_nonempty)()
|
||||||
with pytest.raises(TypeError):
|
with pytest.raises(TypeError):
|
||||||
|
|
|
@ -72,6 +72,7 @@ from yt_dlp.utils import (
|
||||||
intlist_to_bytes,
|
intlist_to_bytes,
|
||||||
iri_to_uri,
|
iri_to_uri,
|
||||||
is_html,
|
is_html,
|
||||||
|
join_nonempty,
|
||||||
js_to_json,
|
js_to_json,
|
||||||
limit_length,
|
limit_length,
|
||||||
locked_file,
|
locked_file,
|
||||||
|
@ -2157,6 +2158,10 @@ Line 1
|
||||||
assert int_or_none(v=10) == 10, 'keyword passed positional should call function'
|
assert int_or_none(v=10) == 10, 'keyword passed positional should call function'
|
||||||
assert int_or_none(scale=0.1)(10) == 100, 'call after partial application should call the function'
|
assert int_or_none(scale=0.1)(10) == 100, 'call after partial application should call the function'
|
||||||
|
|
||||||
|
assert callable(join_nonempty(delim=', ')), 'varargs positional should apply partially'
|
||||||
|
assert callable(join_nonempty()), 'varargs positional should apply partially'
|
||||||
|
assert join_nonempty(None, delim=', ') == '', 'passed varargs should call the function'
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
|
@ -4381,9 +4381,7 @@ class YoutubeDL:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
for idx, t in list(enumerate(thumbnails))[::-1]:
|
for idx, t in list(enumerate(thumbnails))[::-1]:
|
||||||
thumb_ext = t.get('ext') or determine_ext(t['url'], 'jpg')
|
thumb_ext = (f'{t["id"]}.' if multiple else '') + determine_ext(t['url'], 'jpg')
|
||||||
if multiple:
|
|
||||||
thumb_ext = f'{t["id"]}.{thumb_ext}'
|
|
||||||
thumb_display_id = f'{label} thumbnail {t["id"]}'
|
thumb_display_id = f'{label} thumbnail {t["id"]}'
|
||||||
thumb_filename = replace_extension(filename, thumb_ext, info_dict.get('ext'))
|
thumb_filename = replace_extension(filename, thumb_ext, info_dict.get('ext'))
|
||||||
thumb_filename_final = replace_extension(thumb_filename_base, thumb_ext, info_dict.get('ext'))
|
thumb_filename_final = replace_extension(thumb_filename_base, thumb_ext, info_dict.get('ext'))
|
||||||
|
|
|
@ -1649,7 +1649,6 @@ from .radiokapital import (
|
||||||
RadioKapitalIE,
|
RadioKapitalIE,
|
||||||
RadioKapitalShowIE,
|
RadioKapitalShowIE,
|
||||||
)
|
)
|
||||||
from .radioradicale import RadioRadicaleIE
|
|
||||||
from .radiozet import RadioZetPodcastIE
|
from .radiozet import RadioZetPodcastIE
|
||||||
from .radlive import (
|
from .radlive import (
|
||||||
RadLiveChannelIE,
|
RadLiveChannelIE,
|
||||||
|
|
|
@ -66,14 +66,6 @@ class AfreecaTVBaseIE(InfoExtractor):
|
||||||
extensions={'legacy_ssl': True}), display_id,
|
extensions={'legacy_ssl': True}), display_id,
|
||||||
'Downloading API JSON', 'Unable to download API JSON')
|
'Downloading API JSON', 'Unable to download API JSON')
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _fixup_thumb(thumb_url):
|
|
||||||
if not url_or_none(thumb_url):
|
|
||||||
return None
|
|
||||||
# Core would determine_ext as 'php' from the url, so we need to provide the real ext
|
|
||||||
# See: https://github.com/yt-dlp/yt-dlp/issues/11537
|
|
||||||
return [{'url': thumb_url, 'ext': 'jpg'}]
|
|
||||||
|
|
||||||
|
|
||||||
class AfreecaTVIE(AfreecaTVBaseIE):
|
class AfreecaTVIE(AfreecaTVBaseIE):
|
||||||
IE_NAME = 'soop'
|
IE_NAME = 'soop'
|
||||||
|
@ -163,7 +155,7 @@ class AfreecaTVIE(AfreecaTVBaseIE):
|
||||||
'uploader': ('writer_nick', {str}),
|
'uploader': ('writer_nick', {str}),
|
||||||
'uploader_id': ('bj_id', {str}),
|
'uploader_id': ('bj_id', {str}),
|
||||||
'duration': ('total_file_duration', {int_or_none(scale=1000)}),
|
'duration': ('total_file_duration', {int_or_none(scale=1000)}),
|
||||||
'thumbnails': ('thumb', {self._fixup_thumb}),
|
'thumbnail': ('thumb', {url_or_none}),
|
||||||
})
|
})
|
||||||
|
|
||||||
entries = []
|
entries = []
|
||||||
|
@ -234,7 +226,8 @@ class AfreecaTVCatchStoryIE(AfreecaTVBaseIE):
|
||||||
|
|
||||||
return self.playlist_result(self._entries(data), video_id)
|
return self.playlist_result(self._entries(data), video_id)
|
||||||
|
|
||||||
def _entries(self, data):
|
@staticmethod
|
||||||
|
def _entries(data):
|
||||||
# 'files' is always a list with 1 element
|
# 'files' is always a list with 1 element
|
||||||
yield from traverse_obj(data, (
|
yield from traverse_obj(data, (
|
||||||
'data', lambda _, v: v['story_type'] == 'catch',
|
'data', lambda _, v: v['story_type'] == 'catch',
|
||||||
|
@ -245,7 +238,7 @@ class AfreecaTVCatchStoryIE(AfreecaTVBaseIE):
|
||||||
'title': ('title', {str}),
|
'title': ('title', {str}),
|
||||||
'uploader': ('writer_nick', {str}),
|
'uploader': ('writer_nick', {str}),
|
||||||
'uploader_id': ('writer_id', {str}),
|
'uploader_id': ('writer_id', {str}),
|
||||||
'thumbnails': ('thumb', {self._fixup_thumb}),
|
'thumbnail': ('thumb', {url_or_none}),
|
||||||
'timestamp': ('write_timestamp', {int_or_none}),
|
'timestamp': ('write_timestamp', {int_or_none}),
|
||||||
}))
|
}))
|
||||||
|
|
||||||
|
|
|
@ -279,7 +279,6 @@ class InfoExtractor:
|
||||||
thumbnails: A list of dictionaries, with the following entries:
|
thumbnails: A list of dictionaries, with the following entries:
|
||||||
* "id" (optional, string) - Thumbnail format ID
|
* "id" (optional, string) - Thumbnail format ID
|
||||||
* "url"
|
* "url"
|
||||||
* "ext" (optional, string) - actual image extension if not given in URL
|
|
||||||
* "preference" (optional, int) - quality of the image
|
* "preference" (optional, int) - quality of the image
|
||||||
* "width" (optional, int)
|
* "width" (optional, int)
|
||||||
* "height" (optional, int)
|
* "height" (optional, int)
|
||||||
|
|
|
@ -1,105 +0,0 @@
|
||||||
from .common import InfoExtractor
|
|
||||||
from ..utils import url_or_none
|
|
||||||
from ..utils.traversal import traverse_obj
|
|
||||||
|
|
||||||
|
|
||||||
class RadioRadicaleIE(InfoExtractor):
|
|
||||||
_VALID_URL = r'https?://(?:www\.)?radioradicale\.it/scheda/(?P<id>[0-9]+)'
|
|
||||||
_TESTS = [{
|
|
||||||
'url': 'https://www.radioradicale.it/scheda/471591',
|
|
||||||
'md5': 'eb0fbe43a601f1a361cbd00f3c45af4a',
|
|
||||||
'info_dict': {
|
|
||||||
'id': '471591',
|
|
||||||
'ext': 'mp4',
|
|
||||||
'title': 'md5:e8fbb8de57011a3255db0beca69af73d',
|
|
||||||
'description': 'md5:5e15a789a2fe4d67da8d1366996e89ef',
|
|
||||||
'location': 'Napoli',
|
|
||||||
'duration': 2852.0,
|
|
||||||
'timestamp': 1459987200,
|
|
||||||
'upload_date': '20160407',
|
|
||||||
'thumbnail': 'https://www.radioradicale.it/photo400/0/0/9/0/1/00901768.jpg',
|
|
||||||
},
|
|
||||||
}, {
|
|
||||||
'url': 'https://www.radioradicale.it/scheda/742783/parlamento-riunito-in-seduta-comune-11a-della-xix-legislatura',
|
|
||||||
'info_dict': {
|
|
||||||
'id': '742783',
|
|
||||||
'title': 'Parlamento riunito in seduta comune (11ª della XIX legislatura)',
|
|
||||||
'description': '-) Votazione per l\'elezione di un giudice della Corte Costituzionale (nono scrutinio)',
|
|
||||||
'location': 'CAMERA',
|
|
||||||
'duration': 5868.0,
|
|
||||||
'timestamp': 1730246400,
|
|
||||||
'upload_date': '20241030',
|
|
||||||
},
|
|
||||||
'playlist': [{
|
|
||||||
'md5': 'aa48de55dcc45478e4cd200f299aab7d',
|
|
||||||
'info_dict': {
|
|
||||||
'id': '742783-0',
|
|
||||||
'ext': 'mp4',
|
|
||||||
'title': 'Parlamento riunito in seduta comune (11ª della XIX legislatura)',
|
|
||||||
},
|
|
||||||
}, {
|
|
||||||
'md5': 'be915c189c70ad2920e5810f32260ff5',
|
|
||||||
'info_dict': {
|
|
||||||
'id': '742783-1',
|
|
||||||
'ext': 'mp4',
|
|
||||||
'title': 'Parlamento riunito in seduta comune (11ª della XIX legislatura)',
|
|
||||||
},
|
|
||||||
}, {
|
|
||||||
'md5': 'f0ee4047342baf8ed3128a8417ac5e0a',
|
|
||||||
'info_dict': {
|
|
||||||
'id': '742783-2',
|
|
||||||
'ext': 'mp4',
|
|
||||||
'title': 'Parlamento riunito in seduta comune (11ª della XIX legislatura)',
|
|
||||||
},
|
|
||||||
}],
|
|
||||||
}]
|
|
||||||
|
|
||||||
def _entries(self, videos_info, page_id):
|
|
||||||
for idx, video in enumerate(traverse_obj(
|
|
||||||
videos_info, ('playlist', lambda _, v: v['sources']))):
|
|
||||||
video_id = f'{page_id}-{idx}'
|
|
||||||
formats = []
|
|
||||||
subtitles = {}
|
|
||||||
|
|
||||||
for m3u8_url in traverse_obj(video, ('sources', ..., 'src', {url_or_none})):
|
|
||||||
fmts, subs = self._extract_m3u8_formats_and_subtitles(m3u8_url, video_id)
|
|
||||||
formats.extend(fmts)
|
|
||||||
self._merge_subtitles(subs, target=subtitles)
|
|
||||||
for sub in traverse_obj(video, ('subtitles', ..., lambda _, v: url_or_none(v['src']))):
|
|
||||||
self._merge_subtitles({sub.get('srclang') or 'und': [{
|
|
||||||
'url': sub['src'],
|
|
||||||
'name': sub.get('label'),
|
|
||||||
}]}, target=subtitles)
|
|
||||||
|
|
||||||
yield {
|
|
||||||
'id': video_id,
|
|
||||||
'title': video.get('title'),
|
|
||||||
'formats': formats,
|
|
||||||
'subtitles': subtitles,
|
|
||||||
}
|
|
||||||
|
|
||||||
def _real_extract(self, url):
|
|
||||||
page_id = self._match_id(url)
|
|
||||||
webpage = self._download_webpage(url, page_id)
|
|
||||||
|
|
||||||
videos_info = self._search_json(
|
|
||||||
r'jQuery\.extend\(Drupal\.settings\s*,',
|
|
||||||
webpage, 'videos_info', page_id)['RRscheda']
|
|
||||||
|
|
||||||
entries = list(self._entries(videos_info, page_id))
|
|
||||||
|
|
||||||
common_info = {
|
|
||||||
'id': page_id,
|
|
||||||
'title': self._og_search_title(webpage),
|
|
||||||
'description': self._og_search_description(webpage),
|
|
||||||
'location': videos_info.get('luogo'),
|
|
||||||
**self._search_json_ld(webpage, page_id),
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(entries) == 1:
|
|
||||||
return {
|
|
||||||
**entries[0],
|
|
||||||
**common_info,
|
|
||||||
}
|
|
||||||
|
|
||||||
return self.playlist_result(entries, multi_video=True, **common_info)
|
|
|
@ -216,7 +216,7 @@ def partial_application(func):
|
||||||
sig = inspect.signature(func)
|
sig = inspect.signature(func)
|
||||||
required_args = [
|
required_args = [
|
||||||
param.name for param in sig.parameters.values()
|
param.name for param in sig.parameters.values()
|
||||||
if param.kind in (inspect.Parameter.POSITIONAL_ONLY, inspect.Parameter.POSITIONAL_OR_KEYWORD)
|
if param.kind in (inspect.Parameter.POSITIONAL_ONLY, inspect.Parameter.POSITIONAL_OR_KEYWORD, inspect.Parameter.VAR_POSITIONAL)
|
||||||
if param.default is inspect.Parameter.empty
|
if param.default is inspect.Parameter.empty
|
||||||
]
|
]
|
||||||
|
|
||||||
|
@ -4837,6 +4837,7 @@ def number_of_digits(number):
|
||||||
return len('%d' % number)
|
return len('%d' % number)
|
||||||
|
|
||||||
|
|
||||||
|
@partial_application
|
||||||
def join_nonempty(*values, delim='-', from_dict=None):
|
def join_nonempty(*values, delim='-', from_dict=None):
|
||||||
if from_dict is not None:
|
if from_dict is not None:
|
||||||
values = (traversal.traverse_obj(from_dict, variadic(v)) for v in values)
|
values = (traversal.traverse_obj(from_dict, variadic(v)) for v in values)
|
||||||
|
|
|
@ -332,14 +332,14 @@ class _RequiredError(ExtractorError):
|
||||||
|
|
||||||
|
|
||||||
@typing.overload
|
@typing.overload
|
||||||
def subs_list_to_dict(*, lang: str | None = 'und', ext: str | None = None) -> collections.abc.Callable[[list[dict]], dict[str, list[dict]]]: ...
|
def subs_list_to_dict(*, ext: str | None = None) -> collections.abc.Callable[[list[dict]], dict[str, list[dict]]]: ...
|
||||||
|
|
||||||
|
|
||||||
@typing.overload
|
@typing.overload
|
||||||
def subs_list_to_dict(subs: list[dict] | None, /, *, lang: str | None = 'und', ext: str | None = None) -> dict[str, list[dict]]: ...
|
def subs_list_to_dict(subs: list[dict] | None, /, *, ext: str | None = None) -> dict[str, list[dict]]: ...
|
||||||
|
|
||||||
|
|
||||||
def subs_list_to_dict(subs: list[dict] | None = None, /, *, lang='und', ext=None):
|
def subs_list_to_dict(subs: list[dict] | None = None, /, *, ext=None):
|
||||||
"""
|
"""
|
||||||
Convert subtitles from a traversal into a subtitle dict.
|
Convert subtitles from a traversal into a subtitle dict.
|
||||||
The path should have an `all` immediately before this function.
|
The path should have an `all` immediately before this function.
|
||||||
|
@ -352,7 +352,7 @@ def subs_list_to_dict(subs: list[dict] | None = None, /, *, lang='und', ext=None
|
||||||
`quality` The sort order for each subtitle
|
`quality` The sort order for each subtitle
|
||||||
"""
|
"""
|
||||||
if subs is None:
|
if subs is None:
|
||||||
return functools.partial(subs_list_to_dict, lang=lang, ext=ext)
|
return functools.partial(subs_list_to_dict, ext=ext)
|
||||||
|
|
||||||
result = collections.defaultdict(list)
|
result = collections.defaultdict(list)
|
||||||
|
|
||||||
|
@ -360,15 +360,9 @@ def subs_list_to_dict(subs: list[dict] | None = None, /, *, lang='und', ext=None
|
||||||
if not url_or_none(sub.get('url')) and not sub.get('data'):
|
if not url_or_none(sub.get('url')) and not sub.get('data'):
|
||||||
continue
|
continue
|
||||||
sub_id = sub.pop('id', None)
|
sub_id = sub.pop('id', None)
|
||||||
if not isinstance(sub_id, str):
|
if sub_id is None:
|
||||||
if not lang:
|
|
||||||
continue
|
continue
|
||||||
sub_id = lang
|
if ext is not None and not sub.get('ext'):
|
||||||
sub_ext = sub.get('ext')
|
|
||||||
if not isinstance(sub_ext, str):
|
|
||||||
if not ext:
|
|
||||||
sub.pop('ext', None)
|
|
||||||
else:
|
|
||||||
sub['ext'] = ext
|
sub['ext'] = ext
|
||||||
result[sub_id].append(sub)
|
result[sub_id].append(sub)
|
||||||
result = dict(result)
|
result = dict(result)
|
||||||
|
@ -458,9 +452,9 @@ def trim_str(*, start=None, end=None):
|
||||||
return trim
|
return trim
|
||||||
|
|
||||||
|
|
||||||
def unpack(func, **kwargs):
|
def unpack(func):
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
def inner(items):
|
def inner(items, **kwargs):
|
||||||
return func(*items, **kwargs)
|
return func(*items, **kwargs)
|
||||||
|
|
||||||
return inner
|
return inner
|
||||||
|
|
Loading…
Reference in New Issue
Block a user