python的subprocess调用bash脚本

我整理的一些关于【Python】的项目学习资料（附讲解～～）和大家一起分享、学习一下：

 https://d.51cto.com/Hpqqk2

Python的subprocess模块调用bash脚本

在系统编程和日常任务自动化中，Python的subprocess模块提供了一种强大的方式来调用外部程序和脚本。通过subprocess模块，我们可以方便地执行bash脚本，并与之进行交互。在本文中，我们将探讨如何使用Python的subprocess模块调用bash脚本，并给出具体的代码示例。

什么是subprocess模块？

subprocess模块是一个标准库模块，允许你生成子进程、与这些进程进行交互以及获取它们的返回值。在系统管理、自动化测试或数据处理任务中，能够方便地调用系统命令和脚本是非常重要的。

使用subprocess模块调用bash脚本

一个简单的示例

首先，我们创建一个简单的bash脚本，命名为hello.sh，内容如下：

登录后复制

#!/bin/bash
echo "Hello from bash script!"

• 1.
• 2.

接下来，我们使用Python调用这个bash脚本：

登录后复制

import subprocess

# 调用bash脚本
result = subprocess.run(['bash', 'hello.sh'], capture_output=True, text=True)

# 输出脚本的结果
print("返回码:", result.returncode)
print("输出:", result.stdout)
print("错误信息:", result.stderr)

• 1.
• 2.
• 3.
• 4.
• 5.
• 6.
• 7.
• 8.
• 9.

在这个例子中，我们使用subprocess.run函数来执行hello.sh脚本。注意这里传递了三个参数：

• ['bash', 'hello.sh']：这是一个列表，包含了要执行的命令及其参数。
• capture_output=True：表示我们希望捕获脚本的标准输出和标准错误。
• text=True：表示获得字符串类型的输出，而非字节类型。

处理脚本参数

有时，我们需要向bash脚本传递参数。让我们修改一下hello.sh脚本，增加一个参数的处理：

登录后复制

#!/bin/bash
echo "Hello, $1!"

• 1.
• 2.

现在我们可以通过Python传递参数来调用这个脚本：

登录后复制

import subprocess

name = "Alice"
result = subprocess.run(['bash', 'hello.sh', name], capture_output=True, text=True)

print("返回码:", result.returncode)
print("输出:", result.stdout)
print("错误信息:", result.stderr)

• 1.
• 2.
• 3.
• 4.
• 5.
• 6.
• 7.
• 8.

通过将name变量作为参数传递给脚本，我们可以动态改变输出内容。

错误处理

在调用外部命令时，处理潜在的错误是重要的。我们可以通过检查返回值来实现这一点：

登录后复制

import subprocess

try:
    result = subprocess.run(['bash', 'non_existent_script.sh'], capture_output=True, text=True, check=True)
except subprocess.CalledProcessError as e:
    print("命令执行失败，返回码:", e.returncode)
    print("错误输出:", e.stderr)

• 1.
• 2.
• 3.
• 4.
• 5.
• 6.
• 7.

通过将check=True参数传递给subprocess.run，我们可以确保在命令执行失败时引发异常。这使得我们的代码更加健壮。

进阶用法

管道与重定向

subprocess模块还支持管道与重定向。让我们看一个简单的示例，将echo命令的输出重定向到文件中：

登录后复制

import subprocess

with open('output.txt', 'w') as f:
    subprocess.run(['echo', 'Hello, World!'], stdout=f)

• 1.
• 2.
• 3.
• 4.

这会将Hello, World!写入output.txt文件中。

管道操作示例

假设我们需要对多个命令的输出进行处理，我们可以使用管道：

登录后复制

import subprocess

# 先运行一个命令，再将结果传给下一个命令
proc1 = subprocess.Popen(['echo', 'Hello, World!'], stdout=subprocess.PIPE)
proc2 = subprocess.Popen(['wc', '-w'], stdin=proc1.stdout, stdout=subprocess.PIPE)

proc1.stdout.close()  # 允许proc1收到SIGPIPE信号
output = proc2.communicate()[0]

print("单词数:", output.decode().strip())

• 1.
• 2.
• 3.
• 4.
• 5.
• 6.
• 7.
• 8.
• 9.
• 10.

在这个例子中，echo命令的输出被传递给了wc -w命令，后者用于统计单词数。

任务调度示例：甘特图

下面是一个简化的Gantt图示例，展示了使用Python调用脚本的任务安排：

交互式旅行图

在编程过程中，您可能会遇到不同的场景和解决方案。以下是一个简单的旅行图，展示了在Python中调用bash脚本的不同场景：

结论

通过使用Python的subprocess模块，我们能够高效地调用并与bash脚本交互。这种能力为自动化任务、系统管理和数据处理等领域带来了极大的便利。希望本文的示例和讨论能帮助你在自己的项目中更加自信地使用这一强大的工具。在自动化和脚本处理的道路上，Python将是你不可或缺的好友。

整理的一些关于【Python】的项目学习资料（附讲解～～），需要自取：

 https://d.51cto.com/Hpqqk2

原创作者: u_16213366 转载于: https://blog.51cto.com/u_16213366/11774362