REmove Rich-progress wrapper again

This commit is contained in:
Matthias 2023-04-09 16:21:30 +02:00
parent 40450ebecc
commit b6aac5079b
2 changed files with 0 additions and 28 deletions

View File

@ -1,3 +1,2 @@
from freqtrade.util.ft_precise import FtPrecise # noqa: F401 from freqtrade.util.ft_precise import FtPrecise # noqa: F401
from freqtrade.util.periodic_cache import PeriodicCache # noqa: F401 from freqtrade.util.periodic_cache import PeriodicCache # noqa: F401
from freqtrade.util.rich_progress import FtProgress # noqa: F401

View File

@ -1,27 +0,0 @@
import logging
import sys
from contextlib import contextmanager
from rich.progress import Progress
@contextmanager
def FtProgress(*args, **kwargs):
"""
Wrapper around rich.progress.Progress to fix issues with logging.
"""
try:
__logger = kwargs.pop('logger', None)
streamhandlers = [x for x in __logger.root.handlers if type(x) == logging.StreamHandler]
__prior_stderr = []
with Progress(*args, **kwargs) as progress:
for handler in streamhandlers:
__prior_stderr.append(handler.stream)
handler.setStream(sys.stderr)
yield progress
finally:
for idx, handler in enumerate(streamhandlers):
handler.setStream(__prior_stderr[idx])