如何使用Python连接 SSH 服务器并执行命令

更新时间：2023年11月02日 15:04:46 作者：lemeifei

实际开发中,有时候经常需要查看日志,有时候使用ssh工具打开就为了看一下错误日志又比较麻烦,所以今天带来一个简单的基于python的小工具,感兴趣的朋友跟随小编一起看看吧

实际开发中,有时候经常需要查看日志,有时候使用ssh工具打开就为了看一下错误日志又比较麻烦,所以今天带来一个简单的基于python的小工具.

首先需要先安装一个库 paramiko

使用命令直接安装

pip install paramiko

paramiko库是一个开源的、基于SSH2协议的库，可以实现SSH连接以及数据的传输。

paramiko是用纯Python实现的SSH2客户端，支持身份验证、SFTP客户端以及远程执行命令等功能。paramiko库提供了丰富的类和方法，帮助用户快速实现SSH通信功能。

在实际应用中，paramiko库常用于构建自动化运维系统、远程部署、多机协作等场景.

定义MySshClient 类

class MySshClient:
    def __init__(self, ssh_client):
        self.ssh_client = ssh_client
    def exec_command(self, cmd):
        try:
            stdin, stdout, stderr = self.ssh_client.exec_command(cmd)
            return stdin, stdout, stderr
        except Exception as e:
            print(f"Error executing command {cmd}: {e}")
    def __enter__(self):
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.ssh_client.close()
在此代码中，我们定义了一个 MySshClient 类，用于连接 SSH 服务器并执行命令。您可以使用该类创建一个实例，并通过 connect() 方法连接到 SSH 服务器。
def connect(host, port, username, password):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.WarningPolicy())
    try:
        ssh.load_system_host_keys()
        ssh.connect(host, port, username, password,timeout=3)
    except paramiko.AuthenticationException:
        raise Exception(f"在主机 {host}连接失败,请检查你的参数")
    except paramiko.SSHException as e:
        raise Exception(f"在 {host}连接出错: {e}")
    except paramiko.BadHostKeyException as e:
        raise Exception(f" {host} 无法验证通过: {e}")
    except Exception as e:
        raise Exception(f" 连接到{host}:{port}: {e}超时")
    return ssh

这个是给函数加上异常处理,方便代码的调试

执行命令

在连接到 SSH 服务器后，您可以使用 exec_command() 方法执行命令。该方法返回一个元组，包含服务器的输出、错误和输入。您可以将此元组提供给 with 关键字，以便在需要时进行处理。

2.连接到主机

在连接到 SSH 服务器之前，您需要指定主机名、端口、用户名和密码。例如：

host = input("Host: 请输入主机名")
port = int(input("Port: 默认为22")or 22) 
username = input("Username: 请输入用户名")
password = getpass.getpass("Password: 请输入密码")

port = int(input(“Port: 默认为22”)or 22) 这段代码的意思是

默认端口22,如果你的ssh端口不是22,可以自行修改

密码的话,如果不想被别人看见,使用getpass函数进行加密

创建并连接到 MySshClient 实例

ssh = connect(host, port, username, password)

连接成功后，执行命令并处理输出

连接成功后，您可以使用 exec_command() 方法执行命令，并使用 with 关键字处理服务器的输出：

with MySshClient(ssh) as client:
    stdin, stdout, stderr = client.exec_command("ls -l")
    with stdout:
        print(stdout.read().decode())
    with stderr:
        print(stderr.read().decode())

这将在连接到 SSH 服务器后，执行 “ls -l” 命令，并输出结果。请注意，此示例仅显示输出，如果有错误，错误信息将显示在控制台。

完整代码如下:

import paramiko
import os
import getpass
class MySshClient:
    def __init__(self, ssh_client):
        self.ssh_client = ssh_client
    def exec_command(self, cmd):
        try:
            stdin, stdout, stderr = self.ssh_client.exec_command(cmd)
            return stdin, stdout, stderr
        except Exception as e:
            print(f"Error executing command {cmd}: {e}")
    def __enter__(self):
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.ssh_client.close()
def connect(host, port, username, password):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.WarningPolicy())
    try:
        ssh.load_system_host_keys()
        ssh.connect(host, port, username, password,timeout=3)
    except paramiko.AuthenticationException:
        raise Exception(f"在主机 {host}连接失败,请检查你的参数")
    except paramiko.SSHException as e:
        raise Exception(f"Error connecting to {host}: {e}")
    except paramiko.BadHostKeyException as e:
        raise Exception(f"Host key for {host} could not be verified: {e}")
    except Exception as e:
        raise Exception(f" 连接到{host}:{port}: {e}超时")
    return ssh
if __name__ == '__main__':  
    host = input("Host: 请输入主机名")
    port = int(input("Port: 默认为22")or 22)
    username = input("Username: 请输入用户名")  
    password = getpass.getpass("Password: 请输入密码")
    print('连接中...........')
    ssh = connect(host, port, username, password,)
    with MySshClient(ssh) as client:
        stdin, stdout, stderr = client.exec_command("ls -l")
        with stdout:
            print(stdout.read().decode())
        with stderr:
            print(stderr.read().decode())

当然你也可以使用sftp进行传输

# 上传
current_path = os.getcwd()   #获取当前路径
print(current_path)
filename = '\\main.py'     		# 文件名
file = current_path + f'{filename}'			# 当前文件的路径
sftp = client.open_sftp() 						# 使用sftp连接
sftp.put(file, '/root/main.py')			#  上传文件
sftp.get('/root/server.sh', current_path+'/code.sh')     #下载文件

到此这篇关于使用 Python 连接 SSH 服务器并执行命令的文章就介绍到这了,更多相关Python 连接 SSH 服务器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

python 根据网易云歌曲的ID 直接下载歌曲的实例
今天小编就为大家分享一篇python 根据网易云歌曲的ID 直接下载歌曲的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-08-08
Python中异常处理及最佳实践举例详解
异常处理在任何一门编程语言里都是值得关注的一个话题,良好的异常处理可以让你的程序更加健壮,下面这篇文章主要给大家介绍了关于Python中异常处理及最佳实践的相关资料,需要的朋友可以参考下
2024-03-03
Python格式化处理JSON数据的完整指南
在Python中,我们经常需要处理JSON数据,而格式化JSON数据是开发过程中的常见需求,本文将详细介绍如何在Python中对JSON数据进行格式化处理,感兴趣的小伙伴可以跟随小编一起学习一下
2026-04-04
python列表字典排序的实现示例
在Python中,对列表字典进行排序是一项常见的任务,本文主要介绍了python列表字典排序的实现示例,具有一定的参考价值,感兴趣的可以了解一下
2023-09-09
Python+Ollama零成本打造一个私有AI助手
零 API 费用、零数据泄露风险、完全离线可用,本文将带大家从安装到实战,详细介绍一下如何使用Python结合Ollama模型30 分钟跑起一个本地 AI 助手,感兴趣的小伙伴可以了解下
2026-04-04
使用Python轻松实现添加与删除Excel工作表的实战指南
这篇文章主要介绍了如何使用Python中的openpyxl库批量处理Excel文件的方法,主要包括添加和删除Excel工作表,感兴趣的小伙伴可以了解一下
2026-04-04
Python深度学习理解pytorch神经网络批量归一化
这篇文章主要是Python深度学习篇，通过示例的详解让大家更好的理解pytorch神经网络批量归一化，有需要的的朋友可以借鉴参考下，希望能够有所帮助
2021-10-10
PyTorch和Keras计算模型参数的例子
今天小编就为大家分享一篇PyTorch和Keras计算模型参数的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
Anaconda安装以及修改环境默认位置图文教程
Anaconda是一个用于科学计算的Python发行版,支持Linux,Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包,下面这篇文章主要给大家介绍了关于Anaconda安装以及修改环境默认位置的相关资料,需要的朋友可以参考下
2023-04-04
Playwright设置base_url的三种方式
本文主要介绍了三种在使用Playwright或pytest-playwright进行Web自动化测试时设置base_url的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
2026-01-01

如何使用Python连接 SSH 服务器并执行命令

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具