Python――远程管理脚本（二）

前言：Python初学者。

脚本实现目的：

    通过一台集中控制的服务器管理其他所有服务器。

    主要功能有：批量多进程远程执行命令、上传文件、获取文件

    服务器的相关信息（IP,username,port,auth_type,password）可以保存在数据库中，也可以写在一个文件中

    类似功能的脚本已经存在并一直在使用，但是由于是shell写的，所以弊端就是速度慢，容错能力差。

脚本思路：

    利用Python的multiprocessing模块（初步思路来源于Python入门老师Alex）

    服务器的相关信息默认保存在数据库中，也可以通过-f参数指定一个文件来获取。

脚本内容：

#!/usr/bin/env python
# conding=UTF-8
#This script is use for multiprocessing host manage
#Create by eleven in 2014-06-01
#Last changed in 2014-06-11

import multiprocessing
import MySQLdb
import sys,os,time

script = 'paramiko_test.py'
mhost="192.168.100.211"
muser="user1"
mpasswd="123"
mdb="test"

#connect mysql and get host information
def con_mysql(mhost,muser,mpasswd,mdb):
	try:
		con = MySQLdb.connect(host=mhost,user=muser,passwd=mpasswd,db=mdb)
		cur = con.cursor()
	except MySQLdb.Error,err_msg:
		print 'Mysql Error Msg:',err_msg
		sys.exit()

	try:	
		cur.execute('select * from host_list')
		info = cur.fetchall()
		return info
	except MySQLdb.Error,err_msg:
		print 'Mysql Error Msg:',err_msg
		cur.close()
		con.close()
		sys.exit()

#get info
all_argv = sys.argv[1:]
if  '-f' in all_argv:
	num=all_argv.index('-f')
	hlist=all_argv[num+1]
	if os.path.exists(hlist):
		f=file(hlist)
		info=[]
		for line2 in f.readlines():
			line2=line2.split()
			info.append(line2)
		f.close()
	else:
		print 'The %s is not exists !' %hlist
else:
	info=con_mysql(mhost,muser,mpasswd,mdb)

#format host information
action = sys.argv[1]
exec_list=[]
for line in info:
	line =  line[1:]
	host = line[0]
	user = line[1]
	atype = line[3]
	passwd = line[4]
	if action == 'cmd': 
		command = sys.argv[2]
		cmd = 'python %s cmd %s %s %s %s %s' %(script,host,user,atype,passwd,command)
	elif action == 'get':
		file1 = sys.argv[2]
		file2 = sys.argv[3]+'_'+host
		cmd = 'python %s get %s %s %s %s %s %s' %(script,host,user,atype,passwd,file1,file2)
	elif action == 'put':
		file1 = sys.argv[2]
		file2 = sys.argv[3]
		cmd = 'python %s put %s %s %s %s %s %s' %(script,host,user,atype,passwd,file1,file2)
	exec_list.append(cmd)
	
def run_cmd(t):
	os.system(t)

#multiprocessing defined
p=multiprocessing.Pool(processes=len(exec_list))
result=[]
for task in exec_list:
	result.append(p.apply_async(run_cmd,(task,)))
p.close()

#get result
for answer in result:
	answer.get(timeout=10)

说明：

    此脚本需要和paramiko的脚本配合一起使用，已经在生产环境中验证过，使用无问题。并打算慢慢替换掉之前shell写的类似功能的脚本。

若有同样感兴趣的朋友，可以留言，到时再详细介绍一下用法，这里就不赘述了。