Skip to content

Commit 1e00235

Browse files
authored
Update componenthandler.py
1 parent b4b6d25 commit 1e00235

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

youtubesearchpython/core/componenthandler.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,28 +1,32 @@
11
from typing import Union, List
22
from urllib.parse import urlparse, parse_qs
3+
import re
34

45

56
def getValue(source: dict, path: List[Union[str, int]]) -> Union[str, int, dict, None]:
67
value = source
8+
79
for key in path:
810
if value is None:
911
return None
12+
1013
if isinstance(key, str):
1114
if not isinstance(value, dict):
1215
return None
13-
if key in value:
14-
value = value[key]
15-
else:
16+
value = value.get(key)
17+
if value is None:
1618
return None
19+
1720
elif isinstance(key, int):
1821
if not isinstance(value, (list, tuple)):
1922
return None
20-
if 0 <= key < len(value):
21-
value = value[key]
22-
else:
23+
if key < 0 or key >= len(value):
2324
return None
25+
value = value[key]
26+
2427
else:
2528
return None
29+
2630
return value
2731

2832

0 commit comments

Comments
 (0)