REmove Rich-progress wrapper again
This commit is contained in:
parent
40450ebecc
commit
b6aac5079b
@ -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])
|
Loading…
Reference in New Issue
Block a user