python批量执行shell

原创

mob64ca12e77061 2024-09-14 07:05:32 ©著作权

文章标签 Shell Python 甘特图 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e77061的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python批量执行Shell命令

在日常的开发和运维中，我们常常需要在多个服务器上批量执行Shell命令。这一过程如果手动操作，既繁琐又容易出错。因此，利用Python自动化这一过程将大大提高我们的效率。本文将向您介绍如何使用Python批量执行Shell命令，并附带代码示例和任务计划甘特图。

Python使用`subprocess`模块执行Shell命令

Python提供了多个模块来执行外部命令，其中最常用的是subprocess模块。subprocess允许我们创建新进程、与进程建立连接并获取其返回值。下面我们将通过一个简单的例子来说明如何使用subprocess模块执行Shell命令。

基础示例

import subprocess

# 定义要执行的Shell命令
command = 'echo Hello, World!'

# 执行命令并获取输出
result = subprocess.run(command, shell=True, capture_output=True, text=True)

# 输出结果
print(result.stdout)

在这个示例中，我们使用subprocess.run来执行一个简单的Shell命令echo Hello, World!。参数capture_output=True意味着我们希望捕获这个命令的输出，text=True表示我们希望输出为字符串类型。最后，我们通过result.stdout来获取命令输出并打印。

批量执行多个命令

如果需要批量执行多个命令，我们可以将命令放在一个列表中，然后利用循环进行遍历。以下是一个更复杂的示例：

import subprocess

# 定义要执行的Shell命令列表
commands = [
    'echo Command 1 executed',
    'echo Command 2 executed',
    'echo Command 3 executed'
]

for command in commands:
    result = subprocess.run(command, shell=True, capture_output=True, text=True)
    print(result.stdout)

在此示例中，我们创建了一个命令列表commands，并使用for循环逐一执行每个命令。这种方式非常适合于需要在多个服务器上运行相同命令的场景。

通过SSH进行远程执行

在实际运维中，我们往往需要在远程服务器上执行Shell命令。可以使用paramiko库来实现SSH连接和命令执行。以下是一个示例：

首先，你需要安装paramiko库：

pip install paramiko

然后，我们可以使用以下代码示例：

import paramiko

# 定义远程服务器的连接信息
hostname = 'remote.server.com'
port = 22
username = 'your_username'
password = 'your_password'

# 创建SSH客户端
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())  # 自动添加主机密钥

# 连接到服务器
ssh.connect(hostname, port=port, username=username, password=password)

# 要执行的命令
commands = [
    'echo Remote Command 1 executed',
    'echo Remote Command 2 executed'
]

for command in commands:
    stdin, stdout, stderr = ssh.exec_command(command)
    print(stdout.read().decode())

# 关闭SSH连接
ssh.close()

在此代码中，我们创建了一个SSH客户端，通过ssh.connect进行连接，然后循环遍历命令并通过ssh.exec_command逐一执行。

任务管理与进度跟踪

在批量执行命令的过程中，任务管理是非常重要的一部分。为了更好地管理任务，我们可以使用甘特图来展示不同任务的执行时间和状态。以下是一个使用Mermaid语法的甘特图示例：

gantt
    title 项目执行时间线
    dateFormat  YYYY-MM-DD
    section 任务1
    执行命令1          :done,  des1, 2023-10-01, 1d
    执行命令2          :active, des2, after des1, 1d
    section 任务2
    执行远程命令1      :         des3, 2023-10-03, 1d
    执行远程命令2      :         des4, after des3, 1d

在这张甘特图中，我们可以清晰地看到每个任务的状态和执行时间，有助于我们有效地进行任务计划和资源分配。