From 575e17a1b912ace50ac383e74a8da5b8e9c8de4e Mon Sep 17 00:00:00 2001 From: pukkandan Date: Sat, 7 Aug 2021 14:24:50 +0530 Subject: [PATCH] [utils] Fix `traverse_obj` depth when is_user_input --- yt_dlp/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/yt_dlp/utils.py b/yt_dlp/utils.py index b04fbd22c..a6e613139 100644 --- a/yt_dlp/utils.py +++ b/yt_dlp/utils.py @@ -6263,6 +6263,8 @@ def traverse_obj( def _traverse_obj(obj, path, _current_depth=0): nonlocal depth + if obj is None: + return None path = tuple(variadic(path)) for i, key in enumerate(path): if isinstance(key, (list, tuple)): @@ -6275,7 +6277,7 @@ def _traverse_obj(obj, path, _current_depth=0): _current_depth += 1 depth = max(depth, _current_depth) return [_traverse_obj(inner_obj, path[i + 1:], _current_depth) for inner_obj in obj] - elif isinstance(obj, dict): + elif isinstance(obj, dict) and not (is_user_input and key == ':'): obj = (obj.get(key) if casesense or (key in obj) else next((v for k, v in obj.items() if _lower(k) == key), None)) else: @@ -6283,7 +6285,7 @@ def _traverse_obj(obj, path, _current_depth=0): key = (int_or_none(key) if ':' not in key else slice(*map(int_or_none, key.split(':')))) if key == slice(None): - return _traverse_obj(obj, (..., *path[i + 1:])) + return _traverse_obj(obj, (..., *path[i + 1:]), _current_depth) if not isinstance(key, (int, slice)): return None if not isinstance(obj, (list, tuple, LazyList)):