如何将Python函数输出内容同时打印到屏幕和文件

在日常的编程过程中，有时我们需要将函数的输出内容同时打印到屏幕和文件中，以便于调试和记录日志。本文将介绍三种实现这一目标的方法，帮助您轻松实现这一需求。

## 方法1：使用`contextlib.redirect_stdout`和自定义类

这种方法通过创建一个自定义的类，将标准输出流重定向到多个目标，包括屏幕和文件。

### 实现步骤：

1. **定义一个自定义的类 `Tee`**：

这个类的作用是将输出内容写入到多个文件对象中。

2. **使用 `contextlib.redirect_stdout` 重定向标准输出**：

通过 `contextlib.redirect_stdout` 将标准输出重定向到 `Tee` 类的实例。

```python

import contextlib

import sys

# 假设 summary 是你的函数

def summary():

print("这是 summary 函数的输出内容。")

# 创建一个写入模式的文件对象

logfile = open('summary_output.log', 'w')

# 将 stdout 重定向到文件和屏幕

class Tee(object):

def __init__(self, *files):

self.files = files

def write(self, obj):

for f in self.files:

f.write(obj)

f.flush() # 确保立即写入文件

def flush(self):

for f in self.files:

f.flush()

# 创建一个 Tee 对象，包含 sys.stdout 和 logfile

tee = Tee(sys.stdout, logfile)

# 使用 contextlib.redirect_stdout 重定向 stdout

with contextlib.redirect_stdout(tee):

summary()

logfile.close()

```

## 方法2：使用 `logging` 模块

Python 的 `logging` 模块提供了一种强大的日志记录方式，可以同时输出日志到屏幕和文件。

### 实现步骤：

1. **配置 `logging` 模块**：

配置 `logging` 模块使其同时将日志信息输出到控制台和文件中。

2. **使用 `logging` 模块记录日志**：

通过 `logging` 模块的 `info` 方法记录日志信息。

```python

import logging

# 配置 logging 模块

logging.basicConfig(level=logging.INFO, format='%(message)s')

logger = logging.getLogger()

# 添加文件处理器

file_handler = logging.FileHandler('summary_output.log')

file_handler.setLevel(logging.INFO)

file_handler.setFormatter(logging.Formatter('%(message)s'))

logger.addHandler(file_handler)

# 假设 summary 是你的函数

def summary():

logger.info("这是 summary 函数的输出内容。")

summary()

```

## 方法3：使用 `print` 函数的 `file` 参数

`print` 函数自带 `file` 参数，可以将输出内容写入到指定的文件对象中。结合自定义的函数实现多目标输出。

### 实现步骤：

1. **定义一个自定义的 `print` 函数**：

这个函数可以同时输出内容到屏幕和文件。

2. **使用该自定义函数进行输出**：

通过自定义的 `print` 函数将内容输出到多个目标。

```python

# 创建一个写入模式的文件对象

logfile = open('summary_output.log', 'w')

# 定义一个自定义的 print 函数

def dual_print(*args, **kwargs):

print(*args, **kwargs) # 打印到屏幕

print(*args, **kwargs, file=logfile) # 打印到文件

# 假设 summary 是你的函数

def summary():

dual_print("这是 summary 函数的输出内容。")

summary()

```

## 总结

本文介绍了三种将Python函数输出内容同时打印到屏幕和文件的方法：使用 `contextlib.redirect_stdout` 和自定义类、使用 `logging` 模块、以及使用 `print` 函数的 `file` 参数。根据具体的需求，您可以选择适合自己项目的方法来实现多目标输出。