paramiko封装

from paramiko.client import SSHClient, AutoAddPolicy


class SSH:
    def __init__(self, hostname, username="root", port=22, password=None, pkey=None, connect_timeout=10):
        self.client = None
        self.connect_dist = {
            "hostname": hostname,
            "username": username,
            "port": port,
            "pkey": pkey,
            #"pkey": paramiko.RSAKey.from_private_key_file(pkey),
            "password": password,
            "timeout": connect_timeout

        }

    def get_connet(self):
        if self.client is None:
            try:
                self.client = SSHClient()
                self.client.set_missing_host_key_policy(AutoAddPolicy)  # 指纹记录
                self.client.connect(**self.connect_dist)
            except Exception as e:
                return None
        return self.client

    def exec_command(self, command, timeout=10):
        with self as cli:  # __enter__方法,并将该方法的返回值给 as 后面的变量
            if cli:  # cli -- self.client
                ssh2 = cli.get_transport().open_session()
                ssh2.settimeout(timeout)
                ssh2.set_combine_stderr(True)  # 正确和错误输出都在一个管道里面输出出来
                ssh2.exec_command(command)
                stdout = ssh2.makefile("rb", -1)
                return stdout.read().decode('GBK')
            return self.connect_dist["hostname"] + "连接失败"

    # with self: 先执行__enter__方法
    def __enter__(self):
        if self.client is None:
            return self.get_connet()

    # with self:语句体内容结束后执行如下方法 先执行__enter__方法
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.client.close()
        self.client = None


if __name__ == '__main__':
    zbb = SSH("121.193", "root", password="zbb")
    aa = zbb.exec_command("pwd")

    print(aa)

posted @   追梦nan  阅读(146)  评论(0编辑  收藏  举报
编辑推荐:
· 一个奇形怪状的面试题:Bean中的CHM要不要加volatile?
· [.NET]调用本地 Deepseek 模型
· 一个费力不讨好的项目,让我损失了近一半的绩效!
· .NET Core 托管堆内存泄露/CPU异常的常见思路
· PostgreSQL 和 SQL Server 在统计信息维护中的关键差异
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 我与微信审核的“相爱相杀”看个人小程序副业
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· DeepSeek R1 简明指南:架构、训练、本地部署及硬件要求
点击右上角即可分享
微信分享提示