1 """批量主机管理接口"""
2
3 import core
4
5 if __name__ == "__main__":
6 core.run()
View Code
1 """核心代码""" 2 import settings 3 import paramiko 4 import threading 5 import os 6 7 8 class REMOTE_HOST(object): 9 #远程操作主机 10 def __init__(self, host, port ,username, password, cmd): 11 self.host = host 12 self.port = port 13 self.username = username 14 self.password = password 15 self.cmd = cmd 16 17 def run(self): 18 """起线程连接远程主机后调用""" 19 cmd_str = self.cmd.split()[0] 20 if hasattr(self, cmd_str): #反射 eg:调用put方法 21 getattr(self, cmd_str)() 22 else: 23 #setattr(x,'y',v)is equivalent to ``x.y=v'' 24 setattr(self, cmd_str, self.command) 25 getattr(self, cmd_str)() #调用command方法,执行批量命令处理 26 27 def command(self): 28 """批量命令处理""" 29 ssh = paramiko.SSHClient() #创建ssh对象 30 #允许连接不在know_hosts文件中的主机 31 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 32 ssh.connect(hostname=self.host,port=self.port,username=self.username,password=self.password) 33 stdin,stdout,stderr = ssh.exec_command(self.cmd) 34 result = stdout.read() 35 print("%s".center(50, "-") % self.host) 36 print(result.decode()) 37 ssh.close() 38 39 def put(self): 40 """上传文件""" 41 filename = self.cmd.split()[1] #要上传的文件 42 transport = paramiko.Transport((self.host, self.port)) 43 transport.connect(username=self.username, password=self.password) 44 sftp = paramiko.SFTPClient.from_transport(transport) 45 sftp.put(filename, filename) 46 print("put sucesss") 47 48 transport.close() 49 50 51 def show_host_list(): 52 """通过选择分组显示主机名与IP""" 53 for index, key in enumerate(settings.msg_dic): 54 print(index + 1, key, len(settings.msg_dic[key])) 55 while True: 56 choose_host_list = input(">>>(eg:group1)").strip() 57 host_dic = settings.msg_dic.get(choose_host_list) 58 if host_dic: 59 #print(host_dic) 60 for key in host_dic: 61 print(key, host_dic[key]["IP"]) 62 return host_dic 63 else: 64 print("NO exit this group!") 65 66 67 def interactive(choose_host_list): 68 """根据选择的分组主机起多个线程进行批量交互""" 69 thread_list = [] 70 while True: 71 cmd = input(">>>").strip() 72 if cmd: 73 for key in choose_host_list: 74 host, port, username, password = choose_host_list[key]["IP"], choose_host_list[key]["port"], \ 75 choose_host_list[key]["username"], choose_host_list[key]["password"] 76 func = REMOTE_H