Python脚本远程批量执行命令

摘要
本文主要写用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

解释：