207 lines
6.6 KiB
Python
207 lines
6.6 KiB
Python
# <AUTO_MAA:A MAA Multi Account Management and Automation Tool>
|
||
# Copyright © <2024> <DLmaster361>
|
||
|
||
# This file is part of AUTO_MAA.
|
||
|
||
# AUTO_MAA is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published
|
||
# by the Free Software Foundation, either version 3 of the License,
|
||
# or (at your option) any later version.
|
||
|
||
# AUTO_MAA is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty
|
||
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
|
||
# the GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
# DLmaster_361@163.com
|
||
|
||
"""
|
||
AUTO_MAA
|
||
AUTO_MAA更新器
|
||
v1.0
|
||
作者：DLmaster_361
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
import zipfile
|
||
import requests
|
||
import subprocess
|
||
|
||
from PySide6.QtWidgets import (
|
||
QApplication,
|
||
QDialog,
|
||
QVBoxLayout,
|
||
QLabel,
|
||
QProgressBar,
|
||
)
|
||
from PySide6.QtGui import QIcon
|
||
from PySide6.QtCore import QObject, QThread, Signal
|
||
|
||
|
||
class UpdateProcess(QThread):
|
||
|
||
info = Signal(str)
|
||
progress = Signal(int, int, int)
|
||
accomplish = Signal()
|
||
|
||
def __init__(self, app_path, name, download_url, version):
|
||
super(UpdateProcess, self).__init__()
|
||
|
||
self.app_path = app_path
|
||
self.name = name
|
||
self.download_url = download_url
|
||
self.version = version
|
||
self.download_path = f"{app_path}/AUTO_MAA_Update.zip" # 临时下载文件的路径
|
||
self.version_path = f"{app_path}/res/version.json"
|
||
|
||
def run(self):
|
||
|
||
# 清理可能存在的临时文件
|
||
try:
|
||
os.remove(self.download_path)
|
||
except FileNotFoundError:
|
||
pass
|
||
|
||
# 下载
|
||
try:
|
||
response = requests.get(self.download_url, stream=True)
|
||
file_size = response.headers.get("Content-Length")
|
||
if file_size is None:
|
||
file_size = 1
|
||
else:
|
||
file_size = int(file_size)
|
||
with open(self.download_path, "wb") as f:
|
||
downloaded_size = 0
|
||
for chunk in response.iter_content(chunk_size=8192):
|
||
f.write(chunk)
|
||
downloaded_size += len(chunk)
|
||
self.info.emit(
|
||
f"正在下载：{self.name} 已下载: {downloaded_size / 1048576:.2f}/{file_size / 1048576:.2f} MB ({downloaded_size / file_size * 100:.2f}%)"
|
||
)
|
||
self.progress.emit(0, 100, int(downloaded_size / file_size * 100))
|
||
except Exception as e:
|
||
e = str(e)
|
||
e = "\n".join([e[_ : _ + 75] for _ in range(0, len(e), 75)])
|
||
self.info.emit(f"下载{self.name}时出错：\n{e}")
|
||
return None
|
||
|
||
# 解压
|
||
try:
|
||
self.info.emit("正在解压更新文件")
|
||
self.progress.emit(0, 0, 0)
|
||
with zipfile.ZipFile(self.download_path, "r") as zip_ref:
|
||
zip_ref.extractall(self.app_path)
|
||
|
||
self.info.emit("正在删除临时文件")
|
||
self.progress.emit(0, 0, 0)
|
||
os.remove(self.download_path)
|
||
|
||
self.info.emit(f"{self.name}更新成功！")
|
||
self.progress.emit(0, 100, 100)
|
||
except Exception as e:
|
||
e = str(e)
|
||
e = "\n".join([e[_ : _ + 75] for _ in range(0, len(e), 75)])
|
||
self.info.emit(f"解压更新时出错：\n{e}")
|
||
return None
|
||
|
||
# 更新version文件
|
||
with open(self.version_path, "r", encoding="utf-8") as f:
|
||
version_info = json.load(f)
|
||
if self.name == "AUTO_MAA更新器":
|
||
version_info["updater_version"] = self.version
|
||
elif self.name == "AUTO_MAA主程序":
|
||
version_info["main_version"] = self.version
|
||
with open(self.version_path, "w", encoding="utf-8") as f:
|
||
json.dump(version_info, f, indent=4)
|
||
|
||
# 主程序更新完成后打开AUTO_MAA
|
||
if self.name == "AUTO_MAA主程序":
|
||
subprocess.Popen(
|
||
f"{self.app_path}/AUTO_MAA.exe",
|
||
shell=True,
|
||
creationflags=subprocess.CREATE_NO_WINDOW,
|
||
)
|
||
|
||
self.accomplish.emit()
|
||
|
||
|
||
class Updater(QObject):
|
||
|
||
def __init__(self, app_path, name, download_url, version):
|
||
super().__init__()
|
||
|
||
self.ui = QDialog()
|
||
self.ui.setWindowTitle("AUTO_MAA更新器")
|
||
self.ui.resize(500, 70)
|
||
self.ui.setWindowIcon(QIcon(f"{app_path}/gui/ico/AUTO_MAA_Updater.ico"))
|
||
|
||
# 创建垂直布局
|
||
self.Layout_v = QVBoxLayout(self.ui)
|
||
|
||
self.info = QLabel("正在初始化", self.ui)
|
||
self.Layout_v.addWidget(self.info)
|
||
|
||
self.progress = QProgressBar(self.ui)
|
||
self.progress.setRange(0, 0)
|
||
self.Layout_v.addWidget(self.progress)
|
||
|
||
self.update_process = UpdateProcess(app_path, name, download_url, version)
|
||
|
||
self.update_process.info.connect(self.update_info)
|
||
self.update_process.progress.connect(self.update_progress)
|
||
|
||
self.update_process.start()
|
||
|
||
def update_info(self, text):
|
||
self.info.setText(text)
|
||
|
||
def update_progress(self, begin, end, current):
|
||
self.progress.setRange(begin, end)
|
||
self.progress.setValue(current)
|
||
|
||
|
||
class AUTO_MAA_Updater(QApplication):
|
||
def __init__(self, app_path, name, download_url, version):
|
||
super().__init__()
|
||
|
||
self.main = Updater(app_path, name, download_url, version)
|
||
self.main.ui.show()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
|
||
# 获取软件自身的路径
|
||
app_path = os.path.dirname(os.path.realpath(sys.argv[0])).replace("\\", "/")
|
||
|
||
# 从本地版本信息文件获取当前版本信息
|
||
if os.path.exists(f"{app_path}/res/version.json"):
|
||
with open(f"{app_path}/res/version.json", "r", encoding="utf-8") as f:
|
||
version_current = json.load(f)
|
||
main_version_current = list(
|
||
map(int, version_current["main_version"].split("."))
|
||
)
|
||
else:
|
||
main_version_current = [0, 0, 0, 0]
|
||
|
||
# 从远程服务器获取最新版本信息
|
||
response = requests.get(
|
||
"https://ghp.ci/https://github.com/DLmaster361/AUTO_MAA/blob/main/res/version.json"
|
||
)
|
||
version_remote = response.json()
|
||
main_version_remote = list(map(int, version_remote["main_version"].split(".")))
|
||
|
||
# 启动更新线程
|
||
if main_version_remote > main_version_current:
|
||
app = AUTO_MAA_Updater(
|
||
app_path,
|
||
"AUTO_MAA主程序",
|
||
version_remote["main_download_url"],
|
||
version_remote["main_version"],
|
||
)
|
||
sys.exit(app.exec())
|