Python 在subprocess.check_output()中抑制stderr输出

在本文中，我们将介绍如何在Python的subprocess模块中使用check_output()函数时，抑制标准错误输出(stderr)的方法。通常情况下，当我们运行一条命令并使用check_output()函数来获取其输出时，终端上会同时显示标准输出和标准错误输出。但有时我们希望只获取标准输出而忽略标准错误输出，这时就需要使用一些技巧来抑制stderr的输出。

阅读更多：Python 教程

subprocess.check_output()函数简介

首先，我们来简单介绍一下subprocess模块和check_output()函数。subprocess模块是Python中用于创建子进程的一个强大工具，它提供了一组用于创建、启动和管理子进程的函数和类。而check_output()函数是subprocess模块中的一个常用函数，用于运行一个命令，并返回命令的输出结果。

下面是一个简单的示例，演示了如何使用check_output()函数运行一个命令并获取输出结果：

import subprocess

output = subprocess.check_output("ls", shell=True)
print(output)

以上代码运行时，会调用ls命令，并将其输出结果赋值给变量output，然后打印输出结果。

抑制stderr输出的方法

在subprocess.check_output()函数中，如果我们只想获取标准输出而忽略标准错误输出，可以通过重定向stderr来实现。具体而言，可以将stderr重定向到一个临时文件，然后再读取该临时文件获取标准错误输出。

下面是一个示例，演示了如何抑制stderr输出：

import subprocess
import os

# 重定向stderr到临时文件
stderr_file = open(os.devnull, 'w')
output = subprocess.check_output("ls nonexistent_file", shell=True, stderr=stderr_file)
stderr_file.close()

print(output)

以上代码中，ls nonexistent_file是一个不存在的命令，运行后会产生一个标准错误输出。通过将stderr重定向到临时文件os.devnull，我们将忽略这个错误输出，最终只获取了标准输出。

使用contextlib模块简化操作

如果我们经常需要抑制stderr输出，可以使用Python的contextlib模块来简化操作。contextlib模块提供了一个装饰器@contextmanager，可以将一个普通的生成器函数转化为一个上下文管理器。

下面是一个使用contextlib模块的示例：

import subprocess
import os
from contextlib import contextmanager

@contextmanager
def suppress_stderr():
    stderr_file = open(os.devnull, 'w')
    yield stderr_file
    stderr_file.close()

with suppress_stderr():
    output = subprocess.check_output("ls nonexistent_file", shell=True)

print(output)

以上代码中，我们定义了一个名为suppress_stderr()的生成器函数，并使用@contextmanager装饰器将其转化为上下文管理器。在上下文管理器中，我们打开了一个临时文件，将stderr重定向到该文件，并在yield语句之前进行输出操作。最后，我们使用with语句来使用上下文管理器，并在其中执行需要抑制stderr的代码。这样，我们就可以通过with语句块中的代码来获取标准输出，而忽略标准错误输出。

总结

本文介绍了在Python的subprocess.check_output()函数中抑制stderr输出的方法。我们可以通过将stderr重定向到临时文件，然后再读取该临时文件来抑制标准错误输出。此外，我们还介绍了使用contextlib模块来简化操作的方法。通过使用contextlib模块的@contextmanager装饰器，我们可以将一个生成器函数转化为上下文管理器，从而更方便地抑制stderr输出。

希望本文对大家在使用Python的subprocess模块时可以有所帮助！