def after_request(response: Response):
response_data = json.loads(response.get_data())
if 'ok' not in response_data or response_data['ok']:
data = dict()
data['ok'] = True
data['response'] = json.loads(response.get_data())
response.set_data(json.dumps(data, sort_keys=False))
return response
def get_content(http_out: Response) -> Any:
try:
response = json.loads(http_out.get_data().decode())
except Exception as e: # pragma: no cover
log.error("Failed to load response:\n{}", e)
raise ValueError(f"Malformed response: {http_out}")
return response
