自定义 TqdmToLogger 类 用法示例 tqdm_out = TqdmToLogger(logger) 进度条的输出重定向到日志 for step in tqdm(range(total_steps), desc="xxx", file=tqdm_out):
| 119 | |
| 120 | |
| 121 | class TqdmToLogger: |
| 122 | ''' |
| 123 | 自定义 TqdmToLogger 类 |
| 124 | 用法示例 |
| 125 | tqdm_out = TqdmToLogger(logger) |
| 126 | |
| 127 | 进度条的输出重定向到日志 |
| 128 | for step in tqdm(range(total_steps), desc="xxx", file=tqdm_out): |
| 129 | ''' |
| 130 | def __init__(self, logger, level=logging.INFO): |
| 131 | self.logger = logger |
| 132 | self.level = level |
| 133 | self.buf = '' |
| 134 | |
| 135 | def write(self, buf): |
| 136 | # 去除多余的空白字符，记录非空的进度信息 |
| 137 | self.buf = buf.strip() |
| 138 | if self.buf: |
| 139 | self.logger.log(self.level, self.buf) |
| 140 | |
| 141 | def flush(self): |
| 142 | pass |
| 143 | |
| 144 | |
| 145 | def test_setup_setup_logger(): |
nothing calls this directly
no outgoing calls
no test coverage detected