python批量更新服务器上的文件

因为公司业务经常需要升级全国服务器的一些脚本,每次都要手工来上传文件,然后再执行,非常烦琐。所以写了一个小脚本来批量执行，顺便炼炼手。

程序需求：
1、可以指定升级哪个模块，从配置文件中调出哪些省有此模块
2、指定升级web还是升级数据库

配置文件areamap.ini

[江苏电信]
abc = 172.16.0.1
def = 172.16.0.2


[江西电信]
abc = 172.16.2.1
def = 172.16.2.2

import os,sys
import ConfigParser

class UpDate:
	def __init__(self,ModName):
		self.modname = ModName
		self.conf = "areamap.ini"

# 到到这个模块所有服务器的地区和IP地址
	def getmodip(self):
		conf=ConfigParser.ConfigParser()
		conf.read(self.conf)
		IP=[]
		for section in conf.sections():
			if self.modname in conf.options(section):
				IP.append((section,conf.get(section,self.modname)))
		return IP
# 取得模块与程序包的对应关系	
	def getwebfile(self):
		ModMap={'abc':'abc.war','def':'def.war',}
		return ModMap[self.modname]		

# 重启tomcat	
	def retomcat(self,IP):
		print 'Now restart tomcat:'
		cmd = "ssh " + IP + " -C 'killall -9 java;sleep 10;tomcat/bin/startup.sh'"
		print cmd
		os.system(cmd)	
	
if __name__ == "__main__":
	if(len(sys.argv) < 3):
		print 'Usage:./up.py modelname db/web sqlfilename'
		sys.exit()		
	elif( sys.argv[2] == 'db' and len(sys.argv) != 4):
		print 'Usage:./up.py modelname db sqlfilename'
		sys.exit()
		

	ModelName = sys.argv[1]	
	update = UpDate(ModName=ModelName)
	IP = update.getmodip()
	
	if(sys.argv[2] == 'web'):		
		for ip in IP:
			File = update.getwebfile()
			if(not os.path.isfile(File)):
				print "The File: %s is not exsit" % File
				sys.exit()			
			print "\n************"+ip[0]+"************\n"
			update.copyfile(File,ip[1],'tomcat/webapps/')
			update.retomcat(ip[1])	

评论

2 楼 ycsk 2008-04-14  

因为war包是直接拷到对应目录下的,经常发现tomcat缓存导致更新后的war包,页面访问未更新,修改了程序:

	def retomcat(self,IP):
		print 'Now restart tomcat:'
		cmd = "ssh " + IP + " -C tomcat/bin/startup.sh'"
		print cmd
		os.system(cmd)
	
	def rmtomcat(self,IP):
		print 'Now remove tomcat webapps:'
		cmd = "ssh " + IP + " -C 'killall -9 java;sleep 10;rm -rf tomcat/webapps/" 
+ self.modname +"'"
		print cmd
		os.system(cmd)

1 楼 ycsk 2008-04-14  

今天在使用此脚本升级发现,在areamap.ini里添加的一个选项怎么也获得不了它的IP.最终发现原来

		for section in conf.sections():
			if self.modname in conf.options(section):
				IP.append((section,conf.get(section,self.modname)))

中conf.options()获得的选项名称都是小写,而我在areamap中是大写的.后来把此段代码修改为:

		for section in conf.sections():
			if self.modname.lower() in conf.options(section):
				IP.append((section,conf.get(section,self.modname)))

最终测试成功