python如何使用ssh登录

一、PYTHON使用SSH登录的方法概述

使用Python进行SSH登录可以通过多种方式实现，最常用的工具有Paramiko、Fabric、以及通过subprocess模块调用SSH命令。这三个工具各有优缺点：Paramiko是一个纯Python实现的SSH协议的库，适用于大多数SSH相关任务；Fabric则是一个基于Paramiko的更高级抽象库，适合自动化任务和远程服务器管理；而subprocess模块则提供了一种简单但不太灵活的方式来调用系统的SSH命令。本文将详细介绍这几种方法及其使用场景。

Paramiko是一个非常流行的Python SSH模块，它提供了一个高层次的API，使得SSH连接和SFTP操作变得非常简单。通过Paramiko，我们可以轻松地创建SSH客户端，执行远程命令，甚至可以进行文件传输。Paramiko的优势在于其灵活性和易用性，它是一个纯Python实现的库，不依赖于操作系统的SSH客户端，因此具有很好的跨平台性。

二、PARAMIKO的使用

Paramiko是一个纯Python实现的SSH协议的库，适用于大多数SSH相关任务。

1、安装和基本使用

首先，我们需要安装Paramiko库。可以通过pip命令来安装：

pip install paramiko

安装完成后，我们可以开始使用Paramiko来进行SSH登录和命令执行。以下是一个简单的示例：

import paramiko
创建SSH客户端对象
ssh = paramiko.SSHClient()
自动添加主机到known_hosts
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
连接到服务器
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())
关闭连接
ssh.close()

2、使用密钥认证

在某些情况下，我们可能需要使用SSH密钥进行认证，而不是使用密码。以下是使用密钥认证的示例：

import paramiko
创建SSH客户端对象
ssh = paramiko.SSHClient()
自动添加主机到known_hosts
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
使用密钥连接到服务器
key = paramiko.RSAKey.from_private_key_file('/path/to/private/key')
ssh.connect(hostname='your-server-ip', port=22, username='your-username', pkey=key)
执行命令
stdin, stdout, stderr = ssh.exec_command('ls -l')
打印输出
print(stdout.read().decode())
关闭连接
ssh.close()

3、文件传输

Paramiko还支持SFTP操作，我们可以使用SFTPClient进行文件上传和下载：

import paramiko
创建SSH客户端对象
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
使用密钥连接到服务器
key = paramiko.RSAKey.from_private_key_file('/path/to/private/key')
ssh.connect(hostname='your-server-ip', port=22, username='your-username', pkey=key)
创建SFTP客户端对象
sftp = ssh.open_sftp()
上传文件
sftp.put('/local/path/to/file', '/remote/path/to/file')
下载文件
sftp.get('/remote/path/to/file', '/local/path/to/file')
关闭SFTP和SSH连接
sftp.close()
ssh.close()

三、FABRIC的使用

Fabric是一个基于Paramiko的更高级抽象库，适合自动化任务和远程服务器管理。

1、安装和基本使用

首先，我们需要安装Fabric库。可以通过pip命令来安装：

pip install fabric

安装完成后，我们可以使用Fabric来执行远程任务。以下是一个简单的示例：

from fabric import Connection
创建连接对象
conn = Connection(
    host='your-server-ip',
    user='your-username',
    connect_kwargs={
        'password': 'your-password',
    },
)
执行命令
result = conn.run('ls -l', hide=True)
打印输出
print(result.stdout)
关闭连接
conn.close()

2、使用密钥认证

Fabric也支持使用密钥进行认证：

from fabric import Connection
创建连接对象
conn = Connection(
    host='your-server-ip',
    user='your-username',
    connect_kwargs={
        'key_filename': '/path/to/private/key',
    },
)
执行命令
result = conn.run('ls -l', hide=True)
打印输出
print(result.stdout)
关闭连接
conn.close()

3、自动化任务

Fabric特别适合用于自动化任务和批量操作。我们可以将多个操作编写成一个脚本，然后通过Fabric一次性执行：

from fabric import task
@task
def deploy(c):
    # 上传文件
    c.put('/local/path/to/file', '/remote/path/to/file')
    # 执行远程命令
    c.run('systemctl restart myservice')
使用fabric命令行工具运行此脚本：fab -H your-server-ip deploy

四、使用SUBPROCESS模块

通过subprocess模块调用SSH命令提供了一种简单但不太灵活的方式来进行SSH操作。

1、基本使用

subprocess模块可以用来调用系统命令，包括SSH命令。以下是一个简单的示例：

import subprocess
使用subprocess调用ssh命令
result = subprocess.run(
    ['ssh', 'your-username@your-server-ip', 'ls -l'],
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE,
    text=True,
)
打印输出
print(result.stdout)

2、使用密钥认证

我们也可以通过subprocess使用密钥进行认证：

import subprocess
使用subprocess调用ssh命令，指定密钥文件
result = subprocess.run(
    ['ssh', '-i', '/path/to/private/key', 'your-username@your-server-ip', 'ls -l'],
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE,
    text=True,
)
打印输出
print(result.stdout)

3、注意事项

使用subprocess模块时，需要注意以下几点：

命令的参数和选项需要以列表的形式传递给subprocess.run。
在某些情况下，可能需要处理命令的标准输入输出和错误输出。
由于subprocess是直接调用系统命令，因此需要确保系统环境中安装并配置了SSH客户端。

五、总结

在Python中使用SSH进行远程连接和操作有多种方法可供选择。Paramiko提供了一个灵活且功能强大的SSH协议实现，适合大多数SSH相关任务；Fabric在Paramiko的基础上提供了更高级的抽象，适合自动化任务和批量操作；而subprocess模块则提供了一种简单但不太灵活的方式来调用系统的SSH命令。根据具体的需求和使用场景，可以选择最适合的方法来实现SSH操作。对于需要跨平台和复杂操作的任务，Paramiko和Fabric是推荐的选择；而对于简单任务或需要直接调用系统命令的场景，subprocess模块则是一个不错的选择。