mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
e787e13d3d
* add IMF Financial Soundness Indicators Set * finish the APIEx * fix docstring * typo * lint * update readme * fix test..? * add Eastern Caribbean Currency Union to country map * mypy * more linting * more linting * improve title consistency between IRFCL and FSI * clear commented out code
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
"""IMF Helper Utilities."""
|
|
|
|
|
|
async def get_data(url: str) -> list[dict]:
|
|
"""Get data from the IMF API."""
|
|
# pylint: disable=import-outside-toplevel
|
|
|
|
from aiohttp.client_exceptions import ContentTypeError # noqa
|
|
from json.decoder import JSONDecodeError
|
|
from openbb_core.provider.utils.helpers import amake_request
|
|
from openbb_core.app.model.abstract.error import OpenBBError
|
|
|
|
try:
|
|
response = await amake_request(url, timeout=20)
|
|
except (JSONDecodeError, ContentTypeError) as e:
|
|
raise OpenBBError(
|
|
"Error fetching data; This might be rate-limiting. Try again later."
|
|
) from e
|
|
|
|
if "ErrorDetails" in response:
|
|
raise OpenBBError(
|
|
f"{response['ErrorDetails'].get('Code')} -> {response['ErrorDetails'].get('Message')}" # type: ignore
|
|
)
|
|
|
|
series = response.get("CompactData", {}).get("DataSet", {}).pop("Series", {}) # type: ignore
|
|
|
|
if not series:
|
|
raise OpenBBError(f"No time series data found -> {response}")
|
|
|
|
# If there is only one series, they ruturn a dict instead of a list.
|
|
if series and isinstance(series, dict):
|
|
series = [series]
|
|
|
|
return series
|