from qtlib import *
          from qtlib.download import download
          from config import add_config, conf
          from tools.progresser import Progresser
          
          import tarfile
          from distutils.dir_util import copy_tree
          SERVER = "http://project.googlecode.com/files/"
          
          def update(program, version):
              """具体更新的方法"""
              if not confirm(None, '开始准备更新, 是否继续?'):
                  return
          
              # 检查是否有更新
              if not check_update(program, version):
                  return
          
              # 复制program
              temp_path = '../%s.backup' % program
              shutil.rmtree(temp_path, True)
              shutil.copytree(os.path.join('..', program),
                              temp_path)
          
              # 下载程序
              tar_file = program + '.tar.gz'
              local_tar_file = os.path.join(temp_path, tar_file)
          
              with Progresser("下载中, 比较忙, 你可以离开位置休息一下..","退出",0,100, None) as p:
                  result = download(SERVER + tar_file,
                                    local_tar_file,
                                    stepper = p)
                  if result == False:
                      showMsg("无法连接到服务器, 请检查是否连上外部网络!")
                      return
                  elif result == None:
                      showMsg('取消下载, 不好意思, 下次还得从头开始下载..')
                      return
          
              showMsg('下载完成, 现在安装新程序.')
              os.chdir(temp_path)
              # 解压
              unzip(tar_file)
              # 标示一下是新程序
              with open('need_update', 'w+') as f:
                  f.write(program)
              # 执行新程序
              program_name = program+'.exe'
              os.execl(program_name, program_name)
          
          def check_update(program, version):
              """检查是否有更新"""
              # 下载versions文件
              if download(SERVER+'versions', 'temp/versions') == False:
                  showMsg("无法连接到服务器, 请检查是否连上外部网络!")
                  return False
          
              # 读取里面的信息
              data = open('temp/versions').read()
              # 格式是: program + 空格 + 版本号
              for line in data.split('\n'):
                  if line.startswith(program):
                      new_version = line.split(' ')[1]
                      # 把版本号当作浮点来检查
                      if float(version) < float(new_version):
                          return True
                      else:
                          showMsg("程序已经是最新版本: %s, %s" % (program, version))
          
          
          def check_finish_update():
              # 检查当前是否是需要更新的临时文件
              if not os.path.isfile('need_update'):
                  return
              # 读取程序名称
              with open('need_update') as f:
                  program = f.read()
              os.remove('need_update')
              # 然后把新程序复制回去
              old_path = '../%s'%program
              copy_tree(program, old_path)
              # 执行程序
              os.chdir(old_path)
              program_name = program + '.exe'
              os.execl(program_name, program_name)
          
          # 加载update.py模块的时候, 判断当前是否为在backup过程中执行的..
          check_finish_update()