-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathutils.py
More file actions
38 lines (27 loc) · 973 Bytes
/
utils.py
File metadata and controls
38 lines (27 loc) · 973 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from __future__ import annotations
from typing import TypeVar, cast, overload
_T = TypeVar("_T")
@overload
def safe_get(obj: object, *path: str | int) -> object | None: ...
@overload
def safe_get(obj: object, *path: str | int, type: type[_T]) -> _T | None: ...
def safe_get(
obj: object, *path: str | int, type: type[object] | None = None
) -> object | None:
current = obj
for part in path:
if isinstance(current, dict):
current_dict = cast(dict[str | int, object], current)
if part not in current_dict:
return None
current = current_dict[part]
continue
if isinstance(current, list) and isinstance(part, int):
if not -len(current) <= part < len(current):
return None
current = current[part]
continue
return None
if type is not None and not isinstance(current, type):
return None
return current