Python脚本远程批量执行命令
摘要本文主要写用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_*_
importmultiprocessing
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 =
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:
p = pool.apply_async(run_task, args=(host,msg,lock))
res_list.append(p)
pool.close()
pool.join()
print '--------All task are finished!-------'
def run_task(host,msg,lock):
ip = host
username = msg
password = msg
cmd = msg
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:%sIP:%s -------------' %(username,ip)
for line in cmd_result:
print line,
lock.release()
s.close()
except Exception,e:
print '----------- HOST:%sIP:%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
node1=10.9.214.10
node2=10.9.214.105
node3=10.9.214.106
node4=10.9.214.113
masternode=10.9.214.15
main.py
1
2
from multi_task import MultiTask, run_task
tasks = MultiTask()
运行方法：
python main.py
解释：
windows能运行么？我在windows下运行了之后，啥反应都没有 好人啊。。。。。。。。。。。。。。。。。。。。。
页:
[1]