基于paramiko的一个类库

最新推荐文章于 2025-07-05 11:46:14 发布

雨中山果落

最新推荐文章于 2025-07-05 11:46:14 发布

阅读量488

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python paramiko

本文链接：https://blog.youkuaiyun.com/longzbc/article/details/76735657

python 专栏收录该内容

1 篇文章

订阅专栏

这个博客介绍了如何使用Python的paramiko库建立SSH连接。通过定义一个Connect类，实现了打开连接、关闭连接以及发送命令的功能。在打开连接时，会检查登录是否成功。send_cmd方法用于发送命令并接收响应，而send_cmd_expect则在发送命令后判断响应是否包含预期的字符串。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding:utf-8 -*-

import paramiko
import time
import sys


class Connect:
    def __init__(self, host, user, pwd, port=22):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.port = port
        self.recv_buffer = 65535
        self.EOF_list = []
        self.is_connected = False

    def open_connect(self, timeout=10):
        try:
            self.ssh_sock = paramiko.Transport(self.host, self.port)
            self.ssh_sock.connect(username=self.user, password=self.pwd)
            self.handle = self.ssh_sock.open_session()
            self.handle.settimeout(timeout)
            self.handle.get_pty()
            self.handle.invoke_shell()
            self.is_connected = True
            time.sleep(5)
            if self.handle.recv_ready():
                rsp = self.handle.recv(self.recv_buffer)
                if 'login' in rsp:
                    print('Login succeed!')
                    sys.stdout.write(rsp)
                self.EOF_list.append(rsp.split('\n')[-1].split()[0])
        except Exception, e:
            print('ssh Open Error: %s' % e)

    def close_connect(self):
        if self.is_connected:
            try:
                self.handle.close()
                self.ssh_sock.close()
                self.is_connected = False
            except Exception, e:
                self.is_connected = False
                print('ssh Close Error: %s' % e)

    def send_cmd(self, cmd, timeout=10):
        if self.is_connected:
            try:
                startTime = time.time()
                self.handle.settimeout(timeout)
                self.handle.send('%s\r' % cmd)
                rsp = ''
                brk_flag = False
                add_flag = False
                while time.time() < startTime + timeout:
                    time.sleep(0.5)
                    if self.handle.recv_ready():
                        rsp = rsp + self.handle.recv(self.recv_buffer)
                        for eof in self.EOF_list:
                            if rsp.rfind(eof) > rsp.rfind('\n'):
                                brk_flag = True
                                add_flag = False
                                break
                        else:
                            add_flag = True
                    if brk_flag:
                        break
                if add_flag:
                    lines = rsp.split('\n')
                    if lines:
                        last_line = lines[-1].split()
                        if last_line:
                            if last_line[0]:
                                self.EOF_list.append(last_line[0])
                sys.stdout.write(rsp)
                return rsp
            except Exception, e:
                print('ssh Cmd Error: %s' % e)

    def send_cmd_expect(self, cmd, expect, timeout=10):
        if self.is_connected:
            rsp = self.send_cmd(cmd, timeout)
            if rsp:
                if str(expect) in rsp:
                    return True
                else:
                    return False