a | b/utils.py | ||
---|---|---|---|
1 | from typing import Dict, List, Union, Any |
||
2 | |||
3 | def get_values(search_hits: Union[Dict[Any, Any], List[Any]], search_key: str) -> List: |
||
4 | results = [] |
||
5 | def extract(data: Union[Dict[Any, Any], List[Any]]): |
||
6 | if isinstance(data, dict): |
||
7 | for key, value in data.items(): |
||
8 | if key == search_key: |
||
9 | results.append(value) |
||
10 | extract(value) |
||
11 | elif isinstance(data, list): |
||
12 | for item in data: |
||
13 | extract(item) |
||
14 | |||
15 | extract(search_hits) |
||
16 | return results |