Python中如何执行Shell命令？-Python教程-PHP中文网

在python中，执行shell命令可以通过subprocess模块实现。1) 使用subprocess.run()执行简单命令，如echo。2) 通过input参数传递数据给命令，如grep。3) 使用check=true处理命令失败，并捕获calledprocesserror。4) 利用subprocess.popen执行一系列命令并传递数据，如grep和wc。使用时需注意安全性、性能和错误处理。

Python中如何执行Shell命令？

在Python中执行Shell命令是个挺酷的功能，特别是当你需要将脚本与操作系统的功能结合起来时。让我们来看看如何做到这一点，以及一些我在实际项目中遇到的小窍门和坑。

首先，Python的subprocess模块是执行Shell命令的首选工具。使用它，你可以轻松地从Python代码中调用任何命令行程序。来看一个简单的例子：

import subprocess

# 执行简单的echo命令
result = subprocess.run(['echo', 'Hello, World!'], capture_output=True, text=True)
print(result.stdout)  # 输出: Hello, World!

这个例子展示了如何使用subprocess.run()来执行一个简单的echo命令。capture_output=True和text=True参数允许我们捕获输出并以文本形式返回。

但这只是冰山一角。让我们深入探讨一下如何处理更复杂的场景，以及一些常见的陷阱。

在实际项目中，我发现处理命令的输入和输出是个关键点。你可能需要将数据传递给命令，或者从命令中读取输出。这里有一个例子，展示了如何将字符串作为输入传递给命令，并读取其输出：

import subprocess

# 执行grep命令并传递输入
input_data = "Hello\nWorld\nPython"
result = subprocess.run(['grep', 'o'], input=input_data, capture_output=True, text=True)
print(result.stdout)  # 输出: Hello\nWorld

这个例子使用grep命令来搜索包含字母'o'的行，并通过input参数传递输入数据。

Shadows Python Sensei

Python 最佳实践助手——代码规范、设计模式、性能优化、测试与类型注解。适用于编写或审查 Python 代码。

下载

不过，执行Shell命令时需要小心处理错误。subprocess.run()默认情况下会抛出异常，如果命令执行失败。可以通过检查result.returncode来处理这种情况：

import subprocess

try:
    result = subprocess.run(['ls', 'non_existent_file'], capture_output=True, text=True, check=True)
except subprocess.CalledProcessError as e:
    print(f"Command failed with error: {e}")
    print(f"Error output: {e.stderr}")

在这个例子中，我们使用check=True来让subprocess.run()在命令失败时抛出CalledProcessError，然后捕获并处理这个异常。

另一个常见的场景是需要执行一系列命令，并在它们之间传递数据。这时，subprocess.Popen就派上用场了。让我们来看一个例子：

import subprocess

# 使用Popen来执行一系列命令
grep_process = subprocess.Popen(['grep', 'o'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
wc_process = subprocess.Popen(['wc', '-l'], stdin=grep_process.stdout, stdout=subprocess.PIPE, text=True)

input_data = "Hello\nWorld\nPython"
grep_process.stdin.write(input_data)
grep_process.stdin.close()

output = wc_process.communicate()[0]
print(output.strip())  # 输出: 2

在这个例子中，我们使用Popen来创建两个进程：一个运行grep，另一个运行wc -l。我们将grep的输出直接传递给wc -l，然后读取最终的输出。

在实际应用中，我发现使用subprocess时需要注意几点：