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.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
|
|
||||||
|
@ -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