用python重写双重提交svn的脚本

之前的shell脚本是 [url=http://rikugun.iteye.com/blog/630203]2个svn root 同步提交[/url]
update:
2010-06-21: 导入文件支持#行注释

现在改成python写,妄图支持windows,未测试.
osx,python2.6下测试通过
目前问题:
optparse的选项默认值不能是unicode,大囧rz

#!/usr/bin/env python
#coding:utf-8
"""
Author:	rikugun
Site: http://rikugun.iteye.com/

同时提交北京和本地SVN的脚本
本地环境需要设置环境变量:
	如
	本地开发编译目录 LOCAL_CRM=/Users/rikugun/Documents/tydic/Projects/LOCAL
	本机实际开发目录 CRM=/Users/rikugun/Documents/tydic/Projects/CRM	
执行
dsvn CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc [...]
后脚本做了以下几件事
 * SVN提交 $CRM/CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc
 * 拷贝/合并(使用合并工具) $CRM/CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc $LOCAL_CRM/CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc
 * 如果$LOCAL_CRM/CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc 不存在SVN中,执行 svn add
 * SVN提交 $LOCAL_CRM/CRM_TRANS/src/ngbssIf/InterfaceElec/CheckInfoOrder.pc
 * TELNET 到开发服务器, 切换到对应目录 执行 svn up  然后编译

"""
import optparse
import sys
import os
import subprocess
import telnetlib
import re

"""需要编译的文件类型"""
SOURCE_FILE_EXT =('.pc','.h','.c')

class TexudoBackend(object):
	"""
	Texudo后台
	可以设置 _DEBUG_LEVEL 开启telnet 调试
	"""

	_DEBUG_LEVEL=0
	def __init__(self, address,username,password=None,quite=False):
		super(TexudoBackend, self).__init__()
		self.address = address
		self.username = username
		self.password = password
		self.instance = telnetlib.Telnet(host=self.address,timeout=60)
		self.instance.set_debuglevel(self._DEBUG_LEVEL)
		self.quite = quite
	def login(self):
		"""登录"""
		if self.password:
			self.instance.read_until("login: ")
			self.instance.write(self.username+'\n')
			self.instance.read_until("ssword: ")
			self.instance.write(self.password+'\n')
			self.instance.read_until('$ ')
		return True	

	def exe_cmd(self,cmd):
		"""执行命令"""
		# print cmd
		self.instance.write(cmd+'\n')
		ret = self.instance.read_until('$ ').decode('GBK')
		if not self.quite:
			print ret
		return ret	
	def close(self):
		"""关闭链接"""
		self.instance.close()	

def getInput(msg):
	"""获取用户输入,默认是NO"""
	return opts.quite or raw_input(msg) not in ['n','no']

def sysexec(cmd):
	"""执行命令幷返回结果"""
	msg = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE).communicate()[0].decode('utf-8')
	if not opts.quite:
		print u'================执行命令==================='
		print cmd
		print u'================我是分割线==================='
		print msg
		print u'================已执行命令==================='
	return msg	


SHARE_DIR_NAME = 'share'		

def sort_dirs(dir1,dir2):
	"""排序目录,优先编译 share 目录,再编译目录名短的"""
	SHARE_DIR_NAME = 'share'
	if dir1.endswith(SHARE_DIR_NAME) and not dir2.endswith(SHARE_DIR_NAME):
		return -1
	elif dir2.endswith(SHARE_DIR_NAME) and not dir1.endswith(SHARE_DIR_NAME):
		return 1
	elif dir1.endswith(SHARE_DIR_NAME) and  dir2.endswith(SHARE_DIR_NAME): 	
		return dir1.find(SHARE_DIR_NAME) - dir2.find(SHARE_DIR_NAME)
	else:
		return len(dir1)-len(dir2)

if __name__ == "__main__":
	opt = optparse.OptionParser("dsvn [-m comments] [-u usernaem] [-p password] [-s host] [-L listfile] filepath [other files]")
	opt.add_option("-m","--comments",help=u"提交的注释",default="commit to compile")
	opt.add_option("-L","--listfile",help=u"提交的文件列表")
	opt.add_option("-u","--username",help=u"登录用户名",default="xxx")
	opt.add_option("-p","--password",help=u"登录用户密码",default="xxx")
	opt.add_option("-s","--server",help=u"登录服务器",default="xxx.xxx.xxx.xxx")
	opt.add_option("-M","--mergetool",help=u"使用工具合并而非拷贝")
	opt.add_option("-q","--quite",action="store_true",help=u"安静模式,所有提示都选择Y")
	opts,files = opt.parse_args()
	if len(files) == 0 and opts.listfile is None:
		print u'请输入要提交文件,或者文件列表'
		print opt.print_usage()
		print __doc__
		exit(2)
	if not files and opts.listfile:
		files = [line for line in open(opts.listfile,'r').readlines() if not line.startswith('#')]	
	#获取环境变量
	# 开发环境
	dev_dir =os.getenv('CRM')
	# 本地编译目录
	local_dir = os.getenv('LOCAL_CRM')

	reg = re.compile(dev_dir+os.sep+'(.+)$')

	#设置合并工具
	if opts.mergetool is not None:
		merge_cmd = opts.mergetool
	elif os.name =='nt':
		merge_cmd = 'copy /Y'
	else:	
		merge_cmd='cp -r'	

	#过滤文件名 如 CRM_TRANS/src/share/EssErrorTrans.txt	
	fileset = set([reg.match(os.path.realpath(os.getcwd()+os.sep+it)).group(1).strip() for it in files])	
	file_list = ','.join(fileset)
	#过滤需要编译的目录
	dir_list = list(set([ os.path.dirname(x) for x in fileset if x.endswith(SOURCE_FILE_EXT)]))
	#排序文件目录
	dir_list.sort(sort_dirs)

	#开发路径
	src_file_list= [ dev_dir+os.sep+x  for x in fileset]
	#同步到本地的路径
	dst_file_list =[local_dir+os.sep+x for x in fileset]
	#过滤新加的文件,需要额外执行svn add命令
	add_dst_file_list = [x for x in dst_file_list if not os.path.exists(x)]

	svn_cmd = ['svn ci -m "%s"' % opts.comments,]
	if getInput('commit such files [%s] to BJ SVN ? (y)es/no :' % file_list) :
		sysexec(' '.join(svn_cmd+src_file_list))
	# 传输文件,本地提交
	if getInput('commit such files [%s] to LOCAL SVN ? (y)es/no :' % file_list) :
		sysexec(' '.join(['svn revert ']+dst_file_list))
		map(lambda ff:sysexec(' '.join([merge_cmd,dev_dir+os.sep+ff,local_dir+os.sep+ff])),fileset)
		if len(add_dst_file_list):
			sysexec(' '.join(['svn add']+dst_file_list))
		sysexec(' '.join(svn_cmd+dst_file_list))
	# 切换到服务器编译
	if getInput('Compile such files [%s] on server[%s] ? (y)es/no :' % (file_list,opts.server)) :
		tn = TexudoBackend(opts.server,opts.username,opts.password)
		try:
			if	tn.login() :
				print u'登录成功'
				#在服务器上更新文件
				tn.exe_cmd('svn up %s' % ' '.join(files))
				#逐个目录编译
				for dir_path in dir_list:
					tn.exe_cmd('cd ~/%s && make && make install ' % dir_path.replace('\\','/'))
		except Exception:	
			print u"传输到服务器中编译发生错误"
			tn.close()	
			exit(2)
	print u'脚本执行成功'