From 9f34153c8424044cf1925d73d6827fced8a93624 Mon Sep 17 00:00:00 2001 From: Matthias Date: Fri, 13 Sep 2024 19:45:30 +0200 Subject: [PATCH] chore: update typing for reload function --- freqtrade/exchange/common.py | 4 ++++ freqtrade/exchange/exchange.py | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/freqtrade/exchange/common.py b/freqtrade/exchange/common.py index cac86ab3c..1bb738dcb 100644 --- a/freqtrade/exchange/common.py +++ b/freqtrade/exchange/common.py @@ -164,6 +164,10 @@ F = TypeVar("F", bound=Callable[..., Any]) def retrier(_func: F) -> F: ... +@overload +def retrier(_func: F, *, retries=API_RETRY_COUNT) -> F: ... + + @overload def retrier(*, retries=API_RETRY_COUNT) -> Callable[[F], F]: ... diff --git a/freqtrade/exchange/exchange.py b/freqtrade/exchange/exchange.py index 30c2aae6d..6e5b2720b 100644 --- a/freqtrade/exchange/exchange.py +++ b/freqtrade/exchange/exchange.py @@ -623,7 +623,7 @@ class Exchange: if self._exchange_ws: self._exchange_ws.reset_connections() - async def _api_reload_markets(self, reload: bool = False) -> None: + async def _api_reload_markets(self, reload: bool = False) -> Dict[str, Any]: try: return await self._api_async.load_markets(reload=reload, params={}) except ccxt.DDoSProtection as e: @@ -662,7 +662,7 @@ class Exchange: logger.debug("Performing scheduled market reload..") try: # on initial load, we retry 3 times to ensure we get the markets - retries = 3 if force else 0 + retries: int = 3 if force else 0 # Reload async markets, then assign them to sync api self._markets = retrier(self._load_async_markets, retries=retries)(reload=True) self._api.set_markets(self._api_async.markets, self._api_async.currencies)