python中如何写ssh登录

Hey小伙伴们，今天要和大家分享一个超级实用的小技巧——如何在Python中实现SSH登录，是不是听起来就觉得很厉害呢？别急，跟着我一步步来，保证你也能轻松这个技能。

我们得知道，SSH（Secure Shell）是一种网络协议，用于加密方式远程登录到服务器，在Python中，我们可以通过第三方库来实现SSH登录，最常用的就是paramiko库，这个库功能强大，可以帮助我们完成SSH连接、执行命令等操作。

准备工作

在开始之前，我们需要安装paramiko库，如果你还没有安装，可以通过pip来安装：

pip install paramiko

连接到SSH

安装好paramiko后，我们就可以开始编写代码了，我们需要创建一个SSH客户端，并连接到服务器，这里是一个基本的连接示例：

import paramiko
创建SSH对象
ssh = paramiko.SSHClient()
添加新的SSH密钥
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
连接到服务器
ssh.connect(hostname='your_server_ip', port=22, username='your_username', password='your_password')

我们首先导入了paramiko库，然后创建了一个SSHClient对象，我们设置了服务器的密钥策略，这里我们选择了自动添加新的主机密钥，我们通过connect方法连接到服务器，需要提供服务器的IP地址、端口号、用户名和密码。

执行命令

连接到服务器后，我们就可以执行命令了。paramiko提供了exec_command方法，允许我们执行任意的SSH命令：

执行命令
stdin, stdout, stderr = ssh.exec_command('ls -l')
获取命令输出
print(stdout.read().decode('utf-8'))

这里我们执行了ls -l命令来列出当前目录下的文件。exec_command方法返回三个对象：stdin、stdout和stderr，分别对应命令的标准输入、输出和错误输出，我们通过stdout.read().decode('utf-8')获取命令的输出，并打印出来。

处理错误

在实际应用中，我们还需要处理可能出现的错误，如果服务器连接失败或者命令执行出错，我们可以通过捕获异常来处理：

try:
    # 连接到服务器
    ssh.connect(hostname='your_server_ip', port=22, username='your_username', password='your_password')
    # 执行命令
    stdin, stdout, stderr = ssh.exec_command('ls -l')
    # 获取命令输出
    print(stdout.read().decode('utf-8'))
except paramiko.AuthenticationException:
    print("认证失败，请检查用户名和密码")
except paramiko.SSHException as e:
    print(f"SSH连接失败：{e}")
except Exception as e:
    print(f"发生错误：{e}")
finally:
    # 关闭连接
    ssh.close()

在这个例子中，我们通过try...except结构来捕获和处理可能发生的异常，如果认证失败，我们打印一条错误消息；如果是SSH连接问题，我们打印出具体的错误信息；无论是否发生异常，我们都确保在finally块中关闭SSH连接。

进阶应用

如果你需要在SSH会话中执行多个命令，或者需要更复杂的交互，paramiko也提供了invoke_shell方法，允许你创建一个交互式的会话：

创建交互式会话
channel = ssh.invoke_shell()
发送命令
channel.send('ls -l
')
等待命令执行
time.sleep(0.5)
获取输出
output = channel.recv(1024).decode('utf-8')
print(output)
关闭会话
channel.close()

在这个例子中，我们首先创建了一个交互式会话，然后发送了一个命令，通过recv方法，我们可以接收命令的输出，不要忘记关闭会话。