本脚本用于游戏服务器端的自动更新,主要流程就是把更新包和脚本的配置文件放在一个中心端center,分支机房client去中心端center下载更新包,并验证MD5,分支机房的游戏服务器根据自身需求通过rsync下载相应的更新包,游戏服务器通过解压、拷贝、验证部分重要文件的MD5来确保更新正确完成,目前此脚本已经用于更新本公司多款游戏,兼容windows、linux系统(根据自身需要更改相关路径),可以用页面去调用,已经写了post页面返回值。
-
-
-
- import os,re,sys,urllib,urllib2,hashlib,time,shutil,platform
-
- def post(status, type, info, err_info=""):
- post_url = args["post_url"]
- aid = args["aid"]
- data = { "aid": aid, "status": status, "type": type, "info": info}
- if err_info:
- data = { "aid": aid, "status": status, "type": type, "info": info, "err_info": err_info}
- print data
- f = urllib2.urlopen(url=post_url, data=urllib.urlencode(data))
-
- def md5sum(file_name):
- if os.path.isfile(file_name):
- f = open(file_name,'rb')
- py_ver = sys.version[:3]
- if py_ver == "2.4":
- import md5 as hashlib
- else:
- import hashlib
- md5 = hashlib.md5(f.read()).hexdigest()
- f.close()
- return md5
- else:
- return 0
-
- def config(args,files):
- try:
- game = args["main_prefix"]
- url = "http://208.asktao.com/autoupdate/%s/config.ini" % game
- get = urllib.urlopen(url)
- aa = get.readlines()
- w = {}
- for i in aa:
- a = i.strip().split()[0]
- if "[" in a:
- x = a.strip("[]")
- w[x] = {}
- continue
- w[x][i.strip().split()[1].strip()] = i.strip().split()[0].strip()
- for key in w:
- if key == files:
- return w[key]
- except Exception,e:
- return 0
-
- class down_start():
-
- def work(self,args):
- aa = config(args,"path_md5")
- game = args["main_prefix"]
- if aa == 0:
- post(2,"read update config","Not find %s config file" % game)
- sys.exit()
- local = '/data/autoupdate/'
- url = "http://208.asktao.com/autoupdate/%s/" % game
- for f in aa:
- md5_r = f.strip().split()[0]
- pkg_name = f.strip().split()[1]
- get = urllib.urlopen(os.path.join(url,pkg_name))
- status = get.getcode()
- if status == 200:
- urllib.urlretrieve(os.path.join(url,pkg_name),os.path.join(local,pkg_name),)
- md5_l = md5sum(os.path.join(local,pkg_name))
- if md5_l == md5_r:
- post(1,"down",pkg_name)
- else:
- urllib.urlretrieve(os.path.join(url,pkg_name),os.path.join(local,pkg_name),)
- md5_l = md5sum(os.path.join(local,pkg_name))
- if md5_l == md5_r:
- post(1,"down",pkg_name)
- else:
- post(2,"down",pkg_name,"Download %s,MD5 not right" % pkg_name)
- sys.exit()
- else:
- post(2,"down",pkg_name,"Not find %s path file" % pkg_name)
-
- while True:
- pid = os.popen("ps auxww | grep 'rsync --daemon' | grep -v grep").read()
- if not pid:
- getso("/usr/bin/rsync --daemon")
- continue
- break
-
- post(1,"down","down_done")
-
- class sync_start():
-
- def rsync(self,update_pkg,args):
- node_ip = args["node_ip"]
- game = args["main_prefix"]
- os_type = platform.system()
- if os_type == "Windows":
- local = "C:\\update\\"
- elif os_type == "Linux":
- local = "/home/update/tmp/"
- if not os.path.isdir(local):
- os.mkdir(local)
- for f in os.listdir(local):
- if os.path.isfile(f):
- os.remove(os.path.join(local,f))
- elif os.path.isdir(f):
- shutil.rmtree(os.path.join(local,f))
- aa = config(args,"path")
- if aa == 0:
- post(2,"read update config","Not find %s config file" % game)
- sys.exit()
- for f in aa:
- if update_pkg in f:
- md5_r = aa[f]
- pkg_name = f
- if os_type == "Windows":
- sync = "rsync -avz %s::update/*%s* /cygdrive/c/update/" % (node_ip,update_pkg)
- elif os_type == "Linux":
- sync = "rsync -avz %s::update/*%s* /home/update/tmp/" % (node_ip,update_pkg)
- result = os.popen(sync).readlines()
- md5_l = md5sum(os.path.join(local,pkg_name))
- if md5_l == md5_r:
- post(1,"sync",pkg_name)
- else:
- if os_type == "Windows":
- sync = "rsync -avz %s::update/*%s* /cygdrive/c/update/" % (node_ip,update_pkg)
- elif os_type == "Linux":
- sync = "rsync -avz %s::update/*%s* /home/update/tmp/" % (node_ip,update_pkg)
- result = os.popen(sync).readlines()
- md5_l = md5sum(os.path.join(local,pkg_name))
- if md5_l == md5_r:
- post(1,"sync",pkg_name)
- else:
- post(2,"rsync",pkg_name,"sync Error,%s not find or file's md5 wrong" % pkg_name)
- sys.exit()
-
- def work(self,args):
- update_pkg = args["update_pkg"].split(',')
- for i in range(len(update_pkg)):
- self.rsync(update_pkg[i],args)
- post(1,"sync","sync_done")
-
- class update_start():
-
- def copy(self,src, dst):
- if os.path.isdir(src):
- base = os.path.basename(src)
- if os.path.exists(dst):
- dst = os.path.join(dst, base)
- if not os.path.exists(dst):
- os.makedirs(dst)
- names = os.listdir(src)
- for name in names:
- srcname = os.path.join(src, name)
- self.copy(srcname, dst)
- else:
- shutil.copy2(src, dst)
-
- def unrar(self,src,dst,args):
- os_type = platform.system()
- try:
- if not os.path.exists(dst) or not os.path.exists(src):
- raise Exception, "%s or %s not exist!" % (src, dst)
- if os_type == "Windows":
- os.system(r'C:\Progra~1\WinRAR\rar x -o+ -inul %s %s' % (src, dst))
- elif os_type == "Linux":
- if os.path.splitext(src)[1] == ".tgz":
- os.system("tar -zxf %s -C %s" % (src, dst))
- elif os.path.splitext(src)[1] == ".zip":
- os.system("unzip -oq %s -d %s" % (src, dst))
- return 0
- except Exception,e:
- return e
-
- def work(self,args):
- os_type = platform.system()
- if os_type == "Windows":
- update = "C:\\update\\"
- elif os_type == "Linux":
- update = "/home/update/tmp/"
- server_dir = { "tmcs":"C:\\Server\\","wd":"/home/asktao/"}
- server = server_dir[args["main_prefix"]]
- update_pkg = args["update_pkg"].split(',')
- game = args["main_prefix"]
- for i in range(len(update_pkg)):
-
- for tgz in os.listdir(update):
- if update_pkg[i] in tgz:
- src = os.path.join(update,tgz)
- r = self.unrar(src,update,args)
- if r == 0:
- post(1,"update","%s unzip success" % src)
- else:
- post(2,"update",src,"unzip fail:%s" % r)
-
- ser_list = os.listdir(server)
- up_list = os.listdir(update)
- for dir in up_list:
- for line in ser_list:
- filepath = os.path.join(update,dir+"\\")
- serv = os.path.join(server,line+"\\")
- if dir in line:
- self.copy(filepath,serv)
- post(1,"update","%s files copy success" % line)
-
- aa = config(args,"files")
- if aa == 0:
- post(2,"read update config","Not find %s config file" % game)
- sys.exit()
- for f in aa:
- md5_r = aa[f]
- pkg_name = f
- if os.path.exists(pkg_name):
- md5_s = md5sum(pkg_name)
- if md5_r == md5_s:
- post(1,"update","%s md5 Ok,update success" % pkg_name)
- else:
- post(2,"update","%s md5 Error,update fail" % pkg_name)
- sys.exit()
- shutil.rmtree(update)
- os.mkdir(update)
- post(1,"update","All_done")
-
- if __name__ == "__main__":
- 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"}
-
-
- sync = sync_start()
- sync.work(args)
- update = update_start()
- update.work(args)
-
-
- ''
-
-
-
-
-
-
-
本文转自 rickyfang 51CTO博客,原文链接:http://blog.51cto.com/wangwei007/957106
,如需转载请自行联系原作者