freqtrade_origin/freqtrade/util/rich_progress.py

from collections.abc import Callable

from rich.console import ConsoleRenderable, Group, RichCast
from rich.progress import Progress


class CustomProgress(Progress):
    def __init__(
        self,
        *args,
        cust_objs: list[ConsoleRenderable] | None = None,
        cust_callables: list[Callable[[], ConsoleRenderable]] | None = None,
        **kwargs,
    ) -> None:
        self._cust_objs = cust_objs or []
        self._cust_callables = cust_callables or []

        super().__init__(*args, **kwargs)

    def get_renderable(self) -> ConsoleRenderable | RichCast | str:
        objs = [obj for obj in self._cust_objs]
        for cust_call in self._cust_callables:
            objs.append(cust_call())
        renderable = Group(*objs, *self.get_renderables())
        return renderable