如何实现一个Python SSH链接工具类？

本文共计319个文字，预计阅读时间需要2分钟。

代码结构如下：

plaintext模块/函数名：功能描述：输入参数：返回值：实现步骤：

1.初始化变量

2.执行特定操作

3.返回结果

代码结构

# coding:utf-8
import paramiko

'''文件上传与下载'''

class SSHConnection(object):

def __init__(self, host='ip', port=22, username='root', pwd=''):
self.host = host
self.port = port
self.username = username
self.pwd = pwd
self.__k = None
def connect(self):
transport = paramiko.Transport((self.host, self.port))
transport.connect(username=self.username, password=self.pwd)
self.__transport = transport

def close(self):
self.__transport.close()

def upload(self, local_path, target_path):
# 连接，上传 # file_name = self.create_file()
sftp = paramiko.SFTPClient.from_transport(self.__transport)
# 将location.py 上传至服务器 /tmp/test.py
sftp.put(local_path, target_path)

def download(self, remote_path, local_path):
sftp = paramiko.SFTPClient.from_transport(self.__transport)
sftp.get(remote_path, local_path)

def cmd(self, command):
ssh = paramiko.SSHClient()
ssh._transport = self.__transport
# 执行命令
stdin, stdout, stderr = ssh.exec_command(command)
# 获取命令结果
result = stdout.read()
print(str(result, encoding='utf-8'))
return result


if __name__ == '__main__':
ssh = SSHConnection()
ssh.connect()
# ssh.cmd("ls")
# SHOW INDEX FROM ums_member;
# ALTER TABLE ums_member DROP INDEX phone;
ssh.upload('/Users','/home/script')
# ssh.upload('//saveLogin.py','/home/saveLogin.py')
# path = '/root/data/categoryInfo.txt'
# ssh.download(path, '/')
ssh.close()
print("完成！")