详解Python Paramiko模块

LUIZ • 2024-10-12 09:43 • 编程

一、Paramiko概述

Paramiko是一个用于SSH和SFTP的Python模块，它可以让Python程序连接到远程服务器，并进行远程命令执行、传输文件等操作。使用Paramiko可以轻松实现远程服务器管理和自动化部署。

二、安装Paramiko

安装Paramiko很简单，只需要使用pip命令即可：

pip install paramiko

三、连接远程服务器

使用Paramiko连接远程服务器需要先创建SSHClient对象，并设置连接参数。

下面是一个简单的例子：

import paramiko

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname='192.168.0.1', username='user', password='password')

stdin, stdout, stderr = ssh.exec_command('ls')
print(stdout.read().decode())

ssh.close()

上面的代码连接到了IP地址为192.168.0.1的服务器，使用用户名为user、密码为password进行登录，并执行了一个简单的命令ls，输出了执行结果。

四、传输文件

使用Paramiko可以轻松实现文件的传输，包括上传和下载。下面是一个上传文件的例子：

import paramiko

ssh = paramiko.Transport(('192.168.0.1', 22))
ssh.connect(username='user', password='password')

sftp = paramiko.SFTPClient.from_transport(ssh)
sftp.put('/local/path/to/file', '/remote/path/to/file')

ssh.close()

上面的代码连接到了IP地址为192.168.0.1、端口为22的服务器，使用用户名为user、密码为password进行登录，并将本地路径为/local/path/to/file的文件上传到了远程路径为/remote/path/to/file的目录下。

五、并发处理

使用Paramiko可以轻松实现多台服务器的并发处理。下面是一个使用多线程并发执行命令的例子：

import paramiko
import threading

def run_command(hostname, username, password, command):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname=hostname, username=username, password=password)

    stdin, stdout, stderr = ssh.exec_command(command)
    print(stdout.read().decode())

    ssh.close()

threads = []
hosts = [{'hostname': '192.168.0.1', 'username': 'user', 'password': 'password'},
         {'hostname': '192.168.0.2', 'username': 'user', 'password': 'password'},
         {'hostname': '192.168.0.3', 'username': 'user', 'password': 'password'}]
command = 'ls'

for host in hosts:
    thread = threading.Thread(target=run_command, args=(host['hostname'], host['username'], host['password'], command))
    threads.append(thread)
    thread.start()

for thread in threads:
    thread.join()

上面的代码并发执行了命令ls，连接到了IP地址为192.168.0.1、192.168.0.2、192.168.0.3的服务器，并使用指定的用户名和密码进行登录。

六、异常处理

使用Paramiko时，由于涉及到网络连接和数据传输，可能会出现各种异常情况，需要进行相应的处理。

下面是一个连接服务器异常的处理例子：

import paramiko

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

try:
    ssh.connect(hostname='192.168.0.1', username='user', password='password')
except paramiko.AuthenticationException as e:
    print('Authentication failed:', e)
except paramiko.SSHException as e:
    print('Unable to establish SSH connection:', e)
except paramiko.BadHostKeyException as e:
    print('Host key could not be verified:', e)

ssh.close()

上面的代码使用try-except语句捕获了3种常见的连接异常：身份验证失败、SSH连接失败和主机密钥无法验证。

七、总结

Paramiko是一个强大的Python模块，可以轻松实现SSH和SFTP的各种操作，方便实现远程服务器管理和自动化部署。通过本文的介绍，读者应该对Paramiko的用法有了更深入的了解。

原创文章，作者：LUIZ，如若转载，请注明出处：https://www.506064.com/n/142542.html