python - 全局关闭 Colorama 颜色代码

Question

Python 的 colorama非常适合装扮终端输出:

如您所见，这将添加 ANSI 代码并显示颜色，即使将日志记录模块用于标准输出也是如此。我用这样的语句来这样做:

logging.info(f"{Fore.CYAN}Query received from {current_user.username}{Style.RESET_ALL}")

但是，如果我要将日志记录模块指向一个文件，那么我到处都会得到一大堆我显然不想要的 ANSI 代码。

如果日志记录到文件被打开(或编写一个日志记录函数做同样的事情)，如果没有到处写if语句来抑制colorama，有没有办法在顶部全局关闭colorama脚本？

Answer 1

您可以暂时将 Fore、Back 和 Style 命名空间替换为包含常量空字符串的命名空间。

class DummyFore:
    BLACK=RED=GREEN=YELLOW=BLUE=MAGENTA=CYAN=WHITE=RESET=''

saved_Fore, Fore = Fore, DummyFore

logging.info(f"{Fore.CYAN}Query received from {current_user.username}{Style.RESET_ALL}") # should insert empty strings for the colour codes
Fore = saved_Fore
del saved_Fore # just to be thorough

它还应该可以将保存-恢复逻辑包装在上下文管理器中:

@contextlib.contextmanager
def without_colorama():
    global Fore
    saved_Fore, Fore = Fore, DummyFore
    yield
    Fore = saved_Fore
    # since saved_Fore is now a local, no need to `del`

with without_colorama():
    logging.info(f"{Fore.CYAN}Query received from {current_user.username}{Style.RESET_ALL}")

在所有示例代码中，我只处理了 Fore 命名空间；其他留作练习。