Python脚本远程批量执行命令

23112312

摘要
本文主要写用python脚本远程连接多台服务器，然后批量执行命令，最终返回命令执行结果。

这个可以说是Ansible，Puppet等工具的最简单的雏形。
做运维的同学应该都知道的。

正文
multi_task.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

#_*_coding:utf-8_*_ import  multiprocessing import paramiko import getpass import ConfigParser class MultiTask(object):     '''handles all the multi task works'''     def __init__(self):         msg = self._interactive()         self.__run_cmd(msg)     def __show_cmd_list(self,msg):         '''show all available cmd list'''         help_content = '''         run_cmd     run cmd on multiple hosts                     run_cmd -u remote_user -g group1,group2 -cmd pwd                     run_cmd -u remote_user -re regular expression -cmd pwd         '''           def _interactive(self):

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

msg = []         nodes = []         #parse setting.conf         Config = ConfigParser.ConfigParser()         Config.read("./setting.conf")         groups = Config.sections()         print groups         # Input group name         while True:             try:                 group = raw_input("\033[33;1mPlease Input Group Name:\033[0m").strip()                 if len(group) == 0 : continue                 elif group not in groups:                     print "Wrong group name ! Please input again!"                     continue                 else :

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

print "You have choose group %s , the children in this group are :" % group                     break                          except (KeyboardInterrupt):                 print '\n'                 exit(1)                   items = dict(Config.items(group))         for node in items.values():             nodes.append(node)             print node                   # Input command         while True:             try:                 cmd = raw_input("\033[33;1mPlease Input Command:\033[0m").strip()                 if len(cmd) == 0 : continue                 else: break             except (KeyboardInterrupt):                 print '\n'                 exit(1)         print "Command you input is %s" % cmd

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

# Input username and password         while True:             try:                 username = raw_input("\033[33;1mPlease Input username:\033[0m").strip()                 if len(username) == 0 : continue                 else: break             except (KeyboardInterrupt):                 print '\n'                 exit(1)         password = getpass.getpass()                 msg = [nodes,username,password,cmd]         print msg         return msg     def __run_cmd(self,msg):         pool = multiprocessing.Pool(5)         lock = multiprocessing.Manager().Lock()         res_list = []         #msg = [['10.9.214.10','haohzhang','871102_Hadoop'],['10.9.214.105','haohzhang','

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20

871102_Hadoop'],['10.9.214.106','haohzhang','871102_Hadoop']]         for host in msg[0]:             p = pool.apply_async(run_task, args=(host,msg[1:],lock))             res_list.append(p)         pool.close()         pool.join()         print '--------All task are finished!-------' def run_task(host,msg,lock):           ip = host     username = msg[0]     password = msg[1]     cmd = msg[2]     print "ip %s, username %s, passwd %s, cmd %s" % (ip, username, password, cmd)     port = 22     s = paramiko.SSHClient()    #绑定实例     s.load_system_host_keys()   #加载本机know host主机文件     s.set_missing_host_key_policy(paramiko.AutoAddPolicy())     try:         s.connect(ip,port

1 2 3 4 5 6 7 8 9 10 11 12 13

,username,password,timeout=5)   #连接远程主机         stdin,stdout,stderr = s.exec_command(cmd)   #执行命令         cmd_result = stdout.read(),stderr.read()    #读取命令结果         lock.acquire()         print '----------- HOST:%s  IP:%s -------------' %(username,ip)         for line in cmd_result:             print line,         lock.release()         s.close()     except Exception,e:         print '----------- HOST:%s  IP:%s -------------' %(username,ip)         print '\033[31;1mError:%s\033[0m' % e # vim:ts=4:sw=4:expandtab

setting.conf

1 2 3 4 5 6 7

[datanodes] node1=10.9.214.10 node2=10.9.214.105 node3=10.9.214.106 node4=10.9.214.113 [masternodes] masternode=10.9.214.15

main.py

1 2	from multi_task import MultiTask, run_task tasks = MultiTask()

运行方法：
python main.py

解释：

sunzhongyu

windows能运行么？我在windows下运行了之后，啥反应都没有

sdjxzys

好人啊。。。。。。。。。。。。。。。。。。。。。