mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-02 06:27:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

[YoutubeDL] extend wide format selector

Browse source
This commit is contained in:
Lesmiscore 2022-04-27 16:05:05 +09:00
parent 69b59b4b4b
commit 2ae28147b0
No known key found for this signature in database
GPG key ID: 0EC2B52CF86236FF
1 changed files with 57 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

71
yt_dlp/YoutubeDL.py
View file

@ -2146,21 +2146,64 @@ def selector_function(ctx):
elif selector.type == SINGLE: # atom
format_spec = selector.selector or 'best'
mobj = re.match(
r'(?P<m>merge)?all(?P<t>video|audio)?(?P<l>lang)?', format_spec)
if mobj:
mg, tp, la = mobj.groups()
def selector_function(ctx):
formats = ctx['formats'][::-1]
# TODO: Add allvideo, allaudio etc by generalizing the code with best/worst selector
if format_spec == 'all':
def selector_function(ctx):
yield from _check_formats(ctx['formats'][::-1])
elif format_spec == 'mergeall':
def selector_function(ctx):
formats = list(_check_formats(
f for f in ctx['formats'] if f.get('vcodec') != 'none' or f.get('acodec') != 'none'))
if not formats:
return
merged_format = formats[-1]
for f in formats[-2::-1]:
merged_format = _merge((merged_format, f))
yield merged_format
if tp == 'video':
formats = (f for f in formats if f.get('vcodec') != 'none')
elif tp == 'audio':
formats = (f for f in formats if f.get('acodec') != 'none')
elif mg:
# exclude storyboards
formats = (
f for f in formats if f.get('vcodec') != 'none' or f.get('acodec') != 'none')
if la:
all_fmts = formats if isinstance(formats, list) else list(formats)
langs = set(filter(None, traverse_obj(all_fmts, (..., 'language'), default=[])))
formats = []
if not langs:
self.report_warning('This video has no track with language tag')
formats = all_fmts
elif tp:
# only audio or video
for f in all_fmts:
if not langs:
break
lang = f.get('language')
if not lang or lang not in langs:
continue
langs.remove(lang)
formats.append(f)
else:
# no video/audio specified
# vcodec, acodec, language
langs = set(itertools.product((True, False), (True, False), langs))
for f in all_fmts:
if not langs:
break
lang = f.get('language')
tu = (f.get('vcodec') != 'none', f.get('acodec') != 'none', lang)
if not lang or tu not in langs:
continue
langs.remove(tu)
formats.append(f)
formats = _check_formats(formats)
if mg:
formats = iter(formats)
merged_format = next(formats, None)
if merged_format is None:
return
for f in formats:
merged_format = _merge((merged_format, f))
yield merged_format
else:
yield from formats
else:
format_fallback, seperate_fallback, format_reverse, format_idx = False, None, True, 1