python实现通过SSH访问linux并执行指令

你的名字 2023-10-18 21:59 112阅读 0赞

#!/usr/bin/env python
    # -*- coding:utf-8 -*-
    import os
    import sys
    import paramiko
    from stat import S_ISDIR
    
    class SSHSession:
    
        def __init__(self, ip, name, PWD):
            print 'SSH init...'
            self.client = paramiko.SSHClient()
            # 自动接受远程服务器host key
            self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            self.client.connect(ip, username=name, password=PWD)
            self.sftp = paramiko.SFTPClient.from_transport(self.client.get_transport())
            self.sftp = self.client.open_sftp()
    
        def getRemoteFile(self, remotefile, localfile):
            self.sftp.get(remotefile, localfile)
    
        def get_all_files_in_remote_dir(self, remote_dir):
            all_files = list()
            if remote_dir[-1] == '/':
                remote_dir = remote_dir[0:-1]
            remoteFiles = self.sftp.listdir_attr(remote_dir)
            for remoteFile in remoteFiles:
                filename = remote_dir + '/' + remoteFile.filename
                if S_ISDIR(remoteFile.st_mode):
                    all_files.extend(self.get_all_files_in_remote_dir(filename))
                else:
                    all_files.append(filename)
    
            print "all_files", all_files
            return all_files
    
        def get_all_files_in_local_dir(self, local_dir):
            all_files = list()
            print local_dir
            print local_dir[-2]
            if local_dir[-1:] == "\\":
                local_dir = local_dir[0:-1]
            # local_dir = local_dir[0:-1]
            localFiles = os.listdir(local_dir)
            print "1 localFiles:", localFiles
            for localFile in localFiles:
                # filename = local_dir + '//' + localFile.filename
                filename = os.path.join(local_dir, localFile)
                print filename
                if os.path.isdir(filename):
                    #list param
                    all_files.extend(self.get_all_files_in_local_dir(filename))
                else:
                    #object
                    all_files.append(filename)
            print "2 all_files:", all_files
            return all_files
        def copyRemoteFolder(self, remote_dir, local_dir):
            all_files = self.get_all_files_in_remote_dir(remote_dir)
            for fullfilename in all_files:
                remoteFilePath, filename = os.path.split(str(fullfilename))
                local_path = os.path.join(local_dir, remoteFilePath.split('/')[-1])
                if not os.path.exists(local_path):
                    os.makedirs(local_path)
                local_filename = os.path.join(local_path, filename)
                print 'get files %s transporting...' % filename
                self.sftp.get(fullfilename, local_filename)
    
        def sendCommand(self, command):
            if self.client:
                channel = self.client.get_transport().open_session()
                channel.exec_command(command)
                return channel
            print 'Connection not opened.'
    
        def checkSSHconnection(self):
            if self.client:
                try:
                    self.client.exec_command('ls', timeout=0.5)
                    return True
                except Exception as e:
                    print 'Connection lost : %s' % e
                    return False
    
        def closeSSH(self):
            self.sftp.close()
            self.client.close()
    
        def __put_all_file(self, local_dir):
    #         save all files list
            all_files = list()
    
            files = os.listdir(local_dir)
            for i in files:
    #             
                filename = os.path.join(local_dir, i)
    #             
                if os.path.isdir(i):
                    all_files.extend(self.__put_all_file(filename))
                else:
                    all_files.append(filename)
    
            return all_files
    
        def put_dir(self, local_dir, remote_dir):
            # 去掉字符串最后的字符
            if remote_dir[-1] == '/':
                remote_dir = remote_dir[0:-1]
            # 获取本地指定的目录及目录下的文件
            all_files = self.get_all_files_in_local_dir(local_dir)
            for i in all_files:
                filename = os.path.split(i)[-1]
                print local_dir,remote_dir
                remote_file = os.path.split(i)[0].replace(local_dir, remote_dir)
                path = remote_file.replace("\\", "/")
                print 'filename:' + filename
                print remote_file
    
                self.client.exec_command('mkdir -p ' + path)
                remote_filename = path + '/' + filename
                print u'Put files...' + filename
                self.sftp.put(i, remote_filename)
    
    
                # remote_filename = remote_dir + '/' + filename
                # print(u'Put文件{}传输中...'.format(filename))
                # self.sftp.put(i, remote_filename)
    
    
    def main():
        SSH = SSHSession('192.168.56.101', 'root', 'suming')
        if SSH.checkSSHconnection():
            # SSH.copyRemoteFolder('/home/frank/log_test/', 'C:\\APP\\tool')
            SSH.put_dir('C:\\APP\\tool', '/home/frank/log_test/')
            # SSH.get_all_files_in_local_dir("C:\\APP\\tool\\")
        # for line in stdout:
        #     print(line)
        SSH.closeSSH()
    if __name__ == '__main__':
    
        main()