按功能拆分代码
This commit is contained in:
parent
d687e0426a
commit
effcc2874e
13 changed files with 382 additions and 341 deletions
347
launcher.py
347
launcher.py
|
|
@ -1,351 +1,18 @@
|
|||
import io
|
||||
import json
|
||||
import mimetypes
|
||||
import os
|
||||
import subprocess
|
||||
import threading
|
||||
import time
|
||||
from pathlib import Path
|
||||
from shutil import rmtree
|
||||
from subprocess import CREATE_NO_WINDOW, Popen
|
||||
|
||||
import chardet
|
||||
import py7zr
|
||||
import requests
|
||||
import webview
|
||||
from py7zr.callbacks import ExtractCallback
|
||||
|
||||
from launcher import config
|
||||
from launcher.sys_config import sys_config
|
||||
from log import logger
|
||||
from launcher.constants import upgrade_script_name
|
||||
from launcher.webview import start_webview
|
||||
|
||||
mimetypes.add_type("text/html", ".html")
|
||||
mimetypes.add_type("text/css", ".css")
|
||||
mimetypes.add_type("application/javascript", ".js")
|
||||
|
||||
version = "v0.3"
|
||||
if __name__ == '__main__':
|
||||
|
||||
get_new_version_url = (
|
||||
"https://git.zhaozuohong.vip/api/v1/repos/mower-ng/launcher/releases/latest"
|
||||
)
|
||||
# 如果当前路径存在更新脚本，则删除
|
||||
if Path(upgrade_script_name).exists():
|
||||
os.remove(upgrade_script_name)
|
||||
|
||||
download_git_url = "https://list.zhaozuohong.vip/mower-ng/git.7z"
|
||||
download_python_url = "https://list.zhaozuohong.vip/mower-ng/python.7z"
|
||||
|
||||
upgrade_script_name = "upgrade.bat"
|
||||
|
||||
|
||||
def custom_event(data):
|
||||
logger.info(data)
|
||||
data = json.dumps({"log": data + "\n"})
|
||||
js = f"var event = new CustomEvent('log', {{detail: {data}}}); window.dispatchEvent(event);"
|
||||
window.evaluate_js(js)
|
||||
|
||||
|
||||
def format_size(size_bytes):
|
||||
"""格式化文件大小为人类可读的形式"""
|
||||
for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
|
||||
if size_bytes < 1024:
|
||||
return f"{size_bytes:.2f} {unit}"
|
||||
size_bytes /= 1024
|
||||
return f"{size_bytes:.2f} PB"
|
||||
|
||||
|
||||
def download_file(download_name, download_url, destination_folder):
|
||||
"""
|
||||
下载文件到指定文件夹
|
||||
:param download_name: 下载内容名
|
||||
:param download_url: 文件的下载 URL
|
||||
:param destination_folder: 保存下载文件的目标文件夹
|
||||
:return: 下载是否成功
|
||||
"""
|
||||
if not os.path.exists(destination_folder):
|
||||
os.makedirs(destination_folder)
|
||||
|
||||
filename = os.path.basename(download_url)
|
||||
download_path = os.path.join(destination_folder, filename)
|
||||
custom_event(f"开始下载: {download_name}")
|
||||
|
||||
response = requests.get(download_url, stream=True)
|
||||
if response.status_code == 200:
|
||||
total_size = int(response.headers.get('content-length', 0))
|
||||
downloaded_size = 0
|
||||
start_time = time.time()
|
||||
last_update_time = time.time() # 记录上次更新时间
|
||||
block_size = 4096 # 每次读取的数据块大小
|
||||
|
||||
with open(download_path, "wb") as file:
|
||||
for data in response.iter_content(chunk_size=block_size):
|
||||
file.write(data)
|
||||
downloaded_size += len(data)
|
||||
current_time = time.time()
|
||||
elapsed_time = current_time - start_time
|
||||
|
||||
if elapsed_time > 0:
|
||||
download_speed = downloaded_size / elapsed_time # 字节/秒
|
||||
else:
|
||||
download_speed = 0
|
||||
|
||||
progress_percent = (downloaded_size / total_size) * 100 if total_size != 0 else 0
|
||||
|
||||
# 检查是否需要更新进度信息，每1秒更新一次
|
||||
if current_time - last_update_time >= 1:
|
||||
# 格式化输出
|
||||
formatted_downloaded_size = format_size(downloaded_size)
|
||||
formatted_total_size = format_size(total_size)
|
||||
formatted_speed = format_size(download_speed) + "/s"
|
||||
|
||||
custom_event(
|
||||
f"下载进度: {progress_percent:.2f}% ({formatted_downloaded_size}/{formatted_total_size}), 下载速度: {formatted_speed}")
|
||||
last_update_time = current_time # 更新上次更新时间
|
||||
|
||||
end_time = time.time()
|
||||
total_elapsed_time = end_time - start_time
|
||||
average_download_speed = downloaded_size / total_elapsed_time if total_elapsed_time != 0 else 0
|
||||
# 格式化输出
|
||||
formatted_total_elapsed_time = f"{total_elapsed_time:.2f} 秒"
|
||||
formatted_average_download_speed = format_size(average_download_speed) + "/s"
|
||||
|
||||
custom_event(
|
||||
f"下载完成: {filename}, 耗时: {formatted_total_elapsed_time}, 下载速度: {formatted_average_download_speed}")
|
||||
else:
|
||||
logger.error(f"下载失败: {response.status_code}")
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
class MyExtractCallback(ExtractCallback):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.total_size = 0
|
||||
self.last_print_time = time.time()
|
||||
|
||||
def report_start(self, archive_name, archive_format):
|
||||
pass
|
||||
|
||||
def report_start_preparation(self):
|
||||
pass
|
||||
|
||||
def report_end(self, processing_file_path, wrote_bytes):
|
||||
self.total_size += int(wrote_bytes)
|
||||
current_time = time.time()
|
||||
if current_time - self.last_print_time >= 1.0: # 至少每隔1秒输出一次
|
||||
custom_event(f"已解压: {format_size(self.total_size)}")
|
||||
self.last_print_time = current_time
|
||||
|
||||
def report_postprocess(self):
|
||||
pass
|
||||
|
||||
def report_update(self, message):
|
||||
pass
|
||||
|
||||
def report_warning(self, message):
|
||||
pass
|
||||
|
||||
|
||||
def extract_7z_file(file_name, file_path, destination_folder, delete_after_extract=True):
|
||||
"""
|
||||
解压7z文件到指定文件夹
|
||||
:param file_name: 7z文件的名称
|
||||
:param file_path: 7z文件的路径
|
||||
:param destination_folder: 解压目标文件夹
|
||||
:param delete_after_extract: 解压后删除压缩文件，默认删除
|
||||
:return: 解压是否成功
|
||||
"""
|
||||
if not os.path.exists(destination_folder):
|
||||
os.makedirs(destination_folder)
|
||||
|
||||
custom_event(f"开始解压文件: {file_name}")
|
||||
try:
|
||||
start_time = time.time()
|
||||
with py7zr.SevenZipFile(file_path, mode='r') as z:
|
||||
callback = MyExtractCallback()
|
||||
z.extractall(path=destination_folder, callback=callback)
|
||||
end_time = time.time()
|
||||
total_elapsed_time = end_time - start_time
|
||||
formatted_total_elapsed_time = f"{total_elapsed_time:.2f} 秒"
|
||||
custom_event(
|
||||
f"解压完成: {file_name}, 总大小: {format_size(callback.total_size)}, 耗时: {formatted_total_elapsed_time}")
|
||||
except Exception as e:
|
||||
e.print_exc()
|
||||
custom_event(f"解压失败: {str(e)}")
|
||||
return False
|
||||
|
||||
if delete_after_extract:
|
||||
try:
|
||||
os.remove(file_path)
|
||||
custom_event(f"删除{file_path}成功")
|
||||
except OSError as e:
|
||||
custom_event(f"删除{file_path}失败: {str(e)}")
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def init_download(download_name, download_url, destination_folder):
|
||||
"""
|
||||
初始化中的下载任务，返回一个函数，用于执行下载任务
|
||||
:param download_name: 下载任务的名称，用于在日志中记录
|
||||
:param download_url: 下载文件的 URL
|
||||
:param destination_folder: 下载文件的目标文件夹
|
||||
:return: 一个函数，用于执行下载任务
|
||||
"""
|
||||
|
||||
def download():
|
||||
target_folder = os.path.join(download_name)
|
||||
if os.path.exists(target_folder):
|
||||
custom_event(f"{download_name} 文件夹已存在，跳过下载")
|
||||
return True
|
||||
|
||||
filename = os.path.basename(download_url)
|
||||
if not download_file(filename, download_url, destination_folder):
|
||||
return False
|
||||
download_path = os.path.join(destination_folder, filename)
|
||||
|
||||
if not extract_7z_file(filename, download_path, destination_folder, True):
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
return download
|
||||
|
||||
|
||||
mirror_list = {
|
||||
"pypi": "https://pypi.org/simple",
|
||||
"aliyun": "https://mirrors.aliyun.com/pypi/simple/",
|
||||
"tuna": "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple",
|
||||
"sjtu": "https://mirror.sjtu.edu.cn/pypi/web/simple",
|
||||
}
|
||||
|
||||
command_list = {
|
||||
"download_git": lambda: init_download("git", download_git_url, os.getcwd()),
|
||||
"download_python": lambda: init_download("python", download_python_url, os.getcwd()),
|
||||
"lfs": "git\\bin\\git lfs install",
|
||||
"ensurepip": "python\\python -m ensurepip --default-pip",
|
||||
"clone": "git\\bin\\git -c lfs.concurrenttransfers=100 clone https://git.zhaozuohong.vip/mower-ng/mower-ng.git --branch slow",
|
||||
"fetch": lambda: f"..\\git\\bin\\git fetch origin {config.conf.branch} --progress",
|
||||
"switch": lambda: f"..\\git\\bin\\git -c lfs.concurrenttransfers=100 switch -f {config.conf.branch} --progress",
|
||||
"reset": lambda: f"..\\git\\bin\\git -c lfs.concurrenttransfers=200 reset --hard origin/{config.conf.branch}",
|
||||
"pip_install": lambda: f"..\\python\\Scripts\\pip install --no-cache-dir -i {mirror_list[config.conf.mirror]} -r requirements.txt --no-warn-script-location",
|
||||
"webview": "start ..\\python\\pythonw webview_ui.py",
|
||||
"manager": "start ..\\python\\pythonw manager.py",
|
||||
}
|
||||
|
||||
|
||||
def detect_encoding(stream):
|
||||
# 读取一部分数据来检测编码
|
||||
raw_data = stream.read(4096)
|
||||
result = chardet.detect(raw_data)
|
||||
stream.seek(0) # 将流指针重置到开头
|
||||
return result['encoding']
|
||||
|
||||
|
||||
def read_stream(stream, log_func):
|
||||
detected_encoding = detect_encoding(stream)
|
||||
text_io = io.TextIOWrapper(stream, encoding=detected_encoding, errors='replace')
|
||||
try:
|
||||
for line in iter(text_io.readline, ''):
|
||||
text = line.rstrip('\n')
|
||||
custom_event(text.strip() + "\n")
|
||||
finally:
|
||||
text_io.close()
|
||||
|
||||
|
||||
class Api:
|
||||
|
||||
def load_config(self):
|
||||
logger.info("读取配置文件")
|
||||
return config.conf.model_dump()
|
||||
|
||||
def save_config(self, conf):
|
||||
logger.info(f"更新配置文件{conf}")
|
||||
config.conf = config.Conf(**conf)
|
||||
config.save_conf()
|
||||
|
||||
def get_version(self):
|
||||
return version
|
||||
|
||||
def get_new_version(self):
|
||||
logger.info("获取最新版本号")
|
||||
response = requests.get(get_new_version_url)
|
||||
return response.json()
|
||||
|
||||
# 更新启动器本身
|
||||
def update_self(self, download_url):
|
||||
logger.info(f"开始更新启动器 {download_url}")
|
||||
# 下载压缩包的全路径
|
||||
file_name = os.path.basename(download_url)
|
||||
current_path = os.getcwd()
|
||||
if not download_file("launcher", download_url, current_path):
|
||||
return "下载新版本失败"
|
||||
|
||||
download_path = os.path.join(current_path, file_name)
|
||||
script_path = os.path.join(os.getcwd(), upgrade_script_name)
|
||||
folder_path = os.path.join(os.getcwd(), "_internal")
|
||||
exe_path = os.path.join(os.getcwd(), "launcher.exe")
|
||||
with open(script_path, "w") as b:
|
||||
temp_list = "@echo off\n"
|
||||
temp_list += "timeout /t 3 /nobreak\n" # 等待进程退出
|
||||
temp_list += f"rmdir {folder_path}\n" # 删除_internal
|
||||
temp_list += f"del {exe_path}\n" # 删除exe
|
||||
temp_list += f"tar -xf {download_path} -C ..\n" # 解压压缩包
|
||||
temp_list += "timeout /t 1 /nobreak\n" # 等待解压
|
||||
temp_list += f"start {exe_path}\n" # 启动新程序
|
||||
temp_list += f"del {download_path}\n" # 删除压缩包
|
||||
temp_list += "exit"
|
||||
b.write(temp_list)
|
||||
# 不显示cmd窗口
|
||||
Popen([script_path], creationflags=CREATE_NO_WINDOW)
|
||||
os._exit(0)
|
||||
|
||||
def rm_site_packages(self):
|
||||
site_packages_path = Path("./python/Lib/site-packages")
|
||||
if site_packages_path.exists():
|
||||
rmtree(site_packages_path)
|
||||
return "site-packages目录移除成功"
|
||||
return "python\\Lib\\site-packages目录不存在"
|
||||
|
||||
def rm_python_scripts(self):
|
||||
python_scripts_path = Path("./python/Scripts")
|
||||
if python_scripts_path.exists():
|
||||
rmtree(python_scripts_path)
|
||||
return "Scripts目录移除成功"
|
||||
return "python\\Scripts目录不存在"
|
||||
|
||||
def run(self, command, cwd=None):
|
||||
command = command_list[command]
|
||||
if callable(command):
|
||||
command = command()
|
||||
if callable(command):
|
||||
return "success" if command() else "failed"
|
||||
custom_event(command + "\n")
|
||||
try:
|
||||
with subprocess.Popen(
|
||||
command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, cwd=cwd, bufsize=0,
|
||||
universal_newlines=False
|
||||
) as p:
|
||||
stdout_thread = threading.Thread(target=read_stream, args=(p.stdout, logger.info))
|
||||
stderr_thread = threading.Thread(target=read_stream, args=(p.stderr, logger.info))
|
||||
|
||||
stdout_thread.start()
|
||||
stderr_thread.start()
|
||||
|
||||
stdout_thread.join()
|
||||
stderr_thread.join()
|
||||
if p.returncode == 0:
|
||||
return "success"
|
||||
except Exception as e:
|
||||
logger.error(f"command {command} 执行异常{e}")
|
||||
logger.exception(e)
|
||||
custom_event(str(e))
|
||||
return "failed"
|
||||
|
||||
|
||||
# 如果当前路径存在更新脚本，则删除
|
||||
if Path(upgrade_script_name).exists():
|
||||
os.remove(upgrade_script_name)
|
||||
|
||||
window = webview.create_window(
|
||||
f"mower-ng launcher {version}", sys_config.get('url'), js_api=Api()
|
||||
)
|
||||
webview.start()
|
||||
start_webview()
|
||||
|
|
|
|||
24
launcher/constants.py
Normal file
24
launcher/constants.py
Normal file
|
|
@ -0,0 +1,24 @@
|
|||
"""
|
||||
constants.py
|
||||
|
||||
该模块定义了应用程序中使用的各种常量。这些常量包括API URL和其他配置参数。
|
||||
"""
|
||||
|
||||
# 更新脚本名
|
||||
upgrade_script_name = "upgrade.bat"
|
||||
|
||||
# 获取最新版本发布信息
|
||||
get_new_version_url = "https://git.zhaozuohong.vip/api/v1/repos/mower-ng/launcher/releases/latest"
|
||||
|
||||
# 下载地址
|
||||
download_git_url = "https://list.zhaozuohong.vip/mower-ng/git.7z"
|
||||
download_python_url = "https://list.zhaozuohong.vip/mower-ng/python.7z"
|
||||
|
||||
# pip镜像地址
|
||||
mirror_list = {
|
||||
"pypi": "https://pypi.org/simple",
|
||||
"aliyun": "https://mirrors.aliyun.com/pypi/simple/",
|
||||
"tuna": "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple",
|
||||
"sjtu": "https://mirror.sjtu.edu.cn/pypi/web/simple",
|
||||
}
|
||||
|
||||
0
launcher/file/__init__.py
Normal file
0
launcher/file/__init__.py
Normal file
100
launcher/file/download.py
Normal file
100
launcher/file/download.py
Normal file
|
|
@ -0,0 +1,100 @@
|
|||
import os
|
||||
import time
|
||||
|
||||
import requests
|
||||
|
||||
from launcher.file.extract import extract_7z_file
|
||||
from launcher.file.utils import format_size
|
||||
from launcher.webview.events import custom_event
|
||||
|
||||
|
||||
def download_file(download_name, download_url, destination_folder):
|
||||
"""
|
||||
下载文件到指定文件夹
|
||||
:param download_name: 下载内容名
|
||||
:param download_url: 文件的下载 URL
|
||||
:param destination_folder: 保存下载文件的目标文件夹
|
||||
:return: 下载是否成功
|
||||
"""
|
||||
if not os.path.exists(destination_folder):
|
||||
os.makedirs(destination_folder)
|
||||
|
||||
filename = os.path.basename(download_url)
|
||||
download_path = os.path.join(destination_folder, filename)
|
||||
custom_event(f"开始下载: {download_name}")
|
||||
|
||||
response = requests.get(download_url, stream=True)
|
||||
if response.status_code == 200:
|
||||
total_size = int(response.headers.get('content-length', 0))
|
||||
downloaded_size = 0
|
||||
start_time = time.time()
|
||||
last_update_time = time.time() # 记录上次更新时间
|
||||
block_size = 4096 # 每次读取的数据块大小
|
||||
|
||||
with open(download_path, "wb") as file:
|
||||
for data in response.iter_content(chunk_size=block_size):
|
||||
file.write(data)
|
||||
downloaded_size += len(data)
|
||||
current_time = time.time()
|
||||
elapsed_time = current_time - start_time
|
||||
|
||||
if elapsed_time > 0:
|
||||
download_speed = downloaded_size / elapsed_time # 字节/秒
|
||||
else:
|
||||
download_speed = 0
|
||||
|
||||
progress_percent = (downloaded_size / total_size) * 100 if total_size != 0 else 0
|
||||
|
||||
# 检查是否需要更新进度信息，每1秒更新一次
|
||||
if current_time - last_update_time >= 1:
|
||||
# 格式化输出
|
||||
formatted_downloaded_size = format_size(downloaded_size)
|
||||
formatted_total_size = format_size(total_size)
|
||||
formatted_speed = format_size(download_speed) + "/s"
|
||||
|
||||
custom_event(
|
||||
f"下载进度: {progress_percent:.2f}% ({formatted_downloaded_size}/{formatted_total_size}), 下载速度: {formatted_speed}")
|
||||
last_update_time = current_time # 更新上次更新时间
|
||||
|
||||
end_time = time.time()
|
||||
total_elapsed_time = end_time - start_time
|
||||
average_download_speed = downloaded_size / total_elapsed_time if total_elapsed_time != 0 else 0
|
||||
# 格式化输出
|
||||
formatted_total_elapsed_time = f"{total_elapsed_time:.2f} 秒"
|
||||
formatted_average_download_speed = format_size(average_download_speed) + "/s"
|
||||
|
||||
custom_event(
|
||||
f"下载完成: {filename}, 耗时: {formatted_total_elapsed_time}, 下载速度: {formatted_average_download_speed}")
|
||||
else:
|
||||
custom_event(f"下载失败: {response.status_code}")
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def init_download(download_name, download_url, destination_folder):
|
||||
"""
|
||||
初始化中的下载任务，返回一个函数，用于执行下载任务
|
||||
:param download_name: 下载任务的名称，用于在日志中记录
|
||||
:param download_url: 下载文件的 URL
|
||||
:param destination_folder: 下载文件的目标文件夹
|
||||
:return: 一个函数，用于执行下载任务
|
||||
"""
|
||||
|
||||
def download():
|
||||
target_folder = os.path.join(download_name)
|
||||
if os.path.exists(target_folder):
|
||||
custom_event(f"{download_name} 文件夹已存在，跳过下载")
|
||||
return True
|
||||
|
||||
filename = os.path.basename(download_url)
|
||||
if not download_file(filename, download_url, destination_folder):
|
||||
return False
|
||||
download_path = os.path.join(destination_folder, filename)
|
||||
|
||||
if not extract_7z_file(filename, download_path, destination_folder, True):
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
return download
|
||||
75
launcher/file/extract.py
Normal file
75
launcher/file/extract.py
Normal file
|
|
@ -0,0 +1,75 @@
|
|||
import os
|
||||
import time
|
||||
|
||||
from py7zr import py7zr
|
||||
from py7zr.callbacks import ExtractCallback
|
||||
|
||||
from launcher.file.utils import format_size
|
||||
from launcher.webview.events import custom_event
|
||||
|
||||
|
||||
class MyExtractCallback(ExtractCallback):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.total_size = 0
|
||||
self.last_print_time = time.time()
|
||||
|
||||
def report_start(self, archive_name, archive_format):
|
||||
pass
|
||||
|
||||
def report_start_preparation(self):
|
||||
pass
|
||||
|
||||
def report_end(self, processing_file_path, wrote_bytes):
|
||||
self.total_size += int(wrote_bytes)
|
||||
current_time = time.time()
|
||||
if current_time - self.last_print_time >= 1.0: # 至少每隔1秒输出一次
|
||||
custom_event(f"已解压: {format_size(self.total_size)}")
|
||||
self.last_print_time = current_time
|
||||
|
||||
def report_postprocess(self):
|
||||
pass
|
||||
|
||||
def report_update(self, message):
|
||||
pass
|
||||
|
||||
def report_warning(self, message):
|
||||
pass
|
||||
|
||||
|
||||
def extract_7z_file(file_name, file_path, destination_folder, delete_after_extract=True):
|
||||
"""
|
||||
解压7z文件到指定文件夹
|
||||
:param file_name: 7z文件的名称
|
||||
:param file_path: 7z文件的路径
|
||||
:param destination_folder: 解压目标文件夹
|
||||
:param delete_after_extract: 解压后删除压缩文件，默认删除
|
||||
:return: 解压是否成功
|
||||
"""
|
||||
if not os.path.exists(destination_folder):
|
||||
os.makedirs(destination_folder)
|
||||
|
||||
custom_event(f"开始解压文件: {file_name}")
|
||||
try:
|
||||
start_time = time.time()
|
||||
with py7zr.SevenZipFile(file_path, mode='r') as z:
|
||||
callback = MyExtractCallback()
|
||||
z.extractall(path=destination_folder, callback=callback)
|
||||
end_time = time.time()
|
||||
total_elapsed_time = end_time - start_time
|
||||
formatted_total_elapsed_time = f"{total_elapsed_time:.2f} 秒"
|
||||
custom_event(
|
||||
f"解压完成: {file_name}, 总大小: {format_size(callback.total_size)}, 耗时: {formatted_total_elapsed_time}")
|
||||
except Exception as e:
|
||||
e.print_exc()
|
||||
custom_event(f"解压失败: {str(e)}")
|
||||
return False
|
||||
|
||||
if delete_after_extract:
|
||||
try:
|
||||
os.remove(file_path)
|
||||
custom_event(f"删除{file_path}成功")
|
||||
except OSError as e:
|
||||
custom_event(f"删除{file_path}失败: {str(e)}")
|
||||
|
||||
return True
|
||||
7
launcher/file/utils.py
Normal file
7
launcher/file/utils.py
Normal file
|
|
@ -0,0 +1,7 @@
|
|||
def format_size(size_bytes):
|
||||
"""格式化文件大小为人类可读的形式"""
|
||||
for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
|
||||
if size_bytes < 1024:
|
||||
return f"{size_bytes:.2f} {unit}"
|
||||
size_bytes /= 1024
|
||||
return f"{size_bytes:.2f} PB"
|
||||
|
|
@ -22,7 +22,7 @@ def setup_logger():
|
|||
|
||||
# 文件输出
|
||||
file_handler = RotatingFileHandler(
|
||||
"launcher.log", maxBytes=5 * 1024 * 1024, backupCount=3, encoding='utf-8'
|
||||
"../launcher.log", maxBytes=5 * 1024 * 1024, backupCount=3, encoding='utf-8'
|
||||
)
|
||||
file_handler.setLevel(logging.INFO)
|
||||
|
||||
|
|
@ -8,6 +8,8 @@ class SysConfig:
|
|||
"""
|
||||
读取系统配置文件
|
||||
"""
|
||||
# 版本
|
||||
version: str
|
||||
# ui路径
|
||||
url: str
|
||||
# 日志输出级别
|
||||
|
|
|
|||
|
|
@ -1,2 +1,3 @@
|
|||
version: "v0.3"
|
||||
url: "ui/dist/index.html"
|
||||
log_level: "ERROR"
|
||||
|
|
@ -1,2 +1,3 @@
|
|||
version: "dev"
|
||||
url: "http://localhost:5173/"
|
||||
log_level: "INFO"
|
||||
12
launcher/webview/__init__.py
Normal file
12
launcher/webview/__init__.py
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
import webview
|
||||
|
||||
from launcher.webview.api import Api
|
||||
from launcher.sys_config import sys_config
|
||||
|
||||
window = None
|
||||
|
||||
|
||||
def start_webview():
|
||||
global window
|
||||
window = webview.create_window(f"mower-ng launcher {sys_config.get('version')}", sys_config.get('url'), js_api=Api())
|
||||
webview.start()
|
||||
141
launcher/webview/api.py
Normal file
141
launcher/webview/api.py
Normal file
|
|
@ -0,0 +1,141 @@
|
|||
import io
|
||||
import os
|
||||
import subprocess
|
||||
import threading
|
||||
from _winapi import CREATE_NO_WINDOW
|
||||
from pathlib import Path
|
||||
from shutil import rmtree
|
||||
from subprocess import Popen
|
||||
|
||||
import chardet
|
||||
import requests
|
||||
|
||||
from launcher import config
|
||||
from launcher.constants import download_git_url, download_python_url, get_new_version_url, upgrade_script_name, \
|
||||
mirror_list
|
||||
from launcher.file.download import init_download, download_file
|
||||
from launcher.log import logger
|
||||
from launcher.sys_config import sys_config
|
||||
from launcher.webview.events import custom_event
|
||||
|
||||
command_list = {
|
||||
"download_git": lambda: init_download("git", download_git_url, os.getcwd()),
|
||||
"download_python": lambda: init_download("python", download_python_url, os.getcwd()),
|
||||
"lfs": "git\\bin\\git lfs install",
|
||||
"ensurepip": "python\\python -m ensurepip --default-pip",
|
||||
"clone": "git\\bin\\git -c lfs.concurrenttransfers=100 clone https://git.zhaozuohong.vip/mower-ng/mower-ng.git --branch slow",
|
||||
"fetch": lambda: f"..\\git\\bin\\git fetch origin {config.conf.branch} --progress",
|
||||
"switch": lambda: f"..\\git\\bin\\git -c lfs.concurrenttransfers=100 switch -f {config.conf.branch} --progress",
|
||||
"reset": lambda: f"..\\git\\bin\\git -c lfs.concurrenttransfers=200 reset --hard origin/{config.conf.branch}",
|
||||
"pip_install": lambda: f"..\\python\\Scripts\\pip install --no-cache-dir -i {mirror_list[config.conf.mirror]} -r requirements.txt --no-warn-script-location",
|
||||
"webview": "start ..\\python\\pythonw webview_ui.py",
|
||||
"manager": "start ..\\python\\pythonw manager.py",
|
||||
}
|
||||
|
||||
|
||||
def detect_encoding(stream):
|
||||
# 读取一部分数据来检测编码
|
||||
raw_data = stream.read(4096)
|
||||
result = chardet.detect(raw_data)
|
||||
stream.seek(0) # 将流指针重置到开头
|
||||
return result['encoding']
|
||||
|
||||
|
||||
def read_stream(stream, log_func):
|
||||
detected_encoding = detect_encoding(stream)
|
||||
text_io = io.TextIOWrapper(stream, encoding=detected_encoding, errors='replace')
|
||||
try:
|
||||
for line in iter(text_io.readline, ''):
|
||||
text = line.rstrip('\n')
|
||||
custom_event(text.strip() + "\n")
|
||||
finally:
|
||||
text_io.close()
|
||||
|
||||
|
||||
class Api:
|
||||
|
||||
def load_config(self):
|
||||
logger.info("读取配置文件")
|
||||
return config.conf.model_dump()
|
||||
|
||||
def save_config(self, conf):
|
||||
logger.info(f"更新配置文件{conf}")
|
||||
config.conf = config.Conf(**conf)
|
||||
config.save_conf()
|
||||
|
||||
def get_version(self):
|
||||
return sys_config.get("version")
|
||||
|
||||
def get_new_version(self):
|
||||
logger.info("获取最新版本号")
|
||||
response = requests.get(get_new_version_url)
|
||||
return response.json()
|
||||
|
||||
# 更新启动器本身
|
||||
def update_self(self, download_url):
|
||||
logger.info(f"开始更新启动器 {download_url}")
|
||||
# 下载压缩包的全路径
|
||||
file_name = os.path.basename(download_url)
|
||||
current_path = os.getcwd()
|
||||
if not download_file("launcher", download_url, current_path):
|
||||
return "下载新版本失败"
|
||||
|
||||
download_path = os.path.join(current_path, file_name)
|
||||
script_path = os.path.join(os.getcwd(), upgrade_script_name)
|
||||
folder_path = os.path.join(os.getcwd(), "_internal")
|
||||
exe_path = os.path.join(os.getcwd(), "launcher.exe")
|
||||
with open(script_path, "w") as b:
|
||||
temp_list = "@echo off\n"
|
||||
temp_list += "timeout /t 3 /nobreak\n" # 等待进程退出
|
||||
temp_list += f"rmdir {folder_path}\n" # 删除_internal
|
||||
temp_list += f"del {exe_path}\n" # 删除exe
|
||||
temp_list += f"tar -xf {download_path} -C ..\n" # 解压压缩包
|
||||
temp_list += "timeout /t 1 /nobreak\n" # 等待解压
|
||||
temp_list += f"start {exe_path}\n" # 启动新程序
|
||||
temp_list += f"del {download_path}\n" # 删除压缩包
|
||||
temp_list += "exit"
|
||||
b.write(temp_list)
|
||||
# 不显示cmd窗口
|
||||
Popen([script_path], creationflags=CREATE_NO_WINDOW)
|
||||
os._exit(0)
|
||||
|
||||
def rm_site_packages(self):
|
||||
site_packages_path = Path("./python/Lib/site-packages")
|
||||
if site_packages_path.exists():
|
||||
rmtree(site_packages_path)
|
||||
return "site-packages目录移除成功"
|
||||
return "python\\Lib\\site-packages目录不存在"
|
||||
|
||||
def rm_python_scripts(self):
|
||||
python_scripts_path = Path("./python/Scripts")
|
||||
if python_scripts_path.exists():
|
||||
rmtree(python_scripts_path)
|
||||
return "Scripts目录移除成功"
|
||||
return "python\\Scripts目录不存在"
|
||||
|
||||
def run(self, command, cwd=None):
|
||||
command = command_list[command]
|
||||
if callable(command):
|
||||
command = command()
|
||||
if callable(command):
|
||||
return "success" if command() else "failed"
|
||||
custom_event(command + "\n")
|
||||
try:
|
||||
with subprocess.Popen(
|
||||
command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, cwd=cwd, bufsize=0,
|
||||
universal_newlines=False
|
||||
) as p:
|
||||
stdout_thread = threading.Thread(target=read_stream, args=(p.stdout, logger.info))
|
||||
stderr_thread = threading.Thread(target=read_stream, args=(p.stderr, logger.info))
|
||||
|
||||
stdout_thread.start()
|
||||
stderr_thread.start()
|
||||
|
||||
stdout_thread.join()
|
||||
stderr_thread.join()
|
||||
if p.returncode == 0:
|
||||
return "success"
|
||||
except Exception as e:
|
||||
logger.exception(e)
|
||||
custom_event(str(e))
|
||||
return "failed"
|
||||
11
launcher/webview/events.py
Normal file
11
launcher/webview/events.py
Normal file
|
|
@ -0,0 +1,11 @@
|
|||
import json
|
||||
|
||||
import launcher
|
||||
from launcher.log import logger
|
||||
|
||||
|
||||
def custom_event(data):
|
||||
logger.info(data)
|
||||
data = json.dumps({"log": data + "\n"})
|
||||
js = f"var event = new CustomEvent('log', {{detail: {data}}}); window.dispatchEvent(event);"
|
||||
launcher.webview.window.evaluate_js(js)
|
||||
Loading…
Reference in a new issue