From d8a2c59b040baf5aa913efc4bc11db73089f4847 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 17 Nov 2024 12:40:24 +0100 Subject: [PATCH] chore: Add proper typing to jobContainer response --- freqtrade/rpc/api_server/api_schemas.py | 3 ++- freqtrade/rpc/api_server/webserver_bgwork.py | 8 +++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/freqtrade/rpc/api_server/api_schemas.py b/freqtrade/rpc/api_server/api_schemas.py index eaa94678a..b0f080250 100644 --- a/freqtrade/rpc/api_server/api_schemas.py +++ b/freqtrade/rpc/api_server/api_schemas.py @@ -12,6 +12,7 @@ from pydantic import ( from freqtrade.constants import DL_DATA_TIMEFRAMES, IntOrInf from freqtrade.enums import MarginMode, OrderTypeValues, SignalDirection, TradingMode from freqtrade.ft_types import ValidExchangesType +from freqtrade.rpc.api_server.webserver_bgwork import ProgressTask class ExchangeModePayloadMixin(BaseModel): @@ -50,7 +51,7 @@ class BackgroundTaskStatus(BaseModel): status: str running: bool progress: float | None = None - progress_tasks: dict[str, Any] | None = None + progress_tasks: dict[str, ProgressTask] | None = None error: str | None = None diff --git a/freqtrade/rpc/api_server/webserver_bgwork.py b/freqtrade/rpc/api_server/webserver_bgwork.py index 8b1a2bb6c..7ba205f56 100644 --- a/freqtrade/rpc/api_server/webserver_bgwork.py +++ b/freqtrade/rpc/api_server/webserver_bgwork.py @@ -4,12 +4,18 @@ from uuid import uuid4 from freqtrade.exchange.exchange import Exchange +class ProgressTask(TypedDict): + progress: float + total: float + description: str + + class JobsContainer(TypedDict): category: Literal["pairlist", "download_data"] is_running: bool status: str progress: float | None - progress_tasks: NotRequired[dict[str, Any]] + progress_tasks: NotRequired[dict[str, ProgressTask]] result: Any error: str | None