给Linux进行批量部署 - 滴滴嘟嘟博客

一早上，说什么前阵子装的很多agent，没有发现连接到服务器，可能有些人安装的不对，要重新装一下。

自己以前写过个shell，用于批量配置，这不刚好改改就能用了。

情况是，各个linux的主机有不同的密码，账户都是root，那就需要建立一个账户文件，包含远端机器的IP地址，root，还有密码。

然后逐行读取这三个信息，用来ssh连接远端机器，连接完成后，提前将要执行的动作写入一个shell文件，然后sftp到远端机器去，再进行执行，执行后将输出写入当前的目录的log.txt文件中。

目录结构如下：

├── account.csv
├── command.py
├── install.sh
└── log.txt

其中install.sh是要执行的shell文件，log.txt是执行后生成的日志。

account.csv的格式如下：用空格隔开，IP地址，账户名和密码

1
10.1.2.3 root password
2
10.1.2.5 root newpass

下面是python的代码：

1
import csv
2
import paramiko
3

4

5
def ssh_command(ip, username, password):
6
    ssh_client = paramiko.SSHClient()
7
    ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
8

9
    try:
10
        ssh_client.connect(ip, username=username, password=password)
11

12
        sftp = ssh_client.open_sftp()
13
        local_sh = "install.sh"
14
        remote_path = '/tmp/install.sh'
15
        sftp.put(local_sh, remote_path)
16
        sftp.close()
17

18
        stdin, stdout, stderr = ssh_client.exec_command(
19
            'chmod +x {} && {}'.format(remote_path, remote_path))
20

21
        # stdin, stdout, stderr = ssh_client.exec_command('/tmp/install.sh')
22
        # 等待命令执行完成
23
        exit_status = stdout.channel.recv_exit_status()
24
        # 处理命令执行结果并记录到 log.txt 文件中
25
        with open('log.txt', 'a') as log_file:
26
            log_file.write(f"=== Output for {ip} ===\n")
27
            log_file.write(stdout.read().decode("utf-8"))
28
            log_file.write(f"Exit status: {exit_status}\n")
29
    except paramiko.AuthenticationException:
30
        print(f"Failed to log in to {ip} with provided credentials.")
31
    finally:
32
        ssh_client.close()
33

34

35
with open('account.csv', 'r') as csvfile:
36
    csv_reader = csv.reader(csvfile, delimiter=' ')
37
    # next(csv_reader)  # 跳过标题行
38
    for row in csv_reader:
39
        ip = row[0]
40
        username = row[1]
41
        password = row[2]
42
        # command_to_run = "install.sh"  # 替换为要执行的脚本命令
43
        ssh_command(ip, username, password)

通过测试，可以实现效果。后面也可以用os.remove删掉上传到远端的shell文件。