#!/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)  
 
 
[files]#需要验证MD5的重要文件绝对路径和MD5  
15a6605156e29f68fdfd637e73a889d4  C:\Server\Line1\SAFlashPlayer.exe  
15a6605156e29f68fdfd637e73a889d4  C:\Server\Line2\SAFlashPlayer.exe  
[path]#更新包名字和MD5  
32fcb8932799aa553db13b5b9b41e5e9  auto.rar  
32fcb8932799aa553db13b5b9b41e5e9  update.rar