python ssh链接工具类

代码结构

# 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("完成！")