Python 传递数据给 subprocess.check_output

在本文中，我们将介绍如何使用 Python 中的 subprocess.check_output 将数据传递给子进程（subprocess）。subprocess 是 Python 的一个标准库，用于创建和管理子进程。

subprocess.check_output 简介

subprocess.check_output 是一个用于执行系统命令并返回输出结果的函数。它在执行系统命令时，可以接收一个字符串作为命令，并返回命令执行后的输出结果。

使用 subprocess.check_output 传递数据

要使用 subprocess.check_output 传递数据给子进程，首先需要将数据编码为字节串。Python 中提供了多种编码方式，例如 UTF-8、ASCII 等。下面是一个示例，演示了如何将字符串编码为字节串并通过 subprocess.check_output 传递给子进程。

import subprocess

data = "Hello, World!"
data_bytes = data.encode("utf-8")

output = subprocess.check_output(["echo", data_bytes])

print(output.decode("utf-8"))

在上面的示例中，我们首先定义了一个字符串 data，然后使用 encode() 方法将其编码为 UTF-8 格式的字节串 data_bytes。接下来，我们调用 subprocess.check_output 函数，传递子进程的命令和数据。在这个示例中，我们使用 “echo” 命令将字节串输出到控制台。最后，我们使用 decode() 方法将子进程输出的字节串解码为字符串，并打印出来。

传递其他类型的数据

除了字符串，我们还可以使用其他类型的数据通过 subprocess.check_output 传递给子进程。例如，我们可以传递字节串、整数、列表等。

import subprocess

# 传递字节串
data_bytes = b"\x48\x65\x6c\x6c\x6f\x2c\x20\x57\x6f\x72\x6c\x64\x21"

output = subprocess.check_output(["echo", data_bytes])

print(output.decode("utf-8"))

# 传递整数
data_int = 1234

output = subprocess.check_output(["echo", str(data_int)])

print(output.decode("utf-8"))

# 传递列表
data_list = [1, "two", 3.0]

output = subprocess.check_output(["echo", str(data_list)])

print(output.decode("utf-8"))

在上面的示例中，我们分别传递了一个字节串、一个整数和一个列表给子进程，并将它们通过 “echo” 命令输出到控制台。

注意，当传递列表时，我们需要先将其转换为字符串形式。子进程接收的是字符串参数，因此需要将列表转换为字符串，才能正确传递给子进程。

异常处理

在使用 subprocess.check_output 时，我们需要注意异常处理。如果执行的系统命令出错，或者子进程返回的结果不符合预期，都可能引发异常。

例如，当执行的系统命令返回非零状态码时，subprocess.CalledProcessError 异常将被抛出。我们可以使用 try-except 语句来捕获并处理这个异常。

import subprocess

try:
    output = subprocess.check_output(["nonexistent_command"])
    print(output.decode("utf-8"))
except subprocess.CalledProcessError as e:
    print("Command execution failed!")
    print("Exit code:", e.returncode)
    print("Output:", e.output.decode("utf-8"))