游戏服务器端自动更新脚本(python)

本脚本用于游戏服务器端的自动更新，主要流程就是把更新包和脚本的配置文件放在一个中心端center，分支机房client去中心端center下载更新包，并验证MD5，分支机房的游戏服务器根据自身需求通过rsync下载相应的更新包，游戏服务器通过解压、拷贝、验证部分重要文件的MD5来确保更新正确完成，目前此脚本已经用于更新本公司多款游戏，兼容windows、linux系统（根据自身需要更改相关路径），可以用页面去调用，已经写了post页面返回值。

1. #!/usr/bin/env Python  
2. # -*- coding: utf-8 -*-  
3. #Used update game path to server  
4. import os,re,sys,urllib,urllib2,hashlib,time,shutil,platform  
5.  
6. def post(status, type, info, err_info=""):  
7.     post_url = args["post_url"]  
8.     aid = args["aid"]  
9.     data = {"aid": aid, "status": status, "type": type, "info": info}  
10.     if err_info:  
11.         data = {"aid": aid, "status": status, "type": type, "info": info, "err_info": err_info}  
12.     print data  
13.     f = urllib2.urlopen(url=post_url, data=urllib.urlencode(data))  
14.  
15. def md5sum(file_name):  
16.     if os.path.isfile(file_name):  
17.         f = open(file_name,'rb')  
18.         py_ver = sys.version[:3]  
19.         if py_ver == "2.4":  
20.             import md5 as hashlib  
21.         else:  
22.             import hashlib  
23.             md5 = hashlib.md5(f.read()).hexdigest()  
24.             f.close()  
25.             return md5  
26.     else:  
27.         return 0 
28.  
29. def config(args,files):  
30.     try:  
31.         game = args["main_prefix"]  
32.         url = "http://208.asktao.com/autoupdate/%s/config.ini" % game  
33.         get = urllib.urlopen(url)  
34.         aa = get.readlines()  
35.         w = {}  
36.         for i in aa:  
37.             a = i.strip().split()[0]  
38.             if "[" in a:  
39.                 x = a.strip("[]")  
40.                 w[x] = {}  
41.                 continue 
42.             w[x][i.strip().split()[1].strip()] = i.strip().split()[0].strip()  
43.         for key in w:  
44.             if key == files:  
45.                 return w[key]  
46.     except Exception,e:  
47.         return 0 
48.  
49. class down_start():  
50.  
51.     def work(self,args):  
52.         aa = config(args,"path_md5")  
53.         game = args["main_prefix"]  
54.         if aa == 0:  
55.             post(2,"read update config","Not find %s config file" % game)  
56.             sys.exit()  
57.         local = '/data/autoupdate/' 
58.         url = "http://208.asktao.com/autoupdate/%s/" % game  
59.         for f in aa:  
60.             md5_r = f.strip().split()[0]  
61.             pkg_name = f.strip().split()[1]  
62.             get = urllib.urlopen(os.path.join(url,pkg_name))  
63.             status = get.getcode()  
64.             if status == 200:#验证MD5，MD5错误的话重新下载一次，再次错误就提示失败，退出程序  
65.                 urllib.urlretrieve(os.path.join(url,pkg_name),os.path.join(local,pkg_name),)  
66.                 md5_l = md5sum(os.path.join(local,pkg_name))  
67.                 if md5_l == md5_r:  
68.                     post(1,"down",pkg_name)  
69.                 else:  
70.                     urllib.urlretrieve(os.path.join(url,pkg_name),os.path.join(local,pkg_name),)  
71.                     md5_l = md5sum(os.path.join(local,pkg_name))  
72.                     if md5_l == md5_r:  
73.                         post(1,"down",pkg_name)  
74.                     else:  
75.                         post(2,"down",pkg_name,"Download %s,MD5 not right" % pkg_name)  
76.                         sys.exit()  
77.             else:  
78.                 post(2,"down",pkg_name,"Not find %s path file" % pkg_name)  
79.         # 检查rsync服务是否启动  
80.         while True:  
81.             pid = os.popen("ps auxww | grep 'rsync --daemon' | grep -v grep").read()  
82.             if not pid:  
83.                 getso("/usr/bin/rsync --daemon")  
84.                 continue 
85.             break 
86.         #全部完成，更新中心状态  
87.         post(1,"down","down_done")  
88.  
89. class sync_start():  
90.  
91.     def rsync(self,update_pkg,args):  
92.         node_ip = args["node_ip"]  
93.         game = args["main_prefix"]  
94.         os_type = platform.system()  
95.         if os_type == "Windows":  
96.             local = "C:\\update\\" 
97.         elif os_type == "Linux":  
98.             local = "/home/update/tmp/" 
99.         if not os.path.isdir(local):  
100.             os.mkdir(local)  
101.         for f in os.listdir(local):  
102.             if os.path.isfile(f):  
103.                 os.remove(os.path.join(local,f))  
104.             elif os.path.isdir(f):  
105.                 shutil.rmtree(os.path.join(local,f))  
106.         aa = config(args,"path")  
107.         if aa == 0:  
108.             post(2,"read update config","Not find %s config file" % game)  
109.             sys.exit()  
110.         for f in aa:  
111.             if update_pkg in f:  
112.                 md5_r = aa[f]  
113.                 pkg_name = f  
114.                 if os_type == "Windows":  
115.                     sync = "rsync -avz %s::update/*%s* /cygdrive/c/update/" % (node_ip,update_pkg)  
116.                 elif os_type == "Linux":  
117.                     sync = "rsync -avz %s::update/*%s* /home/update/tmp/" % (node_ip,update_pkg)  
118.                 result = os.popen(sync).readlines()  
119.                 md5_l = md5sum(os.path.join(local,pkg_name))  
120.                 if md5_l == md5_r:  
121.                     post(1,"sync",pkg_name)  
122.                 else:  
123.                     if os_type == "Windows":  
124.                         sync = "rsync -avz %s::update/*%s* /cygdrive/c/update/" % (node_ip,update_pkg)  
125.                     elif os_type == "Linux":  
126.                         sync = "rsync -avz %s::update/*%s* /home/update/tmp/" % (node_ip,update_pkg)  
127.                     result = os.popen(sync).readlines()  
128.                     md5_l = md5sum(os.path.join(local,pkg_name))  
129.                     if md5_l == md5_r:  
130.                         post(1,"sync",pkg_name)  
131.                     else:  
132.                         post(2,"rsync",pkg_name,"sync Error,%s not find or file's md5 wrong" % pkg_name)  
133.                         sys.exit()  
134.  
135.     def work(self,args):  
136.         update_pkg = args["update_pkg"].split(',')  
137.         for i in range(len(update_pkg)):  
138.             self.rsync(update_pkg[i],args)  
139.         post(1,"sync","sync_done")  
140.  
141. class update_start():  
142.  
143.     def copy(self,src, dst):  
144.         if os.path.isdir(src):  
145.             base = os.path.basename(src)  
146.             if os.path.exists(dst):  
147.                 dst = os.path.join(dst, base)  
148.             if not os.path.exists(dst):  
149.                 os.makedirs(dst)  
150.             names = os.listdir(src)  
151.             for name in names:  
152.                 srcname = os.path.join(src, name)  
153.                 self.copy(srcname, dst)  
154.         else:  
155.             shutil.copy2(src, dst)  
156.  
157.     def unrar(self,src,dst,args):  
158.         os_type = platform.system()  
159.         try:  
160.             if not os.path.exists(dst) or not os.path.exists(src):  
161.                 raise Exception, "%s or %s not exist!" % (src, dst)  
162.             if os_type == "Windows":  
163.                 os.system(r'C:\Progra~1\WinRAR\rar x -o+ -inul %s %s' % (src, dst))  
164.             elif os_type == "Linux":  
165.                 if os.path.splitext(src)[1] == ".tgz":  
166.                     os.system("tar -zxf %s -C %s" % (src, dst))  
167.                 elif os.path.splitext(src)[1] == ".zip":  
168.                     os.system("unzip -oq %s -d %s" % (src, dst))  
169.             return 0 
170.         except Exception,e:  
171.             return e  
172.  
173.     def work(self,args):  
174.         os_type = platform.system()  
175.         if os_type == "Windows":  
176.             update = "C:\\update\\" 
177.         elif os_type == "Linux":  
178.             update = "/home/update/tmp/" 
179.         server_dir = {"tmcs":"C:\\Server\\","wd":"/home/asktao/"}  
180.         server = server_dir[args["main_prefix"]]  
181.         update_pkg = args["update_pkg"].split(',')  
182.         game = args["main_prefix"]  
183.         for i in range(len(update_pkg)):   
184. #####################解压更新包  
185.             for tgz in os.listdir(update):  
186.                 if update_pkg[i] in tgz:  
187.                     src = os.path.join(update,tgz)  
188.                     r = self.unrar(src,update,args)  
189.                     if r == 0:  
190.                         post(1,"update","%s unzip success" % src)  
191.                     else:  
192.                         post(2,"update",src,"unzip fail:%s" % r)  
193. #####################拷贝更新文件到游戏目录  
194.             ser_list = os.listdir(server)  
195.             up_list = os.listdir(update)  
196.             for dir in up_list:  
197.                 for line in ser_list:  
198.                     filepath = os.path.join(update,dir+"\\")  
199.                     serv = os.path.join(server,line+"\\")  
200.                     if dir in line:  
201.                         self.copy(filepath,serv)  
202.                         post(1,"update","%s files copy success" % line)  
203. #####################验证重要文件MD5  
204.         aa = config(args,"files")  
205.         if aa == 0:  
206.             post(2,"read update config","Not find %s config file" % game)  
207.             sys.exit()  
208.         for f in aa:  
209.             md5_r = aa[f]  
210.             pkg_name = f  
211.             if os.path.exists(pkg_name):  
212.                 md5_s = md5sum(pkg_name)  
213.                 if md5_r == md5_s:  
214.                     post(1,"update","%s md5 Ok,update success" % pkg_name)  
215.                 else:  
216.                     post(2,"update","%s md5 Error,update fail" % pkg_name)  
217.                     sys.exit()  
218.         shutil.rmtree(update)  
219.         os.mkdir(update)  
220.         post(1,"update","All_done")  
221.  
222. if __name__ == "__main__":  
223.     args = {"pack":"auto_update","func1":"down_start","url":"http://208.2222.com/manage/auto_update","post_url":"http://192.168.50.209/reg.php","update_pkg":"up,up","node_ip":"192.168.50.208","main_prefix":"tmcs","aid":"123"}  
224.     #down = down_start()  
225.     #down.work(args)  
226.     sync = sync_start()  
227.     sync.work(args)  
228.     update = update_start()  
229.     update.work(args)  
230.  
231.  
232. '''''config.ini内容  
233. [files]#需要验证MD5的重要文件绝对路径和MD5  
234. 15a6605156e29f68fdfd637e73a889d4  C:\Server\Line1\SAFlashPlayer.exe  
235. 15a6605156e29f68fdfd637e73a889d4  C:\Server\Line2\SAFlashPlayer.exe  
236. [path]#更新包名字和MD5  
237. 32fcb8932799aa553db13b5b9b41e5e9  auto.rar  
238. 32fcb8932799aa553db13b5b9b41e5e9  update.rar  
239. '''