#!/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()
