REmove Rich-progress wrapper again
This commit is contained in:
		| @@ -1,3 +1,2 @@ | ||||
| from freqtrade.util.ft_precise import FtPrecise  # noqa: F401 | ||||
| from freqtrade.util.periodic_cache import PeriodicCache  # noqa: F401 | ||||
| from freqtrade.util.rich_progress import FtProgress  # noqa: F401 | ||||
|   | ||||
| @@ -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]) | ||||
		Reference in New Issue
	
	Block a user