jriou/yt-mp3-ui
1
0
Fork 0
Code Activity

[utils] Generalize traverse_dict to traverse_obj

Browse source
This commit is contained in:
pukkandan 2021-06-08 14:23:56 +05:30
commit 324ad82006
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
3 changed files with 34 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
yt_dlp/postprocessor/ffmpeg.py
View file

@ -23,7 +23,7 @@ from ..utils import (
ISO639Utils,
process_communicate_or_kill,
replace_extension,
traverse_dict,
traverse_obj,
)
@ -229,7 +229,7 @@ class FFmpegPostProcessor(PostProcessor):
def get_stream_number(self, path, keys, value):
streams = self.get_metadata_object(path)['streams']
num = next(
(i for i, stream in enumerate(streams) if traverse_dict(stream, keys, casesense=False) == value),
(i for i, stream in enumerate(streams) if traverse_obj(stream, keys, casesense=False) == value),
None)
return num, len(streams)