Python复制globals()实现timeit计时时禁用日志与打印的问题

核心问题分析 #

之前尝试copy.deepcopy(globals())失败是因为全局命名空间中的模块对象（如sys、logging）无法被序列化（pickle），抛出TypeError；浅拷贝globals()则因为模块是引用类型，修改沙箱内的logging配置会直接影响主程序，导致日志无法彻底禁用。

以下是两种高效的解决方案，无需拷贝全局命名空间，且能完全隔离沙箱与主程序的输出：

方法一：用上下文管理器封装禁用逻辑（推荐） #

通过上下文管理器临时修改stdout和logging配置，计时结束后自动恢复，代码模块化且复用性强：

import contextlib
import sys
import logging
import timeit

@contextlib.contextmanager
def disable_print():
    # 保存原始stdout
    original_stdout = sys.stdout
    try:
        # 重定向到空设备（兼容Windows）
        null_device = '/dev/null' if sys.platform != 'win32' else 'nul'
        sys.stdout = open(null_device, 'w')
        yield
    finally:
        # 恢复原始输出
        sys.stdout = original_stdout

@contextlib.contextmanager
def disable_logging():
    # 保存原始logging配置
    original_level = logging.root.level
    original_handlers = logging.root.handlers.copy()
    try:
        # 把日志级别设为高于CRITICAL，清空handler
        logging.root.setLevel(logging.CRITICAL + 1)
        logging.root.handlers = []
        yield
    finally:
        # 恢复原始日志配置
        logging.root.setLevel(original_level)
        logging.root.handlers = original_handlers

def time_func(func, *args, number=1000, **kwargs):
    def wrapped():
        # 在沙箱中执行目标函数
        with disable_print(), disable_logging():
            func(*args, **kwargs)
    # 计算平均单次运行时长
    total_time = timeit.timeit(wrapped, number=number)
    return total_time / number

方法二：在timeit的setup中临时修改配置 #

直接在timeit.timeit的setup参数中完成配置修改，最后在finally块恢复，适合简单场景：

import sys
import logging
import timeit

def time_func(func, *args, number=1000, **kwargs):
    # 保存主程序的原始配置
    original_stdout = sys.stdout
    original_log_level = logging.root.level
    original_handlers = logging.root.handlers.copy()
    
    # 沙箱初始化代码
    setup_code = f"""
import sys
import logging
# 重定向stdout到空设备
null_device = '/dev/null' if sys.platform != 'win32' else 'nul'
sys.stdout = open(null_device, 'w')
# 禁用logging输出
logging.root.setLevel(logging.CRITICAL + 1)
logging.root.handlers = []
"""
    
    # 要计时的函数调用语句
    stmt_code = f"func(*{args}, **{kwargs})"
    
    try:
        total_time = timeit.timeit(stmt=stmt_code, setup=setup_code, globals=globals(), number=number)
    finally:
        # 强制恢复主程序配置，避免沙箱影响主程序
        sys.stdout = original_stdout
        logging.root.setLevel(original_log_level)
        logging.root.handlers = original_handlers
    
    return total_time / number

关键注意事项 #

• 两种方法都确保了沙箱内的配置修改不会影响主程序，无论计时过程是否出错，都会执行恢复逻辑。
• 针对Windows系统做了空设备路径兼容，避免出现文件找不到的错误。
• 若需要极端隔离（如函数会修改全局状态），可考虑用子进程执行计时，但会增加额外开销，不适合短函数的高精度计时。

内容的提问来源于stack exchange，提问作者Zach Joseph