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
39
40
|
from typing import Any, Dict, List, TypeVar
from collections.abc import Mapping
from banal.lists import is_sequence, ensure_list
K = TypeVar("K")
V = TypeVar("V")
def is_mapping(obj: Any) -> bool:
return isinstance(obj, Mapping)
def ensure_dict(obj: Any) -> Dict[K, V]:
if is_mapping(obj) or hasattr(obj, "items"):
return dict(obj.items())
return {}
def clean_dict(data: Any) -> Any:
"""Remove None-valued keys from a dictionary, recursively."""
if isinstance(data, Mapping):
out = {}
for k, v in data.items():
if v is not None:
out[k] = clean_dict(v)
return out
elif is_sequence(data):
return [clean_dict(d) for d in data if d is not None]
return data
def keys_values(data: Dict[K, V], *keys: K) -> List[V]:
"""Get an entry as a list from a dict. Provide a fallback key."""
values: List[V] = []
if isinstance(data, Mapping):
for key in keys:
if key in data:
values.extend(ensure_list(data[key]))
return values
|
