# coding: utf-8 import paramiko import re import os from time import sleep # 定义一个类,表示一台远端linux主机 # 参考https://www.cnblogs.com/haigege/p/5517422.html wyc class Linux(object): # 通过IP, 用户名,密码,超时时间初始化一个远程Linux主机 def __init__(self, ip, username, password, timeout=3000): self.ip = ip self.username = username self.password = password self.timeout = timeout # transport和chanel self.t = '' self.chan = '' # 链接失败的重试次数 self.try_times = 3 # 调用该方法连接远程主机 def connect(self): while True: # 连接过程中可能会抛出异常,比如网络不通、链接超时 try: self.t = paramiko.Transport(sock=(self.ip, 22)) self.t.connect(username=self.username, password=self.password) self.chan = self.t.open_session() self.chan.settimeout(self.timeout) self.chan.get_pty() self.chan.invoke_shell() # 如果没有抛出异常说明连接成功,直接返回 print (u'连接%s成功' % self.ip) # 接收到的网络数据解码为str print (self.chan.recv(65535).decode('utf-8')) return # 这里不对可能的异常如socket.error, socket.timeout细化,直接一网打尽 except Exception as e1: if self.try_times != 0: print (u'连接%s失败,进行重试' %self.ip) self.try_times -= 1 else: print (u'重试3次失败,结束程序') exit(1) # 断开连接 def close(self): self.chan.close() self.t.close() # 发送要执行的命令 def send(self, cmd): cmd += ' ' # 通过命令执行提示符来判断命令是否执行完成 p = re.compile(r':~ #') result = '' # 发送要执行的命令 self.chan.send(cmd) # 回显很长的命令可能执行较久,通过循环分批次取回回显 while True: sleep(0.5) ret = self.chan.recv(65535) ret = ret.decode('utf-8') result += ret if p.search(ret): print (result) return result # ------获取本地指定目录及其子目录下的所有文件------ def __get_all_files_in_local_dir(self, local_dir): # 保存所有文件的列表 all_files = list() # 获取当前指定目录下的所有目录及文件,包含属性值 files = os.listdir(local_dir) for x in files: # local_dir目录中每一个文件或目录的完整路径 filename = os.path.join(local_dir, x) # 如果是目录,则递归处理该目录 if os.path.isdir(x): all_files.extend(self.__get_all_files_in_local_dir(filename)) else: all_files.append(filename) return all_files def sftp_put_dir(self, local_dir, remote_dir): t = paramiko.Transport(sock=(self.ip, 22)) t.connect(username=self.username, password=self.password) sftp = paramiko.SFTPClient.from_transport(t) # 去掉路径字符穿最后的字符'/',如果有的话 if remote_dir[-1] == '/': remote_dir = remote_dir[0:-1] # 获取本地指定目录及其子目录下的所有文件 all_files = self.__get_all_files_in_local_dir(local_dir) # 依次put每一个文件 for x in all_files: filename = os.path.split(x)[-1] remote_filename = remote_dir + '/' + filename print (x) print (remote_filename) print (u'Put文件%s传输到%s中...' % (filename,self.ip)) sftp.put(x, remote_filename) if __name__ == '__main__': # host = Linux('172.16.10.40', 'root', 'imlytek!40') # host.connect() # host.send('ls -l') remote_path = r'/imlytek/smpt' local_path = r'D:TSTP' # host.sftp_put_dir(local_path, remote_path) hostArray=[['172.16.10.40','root','iflstek!40'],['172.16.10.51','root','x1f@2013']] for x in hostArray: host = Linux(x[0], x[1], x[2]) host.sftp_put_dir(local_path, remote_path) #host.close()