### 使用装饰器捕获并记录函数中的 `print` 输出和异常错误到日志文件 在 Python 中，可以创建一个装饰器来拦截函数的 `print` 输出并捕获异常，然后将这些信息记录到日志文件中。这种做法在调试和监控应用程序时非常有用，特别是在生产环境中。 以下是一个完整的实现示例： #### 1. 装饰器设计目标 - 捕获函数中的 `print` 输出，并将其重定向到日志文件。 - 捕获函数中发生的异常，并记录异常信息。 - 将捕获的输出和异常信息记录到日志文件中。 #### 2. 实现代码 ```python import logging import sys from functools import wraps import io # 配置日志系统 logging.basicConfig( filename='function_output_and_errors.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s' ) def log_function_output_and_errors(func): @wraps(func) def wrapper(*args, **kwargs): # 创建 StringIO 对象来捕获 print 输出 captured_output = io.StringIO() # 保存原始的 stdout original_stdout = sys.stdout # 重定向 stdout 到 StringIO sys.stdout = captured_output try: # 执行目标函数 result = func(*args, **kwargs) # 获取捕获的输出 output = captured_output.getvalue() # 将输出记录到日志 if output: logging.info(f"Function '{func.__name__}' 输出:\n{output.strip()}") return result except Exception as e: # 记录异常信息 logging.exception(f"Function '{func.__name__}' 发生异常: {e}") finally: # 恢复原始 stdout sys.stdout = original_stdout return wrapper ``` #### 3. 使用示例 ```python @log_function_output_and_errors def example_function(x, y): print(f"参数 x: {x}, y: {y}") result = x / y print(f"结果: {result}") return result # 调用函数 example_function(10, 0) ``` #### 4. 日志文件内容示例 执行上述代码后，`function_output_and_errors.log` 文件中将记录如下内容： ``` 2025-04-05 12:34:56,789 - INFO - Function 'example_function' 输出: 参数 x: 10, y: 0 2025-04-05 12:34:56,789 - ERROR - Function 'example_function' 发生异常: division by zero Traceback (most recent call last): File "your_script.py", line 25, in wrapper result = func(*args, **kwargs) File "your_script.py", line 45, in example_function result = x / y ZeroDivisionError: division by zero ``` #### 5. 关键点说明 - **`io.StringIO()`**：用于临时捕获 `print` 输出。 - **`sys.stdout` 重定向**：将标准输出重定向到内存中的缓冲区，以便记录输出。 - **`logging.exception()`**：记录异常信息，包括堆栈跟踪。 - **`@wraps(func)`**：保留函数的元数据，确保装饰器不会影响函数的签名。 该装饰器可以应用于任何函数，以捕获其输出和异常，并记录到日志文件中，从而帮助开发者更好地了解程序运行状态和调试问题。 ---