mirror of
https://github.com/Yuukiy/JavSP.git
synced 2026-10-27 21:04:47 +08:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
"""改写内置的print函数，将其输出重定向到tqdm"""
|
|
import tqdm
|
|
import inspect
|
|
|
|
|
|
__all__ = ['TqdmOut']
|
|
|
|
|
|
# 普通输出和tqdm的输出混在一起会导致显示错乱，故在使用tqdm时要使用tqdm.write方法。
|
|
# 但是不希望又每个模块都使用tqdm.write方法，这样会显得混乱而且会导致与tqdm强耦合。
|
|
# 这个模块被设计来解决上面的问题：导入此模块后，全局覆盖内置的print，将输出都重定向到tqdm。
|
|
# 导入只在项目入口执行，这将改写所有后续导入的模块中的print的行为；
|
|
# 在单个模块内，不执行导入，这样的话在各个模块内仍然可以直接使用print
|
|
|
|
builtin_print = print
|
|
def flex_print(*args, **kwargs):
|
|
try:
|
|
tqdm.tqdm.write(*args, **kwargs)
|
|
except:
|
|
builtin_print(*args, ** kwargs)
|
|
# 替换内置的print
|
|
inspect.builtins.print = flex_print
|
|
|
|
|
|
class TqdmOut:
|
|
"""用于将logging的stream输出重定向到tqdm"""
|
|
@classmethod
|
|
def write(cls, s, file=None, nolock=False):
|
|
tqdm.tqdm.write(s, file=file, end='', nolock=nolock)
|