Python의 간단한 호스트 배치 관리 도구에 대한 자세한 설명

"""批量主机管理接口""" import core if __name__ == "__main__": core.run()

"""核心代码""" import settings import paramiko import threading import os class REMOTE_HOST(object): #远程操作主机 def __init__(self, host, port ,username, password, cmd): self.host = host self.port = port self.username = username self.password = password self.cmd = cmd def run(self): """起线程连接远程主机后调用""" cmd_str = self.cmd.split()[0] if hasattr(self, cmd_str): #反射 eg:调用put方法 getattr(self, cmd_str)() else: #setattr(x,'y',v)is equivalent to ``x.y=v'' setattr(self, cmd_str, self.command) getattr(self, cmd_str)() #调用command方法，执行批量命令处理 def command(self): """批量命令处理""" ssh = paramiko.SSHClient() #创建ssh对象 #允许连接不在know_hosts文件中的主机 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect(hostname=self.host,port=self.port,username=self.username,password=self.password) stdin,stdout,stderr = ssh.exec_command(self.cmd) result = stdout.read() print("%s".center(50, "-") % self.host) print(result.decode()) ssh.close() def put(self): """上传文件""" filename = self.cmd.split()[1] #要上传的文件 transport = paramiko.Transport((self.host, self.port)) transport.connect(username=self.username, password=self.password) sftp = paramiko.SFTPClient.from_transport(transport) sftp.put(filename, filename) print("put sucesss") transport.close() def show_host_list(): """通过选择分组显示主机名与IP""" for index, key in enumerate(settings.msg_dic): print(index + 1, key, len(settings.msg_dic[key])) while True: choose_host_list = input(">>>(eg:group1)").strip() host_dic = settings.msg_dic.get(choose_host_list) if host_dic: #print(host_dic) for key in host_dic: print(key, host_dic[key]["IP"]) return host_dic else: print("NO exit this group!") def interactive(choose_host_list): """根据选择的分组主机起多个线程进行批量交互""" thread_list = [] while True: cmd = input(">>>").strip() if cmd: for key in choose_host_list: host, port, username, password = choose_host_list[key]["IP"], choose_host_list[key]["port"], \ choose_host_list[key]["username"], choose_host_list[key]["password"] func = REMOTE_HOST(host, port, username, password, cmd) # 实例化类 t = threading.Thread(target=func.run) # 起线程 t.start() thread_list.append(t) for t in thread_list: t.join() # 主线程等待子线程执行完毕 else: continue def run(): choose_host_list = show_host_list() interactive(choose_host_list)

"""配置文件""" msg_dic = { "group1":{ #分组1 "h1":{"IP":"192.168.1.1", "username":"11", "password":"aa", "port":22}, "h2":{"IP":"192.168.1.2", "username":"22", "password":"bb", "port":22}, "h3":{"IP":"192.168.1.3", "username":"33", "password":"cc", "port":22}, "h4":{"IP":"192.168.1.4", "username":"44", "password":"dd", "port":22}, "h5":{"IP":"192.168.1.5", "username":"55", "password":"ee", "port":22}, "h6":{"IP":"192.168.1.6", "username":"66", "password":"ff", "port":22}, }, "group2":{ #分组2 "h1":{"IP":"192.168.2.1", "username":"111", "password":"aaa", "port":22}, "h2":{"IP":"192.168.2.2", "username":"222", "password":"bbb", "port":22}, "h3":{"IP":"192.168.2.3", "username":"333", "password":"ccc", "port":22}, "h4":{"IP":"192.168.2.4", "username":"444", "password":"ddd", "port":22}, "h5":{"IP":"192.168.2.5", "username":"555", "password":"eee", "port":22}, "h6":{"IP":"192.168.2.6", "username":"666", "password":"fff", "port":22}, "h7":{"IP":"192.168.2.7", "username":"777", "password":"ggg", "port":22}, "h8":{"IP":"192.168.2.8", "username":"888", "password":"hhh", "port":22}, }, "group3":{ "h1":{"IP":"192.168.179.133", "username":"root", "password":"zcl", "port":22}, } }

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/host_manage/main.py 1 group1 6 2 group3 1 3 group2 8 >>>(eg:group1)group3 h1 192.168.179.133 >>>put from_windows.py put sucesss >>> >>>ls ------------------------192.168.179.133------------------------ anaconda-ks.cfg database_test from_windows.py install.log install.log.syslog m oot \root tmp\from_windows.py >>>