摘要

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

这个可以说是Ansible,Puppet等工具的最简单的雏形。

做运维的同学应该都知道的。


正文

multi_task.py

#_*_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):
        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 :

 

                  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

 

      # 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','
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
,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

[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

from multi_task import MultiTask, run_task
tasks = MultiTask()


运行方法:

python main.py


解释: