pyQt5 应用程序更新

主程序启动的时候检查是否有更新,用服务端最新版本号与程序中的版本号做对比。

如果有新版本的话,关闭自身,调起更新程序。

更新程序下载跟新包到本地,覆盖本地文件完成更新。

# _*_  coding: utf-8  _*_
import os
import sys
import subprocess
from PyQt5.QtCore import (QThread, pyqtSignal, QRect)
from PyQt5.QtWidgets import (QApplication, QDesktopWidget, QProgressBar, QWidget, QLabel, QFrame, QPushButton)
from PyQt5.QtGui import (QIcon, QPixmap)
import requests
import zipfile
import json

class dfThread(QThread):
    _signal = pyqtSignal(int, str, str)

    def __init__(self, download_url, filesize, fileobj, buffer):
        super().__init__()
        self.download_url = download_url
        self.filesize = filesize
        self.fileobj = fileobj
        self.buffer = buffer

    def run(self):
        try:
            f = requests.get(self.download_url, stream=True)
            offset = 0
            for chunk in f.iter_content(chunk_size=self.buffer):
                if not chunk:
                    break
                self.fileobj.seek(offset)
                self.fileobj.write(chunk)
                offset = offset + len(chunk)
                proess = offset / int(self.filesize) * 100
                self._signal.emit(int(proess), str(offset), str(self.filesize))
            self.fileobj.close()
            self.exit(0)
        except Exception as e:
            print(e)

class zipFile(object):
    def __init__(self, filename, mode='r', basedir=''):
        self.filename = filename
        self.mode = mode
        if self.mode in ('w', 'a'):
            self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)
        else:
            self.zfile = zipfile.ZipFile(filename, self.mode)
        self.basedir = basedir
        if not self.basedir:
            self.basedir = os.path.dirname(filename)

    def close(self):
        self.zfile.close()

    def extract_to(self, path):
        for p in self.zfile.namelist():
            self.extract(p, path)

    def extract(self, filename, path):
        if not filename.endswith('/'):
            f = os.path.join(path, filename)
            dir = os.path.dirname(f)
            if not os.path.exists(dir):
                os.makedirs(dir)
            try:
                open(f, 'wb').write(self.zfile.read(filename))
            except IOError:
                pass

class updateFrm(QWidget):
    def __init__(self):
        super().__init__()
        self.mainfn  = 'QQ'
        #self.fileuri = 'http://a.b.cn/serv/update/%s.zip' % self.mainfn
        self.fileuri = 'https://dldir1.qq.com/qqfile/qq/PCQQ9.1.1/24953/QQ9.1.1.24953.exe'
        #self.servurl = 'http://a.b.cn/serv/version'
        self.baseTitle = 'QQ升级助手'
        self.initUI()

    def initUI(self):
        self.info = QLabel(self)
        self.info.move(65, 28)
        self.info.resize(360, 40)

        map = QLabel(self)
        map.move(30, 25)
        img = QPixmap('res\\info.png')
        map.setPixmap(img)

        self.tips = QLabel(self)
        self.tips.move(420, 55)
        self.tips.resize(30, 10)
        self.tips.hide()

        self.line = QFrame(self)
        self.line.setGeometry(QRect(0, 75, 500, 75))
        self.line.setFrameShape(QFrame.HLine)
        self.line.setFrameShadow(QFrame.Sunken)
        self.line.setObjectName("line")

        self.btnQuit = QPushButton(self)
        self.btnQuit.setText("取消")
        self.btnQuit.setGeometry(380, 120, 100, 30)
        self.btnQuit.clicked.connect(self.cancel)
        self.btnQuit.hide()

        self.btnStart = QPushButton(self)
        self.btnStart.setText('立即更新')
        self.btnStart.setGeometry(380, 120, 100, 30)
        self.btnStart.clicked.connect(self.start)


        self.bar = QProgressBar(self)
        self.bar.setGeometry(0, 110, 530, 2)
        self.bar.setMaximum(100)
        self.bar.hide()

        self.setFixedSize(500, 160)
        self.center()
        self.setWindowTitle(self.baseTitle)
        icon = QIcon()
        icon.addPixmap(QPixmap("res\\6.ico"), QIcon.Normal, QIcon.Off)
        self.setWindowIcon(icon)

        try:
            #res = requests.get(self.servurl)
            #obj = json.loads(res.text)
            obj = {
                'title': '发现新版本',
                'desc' : 'PCQQ 9.1.1 全新改版,给你焕然一新的感觉'
            }
            txt = '%s\n\n%s' % (obj['title'], obj['desc'])
        except Exception as e:
            txt = '未发现新版本\n\n'
            self.btnStart.setDisabled(True)
        self.info.setText(txt)

        self.show()

    def center(self):
        qr = self.frameGeometry()
        cp = QDesktopWidget().availableGeometry().center()
        qr.moveCenter(cp)
        self.move(qr.topLeft())

    def cancel(self):
        self.close()

    def start(self):
        self.btnStart.hide()
        self.bar.show()
        self.tips.show()
        self.line.hide()
        self.btnQuit.setEnabled(False)
        self.btnQuit.show()
        self.downloadfile()

    def downloadfile(self):
        self.bar.setValue(0)
        path = '%s.zip' % self.mainfn
        self.filesize = requests.get(self.fileuri, stream=True).headers['Content-Length']
        self.fileobj  = open(path, 'wb')
        self.dfThread = dfThread(self.fileuri, self.filesize, self.fileobj, buffer=10240)
        self.dfThread._signal.connect(self.progressinfo)
        self.dfThread.start()

    def progressinfo(self, per, f_size_dl, f_size):
        self.bar.setValue(per)
        self.tips.setText('%s%%' % str(per))
        self.info.setText('升级过程中请勿退出\n\n正在下载.. %s/%s KB' % (str(f_size_dl), str(f_size)))
        if per == 100:
            self.info.setText('升级过程中请勿退出\n\n下载完成,正在安装..')
            self.unzip('%s.zip' % self.mainfn)
            self.tips.hide()
            self.info.setText('安装完成\n\n已升级到最新版本')
            self.btnQuit.setText('完成')
            self.btnQuit.setEnabled(True)

    def unzip(self, fname):
        z = zipFile(fname)
        z.extract_to('.')
        z.close()
        os.remove(fname)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    frm = updateFrm()
    #kill = subprocess.call(("taskkill /F /IM %s.exe" % frm.mainfn), shell=True)
    sys.exit(app.exec_())

猜你喜欢

转载自www.cnblogs.com/6min/p/10775977.html