《python每天一小段》--（2）使用Paramiko库，批量操作linux服务器

news2024/9/30 15:24:47

使用Paramiko库，批量操作多Linux服务器

使用Paramiko库建立SSH连接并执行命令获取多个Linux服务器的磁盘信息。通过这个例子，你可以学习到如何使用Python自动化操作远程服务器。

操作如图：

查看了两台服务器的磁盘容量：
在这里插入图片描述

安装模块

使用jupyter安装paramiko模块：pip install paramiko，如果是windows系统使用cmd即可。

在这里插入图片描述

代码解析

以下是代码的详细解释：

import paramiko
import sys

def sshExeCMD(ip, username, password, port):
    ssh_client = paramiko.SSHClient()
    ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    try:
        ssh_client.connect(hostname=ip, port=port, username=username, password=password)
    except Exception as e:
        print("连接服务器%s失败，请检查IP地址和端口是否正确！" % ip)
        print(e)
        sys.exit(1)

    stdin, stdout, stderr = ssh_client.exec_command("df -hT")
    print("服务器%s磁盘信息：" % ip)
    print(stdout.read().decode('utf-8'))

    ssh_client.close()

if __name__ == '__main__':
    servers = {
        "192.168.31.223": {
            "username": "root",
            "password": "123456",
            "port": 22
        },
        "192.168.31.162": {
            "username": "root",
            "password": "123456",
            "port": 22
        }
    }
    for ip, info in servers.items():
        sshExeCMD(
            ip=ip,
            username=info.get("username"),
            password=info.get("password"),
            port=info.get("port")
        )

代码说明

导入必要的模块：paramiko用于SSH连接，sys用于系统相关操作。
定义了一个名为sshExeCMD的函数，用于执行SSH连接和命令执行操作。函数接受四个参数：ip（服务器的IP地址），username（用户名），password（密码）和port（SSH端口号）。
在函数内部，创建了一个SSH客户端对象ssh_client，并使用paramiko.AutoAddPolicy()设置了自动添加服务器主机密钥到本地known_hosts文件的策略。
使用ssh_client.connect()方法连接到指定的服务器。如果连接失败，打印错误信息并退出程序。
如果连接成功，使用ssh_client.exec_command()方法执行命令df -hT，该命令用于获取磁盘信息。返回的结果包括标准输入、标准输出和标准错误。
打印服务器IP和磁盘信息。
关闭SSH连接。
在__name__ == '__main__'的条件下，定义了一个包含多个服务器信息的字典servers。
使用for循环遍历servers字典中的每个服务器信息。
调用sshExeCMD函数，并传入服务器的IP地址、用户名、密码和端口号作为参数。