├── CheungSSH使用手册.docx
├── 功能升级记录.txt
├── README.md
├── Format_Char_Show.py
├── LogCollect.py
├── install-CheungSSH.sh
└── cheungSSH_Manager_Stand_V96_ISSUE.py


/CheungSSH使用手册.docx:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/4021019/CheungSSH/HEAD/CheungSSH使用手册.docx


--------------------------------------------------------------------------------
/功能升级记录.txt:
--------------------------------------------------------------------------------
 1 | 版本升级记录：
 2 | V94:
 3 | 	1 升级了cd功能，可以使用cd切换目录
 4 | V95：
 5 | 	1 升级了使用方向键的功能
 6 | V96:
 7 | 	1 升级了原始日志记录
 8 | 
 9 | 
10 | 简单的用法请到我的博客：
11 | 	http://blog.chinaunix.net/uid-29295703-id-4663051.html
12 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # CheungSSH
2 | 比Ansibel更好用的自动化工具
3 | 这是我（张其川 Cheung Kei-Chuen） 自主开发的基于 SSH协议的自动操作工具，比Ansible更轻量！操作更简单！
4 | 如果您觉得本软件还不错，有需要的话，请与我联系，包括有任何的使用疑问
5 | 我的QQ   741345015
6 | 
7 | 详细的用法介绍请： http://blog.chinaunix.net/uid-29295703-id-4663051.html
8 | 如果有任何试用疑问，请联系我！！
9 | 


--------------------------------------------------------------------------------
/Format_Char_Show.py:
--------------------------------------------------------------------------------
 1 | #!/usr/bin/python
 2 | VERSION=1.3
 3 | import os,sys,commands
 4 | def Get_Char(Char,i):
 5 | 	Char=Char
 6 | 	New_Char=[]
 7 | 	for m in Char.split('\n'):
 8 | 		New_Char.append('|' + m )
 9 | 	b=Char.split('\n')
10 | 	Char_Len=[]
11 | 	for t in b:
12 | 		Char_Len.append(len(t))
13 | 	Max_Char_Len=sorted(Char_Len)[-1]
14 | 	return Max_Char_Len,New_Char
15 | 	
16 | def Show_Line(Max_Char_Len,i=0,Flag='end'):
17 | 	T=''
18 | 	while i<=Max_Char_Len+2:
19 | 		if i<=5 and Flag=='start':
20 | 			#sys.stdout.write('+')
21 | 			T+='+'
22 | 			i+=1
23 | 			continue
24 | 		
25 | 		#sys.stdout.write('-')
26 | 		T+='-'
27 | 		i+=1
28 | 	return T
29 | def Show_Char(New_Char,Color_Status):
30 | 	AllChar=''
31 | 	if Color_Status==0:
32 | 		Color_Start="\033[1;32m"
33 | 		Color_End="\033[0m"
34 | 	else:
35 | 		Color_Start="\033[1;31m"
36 | 		Color_End="\033[0m"
37 | 	New_Char=New_Char
38 | 	AllChar+='\n\n'
39 | 	Len_and_Char=Get_Char(New_Char,i=0)
40 | 	AllChar+=Color_Start
41 | 	AllChar+=Show_Line(Len_and_Char[0],i=0,Flag='start') + '\n'
42 | 	for t in Len_and_Char[1]:
43 | 		AllChar=AllChar + t + '\n'
44 | 	AllChar+=Show_Line(Len_and_Char[0],i=0) +  Color_End +'\n'
45 | 	#print AllChar
46 | 	return AllChar
47 | 
48 | if __name__=='__main__':
49 | 	Show_Char(sys.argv[1],0)
50 | 
51 | 


--------------------------------------------------------------------------------
/LogCollect.py:
--------------------------------------------------------------------------------
 1 | #!/usr/bin/python
 2 | #coding:utf-8
 3 | import os,sys,random,paramiko,time
 4 | def LogCollect(ip,port,username,password,cmd,ListenTime,UseKey):
 5 | 	try:
 6 | 		outchar=''
 7 | 		errchar=''
 8 | 		ssh=paramiko.SSHClient()
 9 | 		if UseKey=='y':
10 | 			KeyPath=os.path.expanduser('~/.ssh/id_rsa')
11 | 			key=paramiko.RSAKey.from_private_key_file(KeyPath)
12 | 			ssh.load_system_host_keys()
13 | 			ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) #使用这个，如果是第一次连接，就不会不会弹出(yes/no) 提示
14 | 			ssh.connect(ip,port,username,pkey=key)  			#paramiko用法  http://docs.paramiko.org/en/latest/api/client.html		
15 | 		else:
16 | 			ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
17 | 			ssh.connect(ip,port,username,password)
18 | 			
19 | 		Timeout_i=1
20 | 		while Timeout_i<=int(ListenTime):
21 | 			stdin,stdout,stderr=ssh.exec_command(cmd)
22 | 			out=stdout.readlines()
23 | 			error_out=stderr.readlines()
24 | 			for o in out:
25 | 				outchar+=o
26 | 			for err in error_out:
27 | 				errchar+=err
28 | 			if errchar:
29 | 				print "\033[1;31m-ERR %s Deployment Faild , Check deployment FlagFile error  (%s)\033[0m\a" %(ip,errchar)
30 | 				return False
31 | 			if outchar=='DoneSucc': #shell的终端输出应该是  echo  -n 'DoneSucc'
32 | 				print "\033[1;32m+OK %s Deployment \033[0m" % (ip)
33 | 				return True
34 | 			#print "Wating %s (%d Sec) Deploeymenting..." %(ip,Timeout_i),
35 | 			Timeout_i+=1
36 | 			time.sleep(1)
37 | 		stdin,stdout,stderr=ssh.exec_command("""killall -9 tail 2&>/dev/null""")
38 | 			
39 | 	except Exception,e:
40 | 			print "\033[1;31m-ERR  %s\033[0m\a" % e
41 | 			return False
42 | 
43 | 
44 | if __name__ =='__main__':
45 | 	#LogCollect(ip,port,username,passwd,cmd,Timeout)
46 | 	 LogCollect('127.0.0.1',22,'sshd','zaq1ZAQ!',"""grep  'a'  /tmp/a -q && echo  -n 'DoneSucc'""",10,'n')
47 | 
48 | 


--------------------------------------------------------------------------------
/install-CheungSSH.sh:
--------------------------------------------------------------------------------
 1 | #!/bin/bash
 2 | #Author=Cheung Kei-Chuen  张其川
 3 | #QQ=741345015
 4 | #如果您在使用过程中，遇到了一点点的问题，我都真诚希望您能告诉我！为了改善这个软件， 方便您的工作！
 5 | export LANG=zh_CN.UTF-8
 6 | if [ `id -u` -ne 0 ]
 7 | then
 8 | 	echo "Must be as root install!"
 9 | 	exit 1
10 | fi
11 | echo  "Installing..."
12 | rpm  -qa|grep gcc -q
13 | if  [ $? -ne 0 ]
14 | then
15 | 	echo  "您的系统当前没有gcc环境！"
16 | 	echo "现在为您安装GCC..."
17 | 	yum  install -y gcc >/dev/null 2>/dev/null
18 | 	if [ $? -ne 0 ]
19 | 	then
20 | 		echo  "安装GCC失败了，请手动安装gcc。"
21 | 		exit 1
22 | 	else
23 | 		echo "GCC安装成功!"
24 | 	fi
25 | 
26 | fi
27 | rpm  -qa|grep python-devel -q
28 | if  [ $? -ne 0 ]
29 | then
30 | 	echo  "您的系统没有python-devle包，现在需要安装..."
31 | 	yum  install -y python-devel >/dev/null  2>/dev/null
32 | 	if  [ $? -ne  0 ]
33 | 	then
34 | 		echo "无法安装python-devel，请手动安装python-devel"
35 | 		exit 1
36 | 	else
37 | 		echo "安装python-devel成功！"
38 | 	fi
39 | fi
40 | 
41 | 
42 | 
43 | 
44 | which pip 2>/dev/null
45 | if [ $? -ne 0 ]
46 | then
47 | 	easy_install pip
48 | 	if [ $? -ne 0 ]
49 | 	then
50 | 		echo "安装pip失败，请手动安装pip"
51 | 		exit 1
52 | 	else
53 | 		echo "安装pip成功"
54 | 	fi
55 | fi
56 | 
57 | pip install  pycrypto
58 | if [ $? -ne 0 ]
59 | then
60 | 	echo "安装pycrypto失败，请手动下载安装:https://pypi.python.org/packages/source/p/pycrypto/pycrypto-2.6.1.tar.gz"
61 | 	exit 1
62 | else
63 | 	echo "安装pycrypto成功"
64 | fi
65 | 
66 | pip install  paramiko
67 | if [ $? -ne 0 ]
68 | then
69 | 	echo "安装paramiko失败，请手动下载后安装 https://pypi.python.org/packages/source/p/paramiko/paramiko-1.9.0.tar.gz"
70 | 	exit 1
71 | else
72 | 	echo "安装paramiko成功"
73 | fi
74 | #########
75 | chmod -R a+x CheungSSH*  2>/dev/null
76 | chmod a+x *.py 2>/dev/null
77 | cat <<EOF|python
78 | #encoding:utf-8
79 | try:
80 |  import paramiko
81 | except AttributeError:
82 |  import os
83 |  os.system("""sed  -i '/You should rebuild using libgmp/d;/HAVE_DECL_MPZ_POWM_SEC/d'  /usr/lib64/python*/site-packages/Crypto/Util/number.py       /usr/lib/python*/site-packages/pycrypto*/Crypto/Util/number.py""") #有的系统可能需要修改一下这个语句，否则会报错
84 | EOF
85 | 
86 | echo "恭喜，您已经安装好了环境，接下来请您使用 ./`ls cheungSSH*` 启动程序"
87 | 
88 | 
89 | 


--------------------------------------------------------------------------------
/cheungSSH_Manager_Stand_V96_ISSUE.py:
--------------------------------------------------------------------------------
  1 | #!/usr/bin/python
  2 | #coding:utf8
  3 | #Author=Cheung Kei-Chuen
  4 | #QQ 741345015
  5 | VERSION=96
  6 | import os,sys
  7 | BUILD_CMD=['exit','flush logs']
  8 | os.sys.path.insert(0,os.path.abspath('./'))
  9 | os.sys.path.insert(0,os.path.abspath('/cheung/bin/'))
 10 | try:
 11 | 	import paramiko,threading,socket,ConfigParser,time,commands,threading,re,getpass,Format_Char_Show,shutil,random,getpass,LogCollect,readline
 12 | except Exception,e:
 13 | 	print "\033[1m\033[1;31m-ERR %s\033[0m\a"	% (e)
 14 | 	sys.exit(1)
 15 | reload(sys)
 16 | sys.setdefaultencoding('utf8')
 17 | LogFile='/cheung/logs/cheungssh.log'
 18 | SLogFile='/cheung/logs/cheungssh.source.log'
 19 | DeploymentFlag="/tmp/DeploymentFlag%s" % (str(random.randint(999999999,999999999999)))
 20 | try:
 21 | 	paramiko.util.log_to_file('/cheung/logs/paramiko.log')
 22 | except Exception,e:
 23 | 	pass
 24 | #os.system('stty erase ^H')
 25 | T_V=sys.version.split()[0]
 26 | if int(T_V.replace(".","")) <240:
 27 | 	print "Python's version can not less than 2.4"
 28 | 	print "Please : yum  update  -y  python"
 29 | 	sys.exit(1)
 30 | 
 31 | 
 32 | def Write_Log(ip,stderr,stdout,Logcmd,LogFile,useroot,username,UseLocalScript,Deployment,DeploymentStatus):
 33 | 	if DeploymentStatus:
 34 | 		DeploymentStatus='Y'
 35 | 	else:
 36 | 		DeploymentStatus='N'
 37 | 	Deployment=Deployment.upper()
 38 | 	
 39 | 	try:
 40 | 		T=open(LogFile,"a")
 41 | 		T.write(ip+ '===' + username  + '===' + time.strftime('%Y%m%d%H%M%S',time.localtime())   + '===' + useroot + '===' + UseLocalScript + '===' + Deployment + '===' + DeploymentStatus + '===' + Logcmd + '===' + stderr + '===' + stdout)
 42 | 		T.close()
 43 | 	except Exception,e:
 44 | 		print "Warning: Can't write log. (%s)" % e
 45 | def WriteSourceLog(MSG):
 46 | 	try:
 47 | 		F=open(SLogFile,"a")
 48 | 		F.write(MSG)
 49 | 		F.close()
 50 | 	except Exception,e:
 51 | 		print "Can not write to log (%s)" % (e)
 52 | 	
 53 | def LocalScriptUpload(ip,port,username,password,s_file,d_file):
 54 | 	try:		
 55 | 		t = paramiko.Transport((ip,port))
 56 |                 if UseKey=='y':
 57 |                         KeyPath=os.path.expanduser('~/.ssh/id_rsa')
 58 |                         key=paramiko.RSAKey.from_private_key_file(KeyPath)
 59 | 			t.connect(username = username,pkey=key)
 60 | 		else:
 61 | 			t.connect(username = username,password = password)
 62 | 		sftp = paramiko.SFTPClient.from_transport(t)
 63 | 		ret=sftp.put(s_file,d_file)
 64 | 	except Exception,e:		
 65 | 		print "LocalScript inited Failed",e
 66 | 		return False	
 67 | 	else:
 68 | 		t.close()
 69 | def InitInstall():
 70 | 	if not os.path.isdir("/cheung"):
 71 | 		if getpass.getuser()=="root":
 72 | 			os.system("mkdir -p /cheung/logs /cheung/.db/.key  /cheung/conf /cheung/bin")
 73 | 		else:
 74 | 			print "Sorry Must be as root install !"
 75 | 			sys.exit(1)
 76 | 	if not os.path.isfile('/cheung/conf/cheung.conf'):
 77 | 		T=open('/cheung/conf/cheung.conf','w')
 78 | 		T.write("""[CheungSSH]
 79 | #Author=Cheung Kei-Chuen
 80 | #QQ=741345015
 81 | Servers=localhost,127.0.0.1,www.baidu.com
 82 | Username=YourServerCount
 83 | Password=Yourcount-Password
 84 | Useroot=N
 85 | #localhost_User=apache
 86 | #localhost_Password=apache-password
 87 | #localhost_Port=222
 88 | #如果您的每个服务器的账户对应的密码不是都一样，那么您可以使用这个配置
 89 | #Passwordroot=root-password
 90 | #Timeout=10
 91 | RunMode=M
 92 | UseKey=n
 93 | Deployment=n
 94 | #ListenFile=/var/log/messages
 95 | #ListenTime=60
 96 | #ListenChar=Server startup
 97 | Port=22""")
 98 | 		T.close()
 99 | 
100 | 
101 | 
102 | def SSH_cmd(ip,username,password,port,cmd,UseLocalScript,OPTime):
103 | 	PATH="export PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin;"
104 | 	global All_Servers_num,All_Servers_num_all,All_Servers_num_Succ,Done_Status,Global_start_time,PWD
105 | 	start_time=time.time()
106 | 	ResultSum=''
107 | 	ResultSumLog=''
108 | 	DeploymentStatus=False
109 | 	DeploymentInfo=None
110 | 	try:
111 | 		o=None
112 | 		err=None
113 | 		ssh=paramiko.SSHClient()
114 | 		if UseKey=='y':
115 | 	
116 | 			KeyPath=os.path.expanduser('~/.ssh/id_rsa')
117 | 			###
118 | 			key=paramiko.RSAKey.from_private_key_file(KeyPath)
119 | 			ssh.load_system_host_keys()
120 | 			ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
121 | 			ssh.connect(ip,port,username,pkey=key)  
122 | 		else:
123 | 			ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
124 | 			ssh.connect(ip,port,username,password)
125 | 		if Deployment=='y':
126 | 			stdin,stdout,stderr=ssh.exec_command(PWD+PATH+ListenLog+cmd)
127 | 		else:
128 | 			stdin,stdout,stderr=ssh.exec_command(PWD+PATH+cmd)
129 | 		out=stdout.readlines()
130 | 		All_Servers_num += 1
131 | 		print "\r"
132 | 		for o in out:
133 | 			ResultSum +=o
134 | 			ResultSumLog +=o.strip('\n') + '\\n'
135 | 		
136 | 		error_out=stderr.readlines()
137 | 		for err in error_out:
138 | 			ResultSum +=err
139 | 			ResultSumLog +=err.strip('\n') + '\\n'
140 | 		if err:
141 | 			ResultSum_count="\033[1m\033[1;32m+OK %s (%0.2f Sec, All %d Done %d \033[1m\033[1;31mCmd:Failed\033[1m\033[1;32m)\033[1m\033[0m" % (ip,float(time.time()-start_time),All_Servers_num_all,All_Servers_num)
142 | 			out='Null\n'
143 | 			if Deployment=='y':
144 | 				DeploymentStatus=False
145 | 			Write_Log(ip,ResultSumLog.strip('\\n'),out,cmd,LogFile,'N',username,UseLocalScript,Deployment,DeploymentStatus)
146 | 		else:
147 | 			error_out='NULL'
148 | 			ResultSum_count="\033[1m\033[1;32m+OK %s (%0.2f Sec, All %d Done %d  Cmd:Sucess)\033[1m\033[0m" % (ip,float(time.time()-start_time),All_Servers_num_all,All_Servers_num)
149 | 			All_Servers_num_Succ+=1
150 | 			if Deployment=='y':
151 | 				print  "Wating %s deployment (for %d Sec)..." % (ip,ListenTime)
152 | 				T=LogCollect.LogCollect(ip,port,username,password,"""grep  -E "%s"  %s -q && echo  -n 'DoneSucc'""" % (ListenChar,DeploymentFlag),ListenTime,UseKey)
153 | 				if T:
154 | 					DeploymentStatus=True
155 | 				else:
156 | 					DeploymentInfo="Main commands excuted success, But deployment havn't check suncess info (%s) " %(ListenChar)
157 | 					DeploymentStatus=False
158 | 					
159 | 
160 | 			Write_Log(ip,error_out,ResultSumLog.strip('\\n') + '\n',cmd,LogFile,'N',username,UseLocalScript,Deployment,DeploymentStatus)
161 | 
162 | 		Show_Result=ResultSum + '\n' +ResultSum_count
163 | 		TmpShow=Format_Char_Show.Show_Char(Show_Result+"Time:"+OPTime,0)  
164 | 		WriteSourceLog(TmpShow)
165 | 		print TmpShow
166 | 	except Exception,e:
167 | 		All_Servers_num += 1
168 | 		ResultSum_count="\n\033[1m\033[1;31m-ERR %s %s (%0.2f Sec All %d Done %d)\033[1m\033[0m\a"	% (ip,e,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
169 | 		Show_Result= ResultSum+ResultSum_count
170 | 
171 |                 TmpShow=Format_Char_Show.Show_Char(Show_Result+"Time:"+OPTime,0)
172 |                 WriteSourceLog(TmpShow)
173 |                 print TmpShow
174 | 		#Format_Char_Show.Show_Char(Show_Result+"Time:"+OPTime,1)  
175 | 		Write_Log(ip,str(e),'NULL\n',cmd,LogFile,'N',username,UseLocalScript,Deployment,DeploymentStatus)
176 | 	else:
177 | 		ssh.close()
178 | 	if Deployment=='y' and not  DeploymentStatus:
179 | 		while True:
180 | 			TT=raw_input("%s Deployment not Success (%s) want contiue deployment next server (yes/no) ? " %(ip,DeploymentInfo))
181 | 			if TT=='yes':
182 | 				break
183 | 			elif TT=='no':
184 | 				sys.exit(1)
185 | 	if All_Servers_num == All_Servers_num_all: #这里防止计数器永远相加下去
186 | 		print "+Done (Succ:%d,Fail:%d, %0.2fSec GUN/Linux Cheung Kei-Chuen All Right Reserved)" % (All_Servers_num_Succ,All_Servers_num_all-All_Servers_num_Succ,time.time()-Global_start_time)
187 | 		All_Servers_num =0
188 | 		All_Servers_num_Succ=0
189 | 		Done_Status='end'
190 | 
191 | def Read_config(file="/cheung/conf/cheung.conf"):
192 | 	global Username,Password,Servers,Port,PsswordType,Useroot,Passwordroot,All_pass,All_user,All_port,Timeout,RunMode,UseKey,Deployment,ListenTime,ListenFile,ListenChar
193 | 	c=ConfigParser.ConfigParser()
194 | 	try:
195 | 		c.read(file)
196 | 	except ConfigParser.ParsingError,e:
197 | 		print "The %s format Error.\a\n\t%s" % (file,e)
198 | 		sys.exit(1)
199 | 	try:
200 | 		Servers=c.get("CheungSSH","Servers")
201 | 	except Exception,e:
202 | 		print "No Servers"
203 | 		sys.exit()
204 | 	try:
205 | 		UseKey=c.get("CheungSSH","UseKey").lower()
206 | 	except Exception,e:
207 | 		UseKey='n'
208 | 	
209 | 	if UseKey=='n':
210 | 		try:
211 | 			Password=c.get("CheungSSH","Password")
212 | 		except Exception,e:
213 | 			print "No Password"
214 | 			sys.exit()
215 | 		try:
216 | 			Username=c.get("CheungSSH","Username")
217 | 		except Exception,e:
218 | 			print "No Username"
219 | 			sys.exit()
220 | 	else:
221 | 		Password=None
222 | 		Username=getpass.getuser()
223 | 	try:
224 | 		Port=int(c.get("CheungSSH","Port"))
225 | 	except Exception,e:
226 | 		Port=22
227 | 		print "No Port default 22"
228 | 	try:
229 | 		RunMode=c.get("CheungSSH","RunMode").upper()
230 | 	except Exception,e:
231 | 		RunMode='M'
232 | 		print "No Runmode default Mutiple(M)"
233 | 	try:
234 | 		Deployment=c.get("CheungSSH","Deployment").lower()
235 | 		if Deployment=='y':
236 | 			try:
237 | 				ListenFile=c.get("CheungSSH","ListenFile")
238 | 			except Exception,e:
239 | 				print "In deployment mode ,must be specify ListenFile"
240 | 				sys.exit(1)
241 | 			try:
242 | 				ListenTime=int(c.get("CheungSSH","ListenTime"))
243 | 			except Exception,e:
244 | 				print  "Warning : ListenTime default is 60"
245 | 				ListenTime=60
246 | 			try:
247 | 				ListenChar=c.get("CheungSSH","ListenChar")
248 | 			except Exception,e:
249 | 				print "In deployment mode ,must be specify ListenChar"
250 | 				sys.exit(1)
251 | 	except Exception,e:
252 | 		Deployment='n'
253 | 	if RunMode=='M' and Deployment=='y':
254 | 		print "In Mutiple-threading mode,do not support deployment mode!"
255 | 		sys.exit(1)
256 | 			
257 | 		
258 | 	try:
259 | 		Useroot=c.get("CheungSSH","Useroot").lower()
260 | 		if Useroot=='y' and Deployment=='y':
261 | 			print "In Deployment no support su  - root "
262 | 			sys.exit(1)
263 | 	except Exception,e:
264 | 		Useroot="n"
265 | 	try:
266 | 		Timeout=c.get("CheungSSH","Timeout")
267 | 		try:
268 | 			Timeout=socket.setdefaulttimeout(int(Timeout))
269 | 		except Exception,e:
270 | 			Timeout=10
271 | 			print "Warning: Timeout's Value Error, default=10 (Sec)"
272 | 	except Exception,e:
273 | 		Timeout=socket.setdefaulttimeout(10)
274 | 
275 | 	if Useroot == "y":
276 | 		try:
277 | 			Passwordroot=c.get("CheungSSH","Passwordroot")
278 | 		except Exception,e:
279 | 			print "Need root's Password"
280 | 			sys.exit(1)
281 | 	All_pass={}
282 | 	All_user={}
283 | 	All_port={}
284 | 	for t_server in Servers.split(","):
285 | 		try:
286 | 			t=c.get("CheungSSH","%s_Password" % (t_server))
287 | 			All_pass["%s_Password" % (t_server)]=t
288 | 		except Exception,e:
289 | 			pass
290 | 		try:
291 | 			t=c.get("CheungSSH","%s_Username" % (t_server))
292 | 			All_user["%s_Username" % (t_server)]=t
293 | 		except:
294 | 			pass
295 | 		try:
296 | 			t=c.get("CheungSSH","%s_Port" % (t_server))
297 | 			All_port["%s_Port" % (t_server)]=int(t)
298 | 		except:
299 | 			pass
300 | 	print "Servers:%d|RunMode:%s|UseKey:%s|Deployment:%s|Username:%s  \n" % (len(Servers.split(',')),RunMode,UseKey,Deployment,Username)
301 | 
302 | 
303 | def Upload_file(ip,port,username,password):
304 | 	start_time=time.time()
305 | 	global All_Servers_num,All_Servers_num_all,All_Servers_num_Succ,Global_start_time
306 | 	try:
307 | 		t = paramiko.Transport((ip,port))
308 | 		if UseKey=='y':
309 |                         KeyPath=os.path.expanduser('~/.ssh/id_rsa')
310 |                         key=paramiko.RSAKey.from_private_key_file(KeyPath)
311 | 			try:
312 | 				t.connect(username = username,pkey=key)
313 | 			except EOFError:
314 | 				print "Try use RunMode=D"
315 | 			##########################################
316 | 		else:
317 | 			try:
318 | 				t.connect(username = username,password = password)
319 | 			except EOFError:
320 | 				print "Try use RunMode=D"
321 | 		sftp = paramiko.SFTPClient.from_transport(t)
322 | 		New_d_file=re.sub('//','/',d_file + '/')+ os.path.split(s_file)[1]
323 | 		Bak_File=New_d_file+'.bak'+"%d" % (Global_start_time)
324 | 		try:
325 | 			sftp.rename(New_d_file,Bak_File)
326 | 			SftpInfo="Warning: %s %s  already exists, and has been backed up to %s \n" % (ip,New_d_file,Bak_File)
327 | 		except Exception,e:
328 | 			SftpInfo='\n'
329 | 		ret=sftp.put(s_file,New_d_file)
330 | 		All_Servers_num += 1
331 | 		All_Servers_num_Succ+=1
332 | 		print SftpInfo + "\033[1m\033[1;32m+OK %s (%0.2f Sec  All %d Done %d)\033[1m\033[0m" % (ip,time.time() - start_time,All_Servers_num_all,All_Servers_num)
333 | 	except Exception,e:
334 | 		All_Servers_num += 1
335 | 		print "\033[1m\033[1;31m-ERR %s %s(%0.2f Sec,All %d Done %d)\033[1m\033[0m" % (ip,e,float(time.time() -start_time),All_Servers_num_all,All_Servers_num)	
336 | 	else:
337 | 		t.close()
338 | 
339 | 	if All_Servers_num_all == All_Servers_num:
340 | 		print "+Done (Succ:%d,Fail:%d, %0.2fSec GUN/Linux Cheung Kei-Chuen All Right Reserved)" % (All_Servers_num_Succ,All_Servers_num_all-All_Servers_num_Succ,time.time()-Global_start_time)
341 | 		All_Servers_num =0
342 | 		All_Servers_num_Succ=0
343 | 
344 | 
345 | def Download_file_regex(ip,port,username,password):
346 | 	global All_Servers_num_all,All_Servers_num,All_Servers_num_Succ
347 | 	start_time=time.time()
348 | 	try:
349 | 		t = paramiko.Transport((ip,port))
350 |                 if UseKey=='y':
351 |                         KeyPath=os.path.expanduser('~/.ssh/id_rsa')
352 |                         key=paramiko.RSAKey.from_private_key_file(KeyPath)
353 | 			t.connect(username = username,pkey=key)
354 | 		else:
355 | 			t.connect(username = username,password = password)
356 | 		sftp = paramiko.SFTPClient.from_transport(t)
357 | 		t_get=sftp.listdir(os.path.dirname(s_file))
358 | 		for getfilename in t_get:
359 | 			if re.search(os.path.basename(s_file),getfilename):
360 | 				download_fullpath=os.path.join(os.path.dirname(s_file),getfilename)
361 | 				try:
362 | 					ret=sftp.get(download_fullpath,"%s_%s" % (os.path.join(d_file,getfilename),ip))
363 | 					print  '\t\033[1m\033[1;32m+OK %s : %s' % (ip,download_fullpath)
364 | 				except Exception,e:
365 | 					print  '\t\033[1m\033[1;33m-Failed %s : %s %s' % (ip,download_fullpath,e)
366 | 		All_Servers_num +=1
367 | 		All_Servers_num_Succ+=1
368 | 		print "\033[1m\033[1;32m+OK %s (%0.2f Sec All %d Done %d)\033[1m\033[0m" % (ip,float(time.time()) - start_time,All_Servers_num_all,All_Servers_num)
369 | 	except Exception,e:
370 | 		All_Servers_num +=1
371 | 		print "\033[1m\033[1;31m-ERR %s %s (%0.2f Sec All %d Done %d)\033[1m\033[0m" % (ip,e,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
372 | 	else:
373 | 		t.close()
374 | 	if All_Servers_num_all == All_Servers_num:
375 | 		All_Servers_num = 0
376 | 		print "+Done (Succ:%d,Fail:%d, %0.2fSec GUN/Linux Cheung Kei-Chuen All Right Reserved)" % (All_Servers_num_Succ,All_Servers_num_all-All_Servers_num_Succ,time.time()-Global_start_time)
377 | 		
378 | 
379 | def Download_file(ip,port,username,password):
380 | 	global All_Servers_num_all,All_Servers_num,All_Servers_num_Succ
381 | 	start_time=time.time()
382 | 	try:
383 | 		t = paramiko.Transport((ip,port))
384 |                 if UseKey=='y':
385 |                         KeyPath=os.path.expanduser('~/.ssh/id_rsa')
386 |                         key=paramiko.RSAKey.from_private_key_file(KeyPath)
387 | 			t.connect(username = username,pkey=key)
388 |                	else:
389 | 			t.connect(username = username,password = password)
390 | 		sftp = paramiko.SFTPClient.from_transport(t)
391 | 		New_d_file=re.sub('//','/',d_file + '/')
392 | 		ret=sftp.get(s_file,"%s%s_%s" % (New_d_file,os.path.basename(s_file),ip))
393 | 		All_Servers_num +=1
394 | 		All_Servers_num_Succ+=1
395 | 		print "\033[1m\033[1;32m+OK %s (%0.2f Sec All %d Done %d)\033[1m\033[0m" % (ip,float(time.time()) - start_time,All_Servers_num_all,All_Servers_num)
396 | 	except Exception,e:
397 | 		All_Servers_num +=1
398 | 		print "\033[1m\033[1;31m-ERR %s %s (%0.2f Sec All %d Done %d)\033[1m\033[0m" % (ip,e,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
399 | 	else:
400 | 		t.close()
401 | 	if All_Servers_num_all == All_Servers_num:
402 | 		All_Servers_num = 0
403 | 		print "+Done (Succ:%d,Fail:%d, %0.2fSec GUN/Linux Cheung Kei-Chuen All Right Reserved)" % (All_Servers_num_Succ,All_Servers_num_all-All_Servers_num_Succ,time.time()-Global_start_time)
404 | 
405 | 
406 | 
407 | 
408 | 
409 | 	
410 | def Main_p(Server,Port,Username,Password):
411 | 	global s_file,d_file,All_Servers_num_Succ,LocalScript,Global_start_time
412 | 	global All_Servers_num_all,All_Servers_num
413 | 	All_Servers_num_all=len(Servers.split(','))
414 | 	All_Servers_num    =0
415 | 	All_Servers_num_Succ=0
416 | 	try:
417 | 		from optparse import OptionParser
418 | 		p=OptionParser()
419 | 		p.add_option("-t","--excute-type",help="""Description: select excute type
420 | 			Parameter: [cmd|download|upload]
421 | 				cmd     : Excute Shell Command
422 | 				download: Download file
423 | 				upload  : Upload file
424 | 			
425 | 			Example: %s -t cmd""" % sys.argv[0])
426 | 		p.add_option("-s","--source-file",help="""Description:	Specific Source file  path
427 | 			Example:
428 | 				%s  -t upload   -s /local/file  -d /remote/dir
429 | 				%s  -t download -s /remote/file -d /local/dir""" %(sys.argv[0],sys.argv[0]))
430 | 		p.add_option("-d","--destination-file",help="""
431 | 			Description: Specific a destination directory Path""")
432 | 		p.add_option("-r","--regex",action='store_false',default=True,help="""
433 | 			Description: Use regex match filename
434 | 			Example: 
435 | 			%s  -t download -s '^/remote/tomcat/logs/localhost_2015-0[1-3].*log$' -d  /local/dir/
436 | 
437 | 			Notice: This parameter applies only to download""" % sys.argv[0])
438 | 		p.add_option("-f","--File",action='store_false',default=True,help="""Use LocalScript File""")
439 | 		(option,args)=p.parse_args()
440 | 		if option.File :
441 | 			LocalScript='n'
442 | 			#print 'no use'
443 | 		else:
444 | 			LocalScript='y'
445 | 			#print 'use'
446 | 			
447 | 		if option.excute_type == "cmd":
448 | 			Excute_cmd()
449 | 		elif option.excute_type == "upload":
450 | 			if option.source_file and option.destination_file:
451 | 				s_file=option.source_file
452 | 				d_file=option.destination_file
453 | 			else:
454 | 				print "Upload File"
455 | 				s_file=raw_input("Local Source Path>>>")
456 | 				d_file=raw_input("Remote Destination Full-Path>>>")
457 | 			Global_start_time=time.time()
458 | 			for s in Servers.split(","):
459 | 				try:
460 | 						t_password=All_pass["%s_Password" % (s)]
461 | 				except:
462 | 						t_password=Password
463 | 				try:
464 | 						t_username=All_user["%s_Username" % (s)]
465 | 				except:
466 | 						t_username=Username
467 | 				try:
468 | 						t_port=All_port["%s_Port" % (s)]
469 | 				except:
470 | 						t_port=Port
471 | 				if RunMode.upper()=='M':
472 | 					a=threading.Thread(target=Upload_file,args=(s,t_port,t_username,t_password))
473 | 					a.start()
474 | 				else:
475 | 					Upload_file(s,t_port,t_username,t_password)
476 | 		elif option.excute_type == "download":
477 | 			if option.source_file and option.destination_file:
478 | 				s_file=option.source_file
479 | 				d_file=option.destination_file
480 | 			else:
481 | 				print "Download File"
482 | 				s_file=raw_input("Remote Source Full-Path>>>")
483 | 				d_file=raw_input("Local Destination Path>>>")
484 | 			if not os.path.isdir(d_file):
485 | 				print 'Recv location must be a directory'
486 | 				sys.exit(1)
487 | 			Global_start_time=time.time()
488 | 			for s in Servers.split(","):
489 | 				
490 | 				try:
491 | 					t_password=All_pass["%s_Password" % (s)]
492 | 				except:
493 | 					t_password=Password
494 | 				try:
495 | 					t_username=All_user["%s_Username" % (s)]
496 | 				except:
497 | 					t_username=Username
498 | 				try:
499 | 					t_port=All_port["%s_Port" % (s)]
500 | 				except:
501 | 					t_port=Port
502 | 
503 | 				if option.regex:
504 | 					a=threading.Thread(target=Download_file,args=(s,t_port,t_username,t_password))
505 | 				else:
506 | 					a=threading.Thread(target=Download_file_regex,args=(s,t_port,t_username,t_password))
507 | 				a.start()
508 | 		elif not option.excute_type:
509 | 			Excute_cmd()
510 | 			sys.exit(0)
511 | 		else:
512 | 			print "Parameter does not currently support\t(%s)\a" % (option.excute_type)
513 | 			Excute_cmd()
514 | 	except KeyboardInterrupt:
515 | 		print "exit"
516 | 	except EOFError:
517 | 		print "exit"
518 | 
519 | def Excute_cmd_root(s,Port,Username,Password,Passwordroot,cmd,UseLocalScript,OPTime):
520 | 	global All_Servers_num_all,All_Servers_num,All_Servers_num_Succ,Done_Status,bufflog
521 | 	Done_Status='start'
522 | 	bufflog=''
523 | 	start_time=time.time()
524 | 	ResultSum=''
525 | 	Result_status=False
526 | 	try:
527 | 		t=paramiko.SSHClient()
528 |                 if UseKey=='y':
529 | 			KeyPath=os.path.expanduser('~/.ssh/id_rsa')
530 |                         key=paramiko.RSAKey.from_private_key_file(KeyPath)
531 |                         t.load_system_host_keys()
532 | 			t.set_missing_host_key_policy(paramiko.AutoAddPolicy())
533 |                         t.connect(s,Port,Username,pkey=key) 
534 |                 else:
535 | 			t.set_missing_host_key_policy(paramiko.AutoAddPolicy())
536 | 			t.connect(s,Port,Username,Password)
537 | 		ssh=t.invoke_shell()
538 | 		ssh.send("LANG=zh_CN.UTF-8\n")
539 | 		ssh.send("export LANG\n")
540 | 		ssh.send("su - root\n")
541 | 		buff=''
542 | 		while not re.search("Password:",buff) and not re.search("：", buff):
543 | 			resp=ssh.recv(9999)
544 | 			buff += resp
545 | 		#print buff #show su result
546 | 		ssh.send("%s\n" % (Passwordroot))
547 | 		buff1=''
548 | 		while True:
549 | 			resp=ssh.recv(500)
550 | 			buff1 += resp
551 | 			if  re.search('su:',buff1):
552 | 				#print "\033[1;31m-ERR su Failed  %s\033[0m" % (s)
553 | 				break
554 | 			else:
555 | 				if re.search('# *$',buff1):
556 | 					Result_status=True
557 |                 			All_Servers_num_Succ+=1
558 | 					break
559 | 		if Result_status:
560 | 			ssh.send(PWD)
561 | 			ssh.send("%s\n" % (cmd))
562 | 			buff=""
563 | 			bufflog=''
564 | 			while not buff.endswith("# "):
565 | 				resp=ssh.recv(9999)
566 | 				buff  += resp
567 | 				bufflog  += resp.strip('\r\n') + '\\n'
568 | 			t.close()
569 | 			All_Servers_num += 1
570 | 			ResultSum=buff + "\n\033[1m\033[1;32m+OK %s (%0.2f Sec All %d Done %d)\033[1m\033[0m\n" % (s,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
571 | 			
572 | 			bufflog_new=''
573 | 			for t in bufflog.split():
574 | 				if t==cmd:
575 | 					continue
576 | 				bufflog_new+=t
577 | 			bufflog=bufflog_new
578 | 		else:
579 | 			All_Servers_num += 1
580 | 			ResultSum=buff + "\n\033[1m\033[1;31m-ERR Su Failed %s (%0.2f Sec All %d Done %d)\033[1m\033[0m\n" % (s,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
581 | 			
582 | 	except Exception,e:
583 | 		All_Servers_num += 1
584 | 		Result_status=False
585 | 		ResultSum="\n\033[1m\033[1;31m-ERR %s %s (%0.2f Sec All %d Done %d)\033[1m\033[0m\a"   % (e,s,float(time.time() - start_time),All_Servers_num_all,All_Servers_num)
586 | 		bufflog=str(e)
587 | 	if Result_status:
588 | 		Write_Log(s,'NULL',bufflog.strip('\\n') + '\n',cmd,LogFile,'Y',Username,UseLocalScript,'N','N')
589 | 		TmpShow=Format_Char_Show.Show_Char(Show_Result+"Time:"+OPTime,0)
590 | 		WriteSourceLog(TmpShow)
591 | 		print TmpShow
592 | 		#Format_Char_Show.Show_Char(ResultSum+"Time:"+OPTime,0)
593 | 	else:
594 | 		Write_Log(s,bufflog.strip('\\n'),'NULL\n',cmd,LogFile,'Y',Username,UseLocalScript,'N','N')
595 | 		TmpShow=Format_Char_Show.Show_Char(Show_Result+"Time:"+OPTime,0)
596 | 		WriteSourceLog(TmpShow)
597 | 		print TmpShow
598 | 		#Format_Char_Show.Show_Char(ResultSum+"Time:"+OPTime,1)
599 | 	if All_Servers_num_all == All_Servers_num:
600 | 		print "+Done (Succ:%d,Fail:%d, %0.2fSec GUN/Linux Cheung Kei-Chuen All Right Reserved)" % (All_Servers_num_Succ,All_Servers_num_all-All_Servers_num_Succ,time.time()-Global_start_time)
601 |                 All_Servers_num =0
602 |                 All_Servers_num_Succ=0
603 | 		Done_Status='end'
604 | 
605 | def Excute_cmd():
606 | 	global All_Servers_num_all,All_Servers_num,All_Servers_num_Succ,Done_Status,Logcmd,ListenLog,Global_start_time,PWD
607 | 	Done_Status='end'
608 | 	All_Servers_num_all=len(Servers.split(','))
609 | 	All_Servers_num    =0
610 | 	All_Servers_num_Succ=0
611 | 	UseLocalScript='N' #
612 | 	PWD=''
613 | 	IS_PWD=False
614 | 	while True:
615 | 		OPTime=time.strftime('%Y%m%d%H%M%S',time.localtime())
616 | 		if Done_Status=='end':
617 | 			if Useroot == "y":
618 | 				cmd=raw_input("CheungSSH root>>>>")
619 | 			else:
620 | 				cmd=raw_input("CheungSSH>>>>")
621 | 				
622 | 		else:
623 | 			time.sleep(0.05)
624 | 			continue
625 | 
626 | 		try:
627 | 			if not IS_PWD:
628 | 				PWD=re.search("^ *cd.*",cmd).group() +";"
629 | 				Done_Status='end'
630 | 				IS_PWD=True
631 | 				continue
632 | 			else:
633 | 				try:
634 | 					PWD=re.search("^ *cd.*",cmd).group() +";"
635 | 					Done_Status='end'
636 | 					IS_PWD=True
637 | 					continue
638 | 				except:
639 | 					pass
640 | 		except Exception,e:
641 | 			PWD=''
642 | 
643 | 		if re.search("^ *[Ee][Xx][Ii][Tt] *",cmd):
644 | 			sys.exit(0)
645 | 		if re.search("^ *[Cc][Ll][Ee][Aa][Rr] *",cmd):
646 | 			print "请使用ctrl + L 清屏"
647 | 			continue
648 | 		if re.search('^ *[Ff][Ll][Uu][Ss][Hh] *[Ll][Oo][Gg][Ss] *$',cmd):
649 | 			try:
650 | 				Log_Flag=time.strftime('%Y%m%d%H%M%S',time.localtime())
651 | 				shutil.move('/cheung/logs/cheungssh.log','/cheung/logs/cheungssh%s.log' % Log_Flag)
652 | 				print "+OK"
653 | 				continue
654 | 			except Exception,e:
655 | 				print "Waring : %s Failed (%s)" % (cmd,e)
656 | 				continue
657 | 		if cmd=="reload":
658 | 			Read_config()
659 | 			print "+OK"
660 | 			continue
661 | 			
662 | 		
663 | 		if not cmd :
664 | 			continue
665 | 		Global_start_time=time.time()
666 | 		for s in Servers.split(","):
667 | 			
668 | 			try:
669 | 				t_password=All_pass["%s_Password" % (s)]
670 | 			except:
671 | 				t_password=Password
672 | 			try:
673 | 				if UseKey=="y":
674 | 					t_username=Username
675 | 				else:
676 | 					t_username=All_user["%s_Username" % (s)]
677 | 			except:
678 | 				t_username=Username
679 | 			try:
680 | 				t_port=All_port["%s_Port" % (s)]
681 | 			except:
682 | 				t_port=Port
683 | 			if LocalScript.lower() == 'y':
684 | 				if not os.path.isfile(cmd):
685 | 					print "Eroor: %s ScriptFile is not fond " % cmd
686 | 					#continue
687 | 					break
688 | 				else:
689 | 					ScriptFlag=str(random.randint(999999999,999999999999))
690 | 					d_file='/tmp/' + os.path.basename(cmd) + ScriptFlag
691 | 					LocalScriptUpload(s,t_port,t_username,t_password,cmd,d_file)
692 | 					Newcmd="""chmod a+x %s;%s;rm -f %s""" % (d_file,d_file,d_file)
693 | 					Logcmd=cmd
694 | 					UseLocalScript='Y'
695 | 			else:
696 | 				Newcmd=cmd
697 | 				Logcmd=cmd
698 | 			
699 | 			Done_Status='start'
700 | 			if RunMode.upper()=='M':
701 | 				if Useroot=='y':
702 | 					a=threading.Thread(target=Excute_cmd_root,args=(s,t_port,t_username,t_password,Passwordroot,Newcmd,UseLocalScript,OPTime))
703 | 					a.start()
704 | 				else:
705 | 					
706 | 					a=threading.Thread(target=SSH_cmd,args=(s,t_username,t_password,t_port,Newcmd,UseLocalScript,OPTime))
707 | 					a.start()
708 | 					
709 | 			else:
710 | 				if Useroot=='y':
711 | 					Excute_cmd_root(s,Port,t_username,t_password,Passwordroot,Newcmd,UseLocalScript,OPTime)
712 | 				else:
713 | 					if Deployment=='y':
714 | 						ListenLog="""if [ ! -r %s ] ; then echo -e '\033[1m\033[1;31m-ERR ListenFile %s  not exists,so do not excute commands !\033[1m\033[0m\a ' 1>&2 ;exit;else nohup tail -n 0 -f  %s  2&>%s &   fi;""" % (ListenFile,ListenFile,ListenFile,DeploymentFlag)
715 | 						SSH_cmd(s,t_username,t_password,t_port,Newcmd,UseLocalScript,OPTime)
716 | 					else:
717 | 						SSH_cmd(s,t_username,t_password,t_port,Newcmd,UseLocalScript,OPTime)
718 | 							
719 | 			############################################################################################
720 | if  __name__=='__main__':
721 | 	InitInstall()
722 | 	Read_config()
723 | 	Main_p(Servers,Port,Username,Password)
724 | 


--------------------------------------------------------------------------------