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)