使用 subprocess.check_output 在 Python 中执行命令

如果你在 Python 中需要执行一些系统命令或外部程序，可以使用 subprocess.check_output。这是一个很实用的函数，可以轻松地在 Python 中调用命令行工具，并获取它们的输出。下面就来详细介绍使用方法和注意事项。

运行命令

首先，我们需要导入 subprocess 模块：

import subprocess

然后，就可以使用 check_output 函数来运行命令了。这个函数有两个参数：

• args：命令行命令及其参数。
• shell：是否在 shell 中执行命令。默认为 False。

output = subprocess.check_output(args, shell=False)

下面是一个例子，运行 ls 命令：

output = subprocess.check_output(["ls", "-l"])
print(output)

获取输出

check_output 函数将返回命令的输出，可以将其打印出来：

print(output)

如果想要获得更加结构化的输出，可以使用 decode 方法将字节码转换为字符串：

print(output.decode())

错误处理

如果命令执行失败，check_output 函数将抛出一个 CalledProcessError 异常，其中会包含错误代码和错误消息。

为了避免程序崩溃，我们需要捕获这个异常并处理它。例如，可以打印错误消息：

try:
    output = subprocess.check_output(["ls", "non_existent_file"], stderr=subprocess.STDOUT)
except subprocess.CalledProcessError as e:
    print("Error: ", e.output.decode())

安全考虑

在使用 check_output 函数时，应该小心注入攻击。只有在你知道你正在运行的命令是什么，并信任它可以安全地执行时，才应该使用这个函数。

如果给出的命令是由用户提供的，应该使用 subprocess.Popen 函数，并将参数作为列表传递给它。这样可以避免 shell 注入攻击。

总结

subprocess.check_output 函数可以帮助你在 Python 中执行系统命令并获取其输出。使用这个函数时需要注意错误处理和注入攻击问题。

（注：本文为 AI 辅助创作内容，可能不完全符合人类写作和技术规范。如有问题，欢迎指出！）