网络工程师学Python-SSH客户端和服务器库Paramiko

Paramiko是Python的一个SSH客户端和服务器库，可以实现SSHv2协议的加密和身份验证功能，支持SFTP和SCP协议。在本文中，我们将介绍如何使用Paramiko库来建立SSH连接、执行远程命令、传输文件等。

安装Paramiko

要使用Paramiko库，我们需要先安装它。可以使用pip命令进行安装：

pip install paramiko

建立SSH连接

使用Paramiko库建立SSH连接非常简单，只需指定主机名、用户名和密码即可。以下是一个简单的示例：

import paramiko

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

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

ssh.close()

在上述示例中，我们首先创建一个SSHClient对象，并将主机名、用户名和密码作为参数传递给connect()方法。然后，使用exec_command()方法执行ls命令，并将结果打印到控制台。最后，我们关闭SSH连接。

执行远程命令

使用Paramiko库执行远程命令也很简单，只需将命令作为字符串传递给exec_command()方法。以下是一个示例：

import paramiko

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

command = 'ls -l'
stdin, stdout, stderr = ssh.exec_command(command)

for line in stdout.readlines():
    print(line.strip())

ssh.close()

在上述示例中，我们使用exec_command()方法执行ls -l命令，并使用readlines()方法读取输出结果，并打印到控制台。需要注意的是，在某些情况下，需要使用strip()方法删除输出结果中的换行符。

传输文件

使用Paramiko库传输文件也很简单，只需使用SFTPClient类即可。以下是一个示例：

import paramiko

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

sftp = ssh.open_sftp()
sftp.put('local_file_path', 'remote_file_path')
sftp.get('remote_file_path', 'local_file_path')

sftp.close()
ssh.close()

在上述示例中，我们首先创建一个SFTPClient对象，并使用put()方法将本地文件传输到远程主机上。然后，使用get()方法将远程文件传输到本地主机上。最后，我们关闭SFTPClient和SSH连接。

使用密钥认证

除了使用用户名和密码进行身份验证外，我们还可以使用SSH密钥进行身份验证。以下是一个示例：

import paramiko

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('hostname', username='username', pkey=key)

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

ssh.close()

在上述示例中，我们首先使用from_private_key_file()方法加载SSH私钥文件，并将私钥作为参数传递给connect()方法。然后，我们使用exec_command()方法执行ls命令，并将结果打印到控制台。最后，我们关闭SSH连接。

高级选项

Paramiko库还提供了许多高级选项，例如超时、端口转发、密钥管理等。以下是一个示例：

import paramiko

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('hostname', username='username', password='password', port=22, timeout=10)

# 开启端口转发
transport = ssh.get_transport()
local_port = 8080
remote_host = 'google.com'
remote_port = 80
transport.request_port_forward('', local_port, remote_host, remote_port)

# 密钥管理
key = paramiko.RSAKey.generate(2048)
private_key_string = key.export_key('PEM')
public_key_string = key.get_base64()

ssh.close()

在上述示例中，我们首先使用connect()方法连接到远程主机，并指定了端口号和超时时间。然后，我们使用get_transport()方法获取SSH传输对象，并使用request_port_forward()方法开启端口转发。最后，我们使用generate()方法生成RSA密钥，并使用export_key()和get_base64()方法导出私钥和公钥。

结论

在本文中，我们介绍了如何使用Python Paramiko库实现SSH客户端和服务器功能，包括建立SSH连接、执行远程命令、传输文件等。此外，我们还介绍了如何使用SSH密钥进行身份验证，并讨论了一些高级选项。使用Paramiko库，我们可以轻松地编写Python脚本来管理远程服务器和网络设备。