python 游戏自动更新

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

#!/usr/bin/env python  
# -*- coding: utf-8 -*-  
#Used update game path to server  
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:#验证MD5,MD5错误的话重新下载一次,再次错误就提示失败,退出程序  
                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)  
        # 检查rsync服务是否启动  
        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)  
#####################验证重要文件MD5  
        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"}  
    #down = down_start()  
    #down.work(args)  
    sync = sync_start()  
    sync.work(args)  
    update = update_start()  
    update.work(args)  
 
 
'''''config.ini内容  
[files]#需要验证MD5的重要文件绝对路径和MD5  
15a6605156e29f68fdfd637e73a889d4  C:\Server\Line1\SAFlashPlayer.exe  
15a6605156e29f68fdfd637e73a889d4  C:\Server\Line2\SAFlashPlayer.exe  
[path]#更新包名字和MD5  
32fcb8932799aa553db13b5b9b41e5e9  auto.rar  
32fcb8932799aa553db13b5b9b41e5e9  update.rar  
''' 
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,033评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,725评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,473评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,846评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,848评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,691评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,053评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,700评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,856评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,676评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,787评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,430评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,034评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,990评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,218评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,174评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,526评论 2 343

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,598评论 18 139
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,464评论 25 707
  • 国家电网公司企业标准(Q/GDW)- 面向对象的用电信息数据交换协议 - 报批稿:20170802 前言: 排版 ...
    庭说阅读 10,865评论 6 13
  • 周末闲来无事,偶翻相册。那些年的江南游又历历可见。 江南好,风景旧曾谙;日出江花红胜火,春来江水绿如蓝。能不忆江南...
    jiangming888阅读 496评论 24 20
  • 年轻时看《红楼梦》,很讨厌里面有个人物王熙凤,感觉她阴险、狡诈和狠毒,把一个无知的‘贾瑞’几下就活活整死了,又利用...
    半亩方塘yx阅读 582评论 5 5