46 lines
2.1 KiB
Python
46 lines
2.1 KiB
Python
from API import APIError
|
|
|
|
|
|
async def safe_parse_api_data(data: dict or list, *path: str or int, idx: int = 0):
|
|
path = [*path]
|
|
if len(path) == idx+1:
|
|
if isinstance(path[idx], str):
|
|
if isinstance(data, dict):
|
|
if path[idx] in data.keys():
|
|
return data[path[idx]]
|
|
elif isinstance(path[idx], int):
|
|
if isinstance(data, list):
|
|
if len(data) > path[idx]:
|
|
return data[path[idx]]
|
|
else:
|
|
if isinstance(path[idx], str):
|
|
if isinstance(data, dict):
|
|
if path[idx] in data.keys():
|
|
parsed_data = await safe_parse_api_data(data[path[idx]], idx=idx+1, *path)
|
|
if not parsed_data:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return parsed_data
|
|
else:
|
|
if idx == 0:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return False
|
|
else:
|
|
if idx == 0:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return False
|
|
elif isinstance(path[idx], int):
|
|
if isinstance(data, list):
|
|
if len(data) > path[idx]:
|
|
parsed_data = await safe_parse_api_data(data[path[idx]], idx=idx+1, *path)
|
|
if not parsed_data:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return parsed_data
|
|
else:
|
|
if idx == 0:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return False
|
|
else:
|
|
if idx == 0:
|
|
raise APIError(f"Data parsing failed on path index {idx} - \nKey: {path[idx]} \nData: {data}")
|
|
return False
|