summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorhjoaquim <h.joaquim@campus.fct.unl.pt>2024-02-21 13:50:43 +0000
committerhjoaquim <h.joaquim@campus.fct.unl.pt>2024-02-21 13:50:43 +0000
commit0af4b7900122827cf6df01afd9bce87fa4406362 (patch)
tree57bc3f0de06b15a2e181802cb4bcf6c0faf27436
parent4e98eb4895adc4cf444a2b63c32a6274ba2451e0 (diff)
support for multiple symbols
-rw-r--r--openbb_platform/providers/tradingeconomics/openbb_tradingeconomics/models/market_historical.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/openbb_platform/providers/tradingeconomics/openbb_tradingeconomics/models/market_historical.py b/openbb_platform/providers/tradingeconomics/openbb_tradingeconomics/models/market_historical.py
index 6ccb1b41ecb..7419f015174 100644
--- a/openbb_platform/providers/tradingeconomics/openbb_tradingeconomics/models/market_historical.py
+++ b/openbb_platform/providers/tradingeconomics/openbb_tradingeconomics/models/market_historical.py
@@ -10,7 +10,7 @@ from openbb_core.provider.standard_models.market_historical import (
)
from openbb_core.provider.utils.helpers import (
ClientResponse,
- amake_request,
+ amake_requests,
get_querystring,
)
from pandas import to_datetime
@@ -68,9 +68,14 @@ class TEMarketHistoricalFetcher(
"""Return the raw data from the TE endpoint."""
api_key = credentials.get("tradingeconomics_api_key") if credentials else ""
+ symbols = query.symbol.split(",")
+
base_url = "https://api.tradingeconomics.com/markets/historical"
query_str = get_querystring(query.model_dump(), ["symbol"])
- url = f"{base_url}/{query.symbol}?{query_str}&c={api_key}"
+ urls = []
+
+ for symbol in symbols:
+ urls.append(f"{base_url}/{symbol}?{query_str}&c={api_key}")
async def callback(response: ClientResponse, _: Any) -> Union[dict, List[dict]]:
"""Return the response."""
@@ -78,7 +83,7 @@ class TEMarketHistoricalFetcher(
raise RuntimeError(f"Error in TE request -> {await response.text()}")
return await response.json()
- return await amake_request(url, response_callback=callback, **kwargs)
+ return await amake_requests(urls, response_callback=callback, **kwargs)
# pylint: disable=unused-argument
@staticmethod