前端发布脚本

1. #!/usr/bin/python2.7
2. #coding=utf-8
3. '''
4. Created on Oct 19
5. @author wangyunhua@smeyun.com Update Version TO Wan
6. Version 1.1
8. '''
10. import os
11. import time
12. import MySQLdb
13. import json
14. import logging
15. import paramiko
16. import traceback as tb
17. from argparse import ArgumentParser
19. l = logging.getLogger('web')
21. #获取时间用于打包备份命名
22. D_TIME = time.strftime('%Y%m%d%H%M')
23. #作为结果输出
24. logs = []
25. #获取脚本当前目录
26. cfd = os.path.dirname(os.path.abspath(__file__))
27. SystemUpdatePath="/data/BusinesstoCustomer/program/"
28. Version_Url='http://192.168.200.141:8557/webdir/'
30. update_service = []
31. Service_Names = []
32. GROUPS_Names = []
34. MYSQLHOST='127.0.0.1'
35. MYSQLUSER='root'
36. MYSQLPASSWD='Feng!1900'
38. def Server_sql():
39. conn = MySQLdb.connect(host=MYSQLHOST,user=MYSQLUSER,passwd=MYSQLPASSWD,db='ccdb')
40. curs = conn.cursor()
41. try:
42. curs.execute('''select d.WebName,d.WD_path,e.webhosts,e.WI_path,e.Wsystem_class,f.muser,f.mpass,f.rpass from t_web_ID_name d,t_web_list e,t_password f where d.WebID = e.WebID and f.ip = e.webhosts and e.WL_status =1;''')
43. conn.commit()
44. data = curs.fetchall()
45. #print u'fetchall()返回的数据：',data
46. conn.close()
47. jsonData = []
48. for row in data:
49. result = {}
50. result['WebName'] = str(row[0])
51. result['WD_path'] = str(row[1])
52. result['webhosts'] = row[2]
53. result['WI_path'] = str(row[3])
54. result['Wsystem_class'] = str(row[4])
55. result['muser'] = str(row[5])
56. result['mpass'] = str(row[6])
57. result['rpass'] = str(row[7])
58. jsonData.append(result)
59. #l.debug('转换为列表字典的原始数据 : %r',jsonData)
60. return json.dumps(jsonData)
61. except:
62. print 'MySQL connect fail...'
63. else:
64. #使用json.dumps将数据转换为json格式，json.dumps方法默认会输出成这种格式"u5377u76aeu6298u6263"，加ensure_ascii=False，则能够防止中文乱码。
65. #JSON采用完全独立于语言的文本格式，事实上大部分现代计算机语言都以某种形式支持它们。这使得一种数据格式在同样基于这些结构的编程语言之间交换成为可能。
66. #json.dumps()是将原始数据转为json（其中单引号会变为双引号），而json.loads()是将json转为原始数据。
67. jsondatar=json.dumps(jsonData,ensure_ascii=False,sort_keys=True)
68. #去除首尾的中括号
69. return jsondatar[1:len(jsondatar)-1]
71. def pack(ob,webnames):
72. #取出相关服务器信息和状态
73. file_data=json.loads(Server_sql())
74. for wname in webnames:
75. for aa in file_data:
76. for obi in ob:
77. if aa['WebName'] == wname and aa['Wsystem_class'] == obi:
78. service_object = services(aa['WebName'],aa['WD_path'],aa['webhosts'],aa['WI_path'],aa['Wsystem_class'],aa['muser'],aa['mpass'],aa['rpass'])
79. update_service.append(service_object)
81. def pack_info():
82. #取出相关服务器信息和状态
83. file_data=json.loads(Server_sql())
84. for aa in file_data:
85. service_name = aa['WebName']
86. groups_name = aa['Wsystem_class']
87. Service_Names.append(service_name)
88. Service_Names.append('all')
90. GROUPS_Names.append(str(groups_name))
91. GROUPS_Names.append('all')
93. def scall(CMD_LINE):
94. '''linux shell '''
95. l.debug('CMD_LINE : %r',CMD_LINE)
96. return call(CMD_LINE,shell=True)
98. def os_system(cmd):
99. result = os.popen(cmd)
100. res = result.read()
101. #for line in res.splitlines():
102. #print line
103. return res.splitlines()
105. #封装的自定义打印模块
106. def head_print():
107. print '33[1;31;40m'
108. #print '*' * 50
109. #print ' '
111. #封装的自定义打印模块
112. def tail_print():
113. #print ' '
114. #print '*' * 50
115. print '33[0m'
117. #输出操作结果
118. def output_result():
119. head_print()
120. for count in range(len(logs)):
121. print " "
122. for content in logs[count]:
123. if 'MD5校验成功！' in content:
124. print content
125. break
126. print content
127. tail_print()
129. class services:
130. def __init__(self,WebName,WD_path,webhosts,WI_path,Wsystem_class,muser,mpass,rpass):
131. self.WebName = WebName
132. self.WD_path = WD_path
133. self.WI_path = WI_path
134. self.webhosts = webhosts
135. self.Wsystem_class = Wsystem_class
136. self.muser = muser
137. self.mpass = mpass
138. self.rpass = rpass
140. def connect(self,cmd,allow_agent=False,look_for_keys=False):
141. s=paramiko.SSHClient()
142. s.load_system_host_keys()
143. s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
144. s.connect(hostname=self.webhosts,port=22,username=self.muser, password=self.mpass)
145. if self.muser != 'root':
146. ssh = s.invoke_shell()
147. time.sleep(0.1)
148. ssh.send('su - ')
149. buff = ''
150. while not buff.endswith('Password: '):
151. resp = ssh.recv(9999)
152. buff +=resp
153. ssh.send(self.rpass)
154. ssh.send(' ')
155. buff = ''
156. while not buff.endswith('# '):
157. resp = ssh.recv(9999)
158. buff +=resp
159. ssh.send(cmd)
160. ssh.send(' ')
161. buff = ''
162. while not buff.endswith('# '):
163. resp = ssh.recv(9999)
164. buff +=resp
165. s.close()
166. result = buff
167. else:
168. stdin, stdout, stderr = s.exec_command(cmd)
169. result = stdout.read()
170. s.close()
171. print result
172. return result
174. def execute(self,*cmd):
175. stdin, stdout, stderr = self.client.exec_command(*cmd)
176. print stdout.readlines()
177. print stderr.readlines()
179. def putFile(self,localPath,remotePath):
180. ftp = self.client.open_sftp()
181. ftp.chmod(remotePath,777)
182. if os.path.isdir(localPath):
183. for f in os.listdir(localPath):
184. remotefile = os.path.join(remotePath,f)
185. localf = os.path.join(localPath,f)
186. if os.path.isfile(localf):
187. ftp.put(localf,remotefile)
188. else:
189. files = os.path.split(localPath)[-1]
190. remotefile = os.path.join(remotePath,fles)
191. ftp.put(localPath,'%s/'%remotefile)
193. #关闭远端服务器tomcat
194. def test_remote(self):
195. #关闭远端进程
196. shl='''ifconfig eth0|grep inet'''
197. l.debug('%r',shl)
198. self.connect(cmd=shl)
200. def ps_prcee(self):
201. shl='''ps aux|grep nginx'''
202. l.debug('%r',shl)
203. self.connect(cmd=shl)
205. def push_prcee(self):
206. shl='''wget -NP %s %s'''%(self.WI_path,Version_Url+self.WebName+'/'+self.WebName+'.zip')
207. l.debug('%r',shl)
208. self.connect(cmd=shl)
210. def backup_prcee(self):
211. shl='''cd %s/backup;tar zcvf %s %s;cd ..;rm -rf *.zip'''%(self.WI_path,self.WebName+D_TIME+'.tar',self.WI_path+'/'+self.WebName)
212. l.debug('%r',shl)
213. self.connect(cmd=shl)
215. def unzip_prcee(self):
216. shl='''cd %s;unzip -o %s.zip'''%(self.WI_path,self.WebName)
217. l.debug('%r',shl)
218. self.connect(cmd=shl)
222. #检测参数，执行操作
223. def handle(args):
224. logging.basicConfig(format="%(message)s")
225. l.level = logging.INFO
226. if args.verbose:
227. l.level =logging.DEBUG
229. op = args.operation
230. ob = args.object
231. webnames = args.webname
233. ss = services
235. if 'all' in webnames:
236. pack_info()
237. Service_Names.remove('all')
238. webname = set(Service_Names)
239. pack(ob,webname)
240. else:
241. pack(ob,webnames)
243. if op == 'test':
244. for content in update_service:
245. ss.test_remote(content)
247. if op == 'cupdate':
248. for content in update_service:
249. ss.backup_prcee(content)
250. for content in update_service:
251. ss.push_prcee(content)
252. for content in update_service:
253. ss.unzip_prcee(content)
254. for content in update_service:
255. ss.ps_prcee(content)
257. if op == 'backup':
258. for content in update_service:
259. ss.backup_prcee(content)
261. if op == 'ps':
262. for content in update_service:
263. ss.ps_prcee(content)
265. if op == 'push':
266. for content in update_service:
267. ss.push_prcee(content)
269. if op == 'unzip':
270. for content in update_service:
271. ss.unzip_prcee(content)
272. #检测输入，
273. def main():
274. pack_info()
275. ACTIONS = ['test','ps','push','cupdate','backup','unzip']
276. webname = set(Service_Names)
277. GROUPS = set(GROUPS_Names)
278. parser = ArgumentParser()
279. try:
280. parser.add_argument('-v','--verbose',action='store_true',help='detail output')
281. parser.add_argument('operation',choices=ACTIONS,help='action name')
282. parser.add_argument('-o','--object',nargs='+',choices=GROUPS,help='object name')
283. parser.add_argument('-w','--webname',nargs='+',choices=webname,help='update servername')
284. except:
285. print '%r'%tb.format_exc()
286. args = parser.parse_args()
287. handle(args)
288. output_result()
290. if __name__ == '__main__':
291. main()