Python 如何判断 SSH 成功

在自动化运维和开发过程中，我们经常需要通过 Python 脚本远程连接服务器执行命令。SSH（Secure Shell）是一种常用的远程登录协议，它提供了一种安全的方式来与远程服务器进行交互。本文将介绍如何在 Python 中使用 SSH 连接远程服务器，并判断连接是否成功。

1. 使用 Paramiko 库

Paramiko 是一个 Python 实现的 SSHv2 协议，它提供了一个接口来处理 SSH 连接、传输和通道。我们可以通过 Paramiko 库来实现 SSH 连接和命令执行。

首先，需要安装 Paramiko 库。可以通过 pip 命令进行安装：

pip install paramiko

2. 连接 SSH

使用 Paramiko 库连接 SSH 时，需要提供服务器的 IP 地址、端口号、用户名和密码。以下是连接 SSH 的基本代码示例：

import paramiko

def ssh_connect(hostname, port, username, password):
    try:
        # 创建 SSH 客户端对象
        client = paramiko.SSHClient()
        # 允许连接不在 known_hosts 文件中的主机
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        # 连接服务器
        client.connect(hostname, port, username, password)
        return client
    except Exception as e:
        print(f"连接失败: {e}")
        return None

# 连接参数
hostname = '192.168.1.1'
port = 22
username = 'your_username'
password = 'your_password'

# 连接 SSH
ssh_client = ssh_connect(hostname, port, username, password)

if ssh_client:
    print("连接成功")
else:
    print("连接失败")

3. 执行命令

连接成功后，我们可以通过 SSH 客户端执行命令。以下是执行命令的代码示例：

def ssh_execute_command(ssh_client, command):
    try:
        # 获取命令执行通道
        stdin, stdout, stderr = ssh_client.exec_command(command)
        # 获取命令执行结果
        result = stdout.read().decode()
        error = stderr.read().decode()
        if result:
            print(f"命令执行结果: {result}")
        if error:
            print(f"命令执行错误: {error}")
    except Exception as e:
        print(f"命令执行失败: {e}")

# 执行命令
command = 'ls -l'
ssh_execute_command(ssh_client, command)

4. 关闭连接

执行完命令后，需要关闭 SSH 连接。以下是关闭连接的代码示例：

def ssh_close(ssh_client):
    try:
        ssh_client.close()
        print("连接已关闭")
    except Exception as e:
        print(f"关闭连接失败: {e}")

# 关闭连接
ssh_close(ssh_client)

5. 判断 SSH 成功

在实际应用中，我们可能需要判断 SSH 连接是否成功。可以通过捕获连接过程中的异常来判断。以下是判断 SSH 成功的代码示例：

def is_ssh_success(hostname, port, username, password):
    try:
        ssh_client = ssh_connect(hostname, port, username, password)
        if ssh_client:
            return True
        else:
            return False
    except Exception as e:
        print(f"连接失败: {e}")
        return False

# 判断 SSH 成功
is_success = is_ssh_success(hostname, port, username, password)
if is_success:
    print("SSH 成功")
else:
    print("SSH 失败")

6. 关系图

以下是 SSH 连接过程中涉及的实体和它们之间的关系图：

erDiagram
    HOST ||--o| PORT : "has"
    USER ||--o| HOST : "connects_to"
    PASS ||--o| USER : "authenticates_with"
    CMD  ||--o| USER : "executes"
    OUT  ||--o| CMD  : "outputs"
    ERR  ||--o| CMD  : "errors"

结语

通过本文的介绍，我们学习了如何在 Python 中使用 Paramiko 库连接 SSH，并判断连接是否成功。在实际应用中，我们可以根据需要执行不同的命令，并处理命令执行的结果和错误。希望本文对您有所帮助。