CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-01-26 08:50:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: simplify progress-tracker typing

Browse Source
This commit is contained in:
Matthias
2024-11-22 15:24:52 +01:00
parent 19f23106e7
commit b67beaa35a
3 changed files with 6 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
freqtrade/data/history/history_utils.py
View File

@@ -20,7 +20,7 @@ from freqtrade.exchange import Exchange
from freqtrade.plugins.pairlist.pairlist_helpers import dynamic_expand_pairlist
from freqtrade.util import dt_now, dt_ts, format_ms_time
from freqtrade.util.migrations import migrate_data
from freqtrade.util.progress_tracker import ProgressLike, retrieve_progress_tracker
from freqtrade.util.progress_tracker import CustomProgress, retrieve_progress_tracker
logger = logging.getLogger(__name__)
@@ -328,7 +328,7 @@ def refresh_backtest_ohlcv_data(
erase: bool = False,
data_format: str | None = None,
prepend: bool = False,
progress_tracker: ProgressLike | None = None,
progress_tracker: CustomProgress | None = None,
) -> list[str]:
"""
Refresh stored ohlcv data for backtesting and hyperopt operations.
@@ -495,7 +495,7 @@ def refresh_backtest_trades_data(
new_pairs_days: int = 30,
erase: bool = False,
data_format: str = "feather",
progress_tracker: ProgressLike | None = None,
progress_tracker: CustomProgress | None = None,
) -> list[str]:
"""
Refresh stored trades data for backtesting and hyperopt operations.
@@ -597,7 +597,7 @@ def download_data(
config: Config,
exchange: Exchange,
*,
progress_tracker: ProgressLike | None = None,
progress_tracker: CustomProgress | None = None,
) -> None:
"""
Download data function. Used from both cli and API.

1
freqtrade/util/__init__.py
View File

@@ -16,7 +16,6 @@ from freqtrade.util.ft_precise import FtPrecise
from freqtrade.util.measure_time import MeasureTime
from freqtrade.util.periodic_cache import PeriodicCache
from freqtrade.util.progress_tracker import ( # noqa F401
ProgressLike,
get_progress_tracker,
retrieve_progress_tracker,
)

14
freqtrade/util/progress_tracker.py
View File

@@ -1,10 +1,6 @@
from contextlib import AbstractContextManager
from typing import Protocol
from rich.progress import (
BarColumn,
MofNCompleteColumn,
TaskID,
TaskProgressColumn,
TextColumn,
TimeElapsedColumn,
@@ -14,19 +10,13 @@ from rich.progress import (
from freqtrade.util.rich_progress import CustomProgress
class ProgressLike(Protocol, AbstractContextManager["ProgressLike"]):
def add_task(self, description: str, *args, **kwargs) -> TaskID: ...
def update(self, task_id: TaskID, *, advance: float | None = None, **kwargs): ...
def retrieve_progress_tracker(pt: ProgressLike | None) -> ProgressLike:
def retrieve_progress_tracker(pt: CustomProgress | None) -> CustomProgress:
if pt is None:
return get_progress_tracker()
return pt
def get_progress_tracker(**kwargs) -> ProgressLike:
def get_progress_tracker(**kwargs) -> CustomProgress:
"""
Get progress Bar with custom columns.
"""