217 lines
7.3 KiB
Python
217 lines
7.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
版本检查器
|
|||
|
|
定期检查后端是否有新版本，并通知GUI更新
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import threading
|
|||
|
|
import time
|
|||
|
|
import logging
|
|||
|
|
from typing import Callable, Optional
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class VersionChecker:
|
|||
|
|
"""
|
|||
|
|
版本检查器
|
|||
|
|
定期向后端发送版本检查请求，如果有新版本则触发更新回调
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self, backend_client, update_callback: Optional[Callable] = None):
|
|||
|
|
"""
|
|||
|
|
初始化版本检查器
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
backend_client: 后端WebSocket客户端实例
|
|||
|
|
update_callback: 发现新版本时的回调函数 callback(latest_version, download_url)
|
|||
|
|
"""
|
|||
|
|
self.backend_client = backend_client
|
|||
|
|
self.update_callback = update_callback
|
|||
|
|
self.is_running = False
|
|||
|
|
self.check_interval = 600 # 10分钟检查一次
|
|||
|
|
self.thread = None
|
|||
|
|
|
|||
|
|
# 从config获取当前版本
|
|||
|
|
try:
|
|||
|
|
from config import APP_VERSION
|
|||
|
|
self.current_version = APP_VERSION
|
|||
|
|
logger.info(f"✅ 版本检查器初始化成功，当前版本: v{self.current_version}")
|
|||
|
|
except ImportError:
|
|||
|
|
self.current_version = "1.0.0"
|
|||
|
|
logger.warning("⚠️ 无法从config读取APP_VERSION，使用默认版本: 1.0.0")
|
|||
|
|
|
|||
|
|
def start(self):
|
|||
|
|
"""启动版本检查线程"""
|
|||
|
|
if self.is_running:
|
|||
|
|
logger.warning("版本检查器已在运行中")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
self.is_running = True
|
|||
|
|
self.thread = threading.Thread(target=self._check_loop, daemon=True)
|
|||
|
|
self.thread.start()
|
|||
|
|
logger.info(f"✅ 版本检查器已启动，每{self.check_interval}秒检查一次更新")
|
|||
|
|
|
|||
|
|
def stop(self):
|
|||
|
|
"""停止版本检查"""
|
|||
|
|
self.is_running = False
|
|||
|
|
if self.thread:
|
|||
|
|
self.thread.join(timeout=2)
|
|||
|
|
logger.info("版本检查器已停止")
|
|||
|
|
|
|||
|
|
def _check_loop(self):
|
|||
|
|
"""版本检查循环"""
|
|||
|
|
# 等待5秒后首次检查（避免启动时立即检查）
|
|||
|
|
time.sleep(5)
|
|||
|
|
|
|||
|
|
while self.is_running:
|
|||
|
|
try:
|
|||
|
|
self._perform_version_check()
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"版本检查失败: {e}")
|
|||
|
|
|
|||
|
|
# 等待下次检查
|
|||
|
|
time.sleep(self.check_interval)
|
|||
|
|
|
|||
|
|
def _perform_version_check(self):
|
|||
|
|
"""执行版本检查"""
|
|||
|
|
try:
|
|||
|
|
if not self.backend_client or not self.backend_client.is_connected:
|
|||
|
|
logger.debug("后端未连接，跳过版本检查")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 向后端发送版本检查请求
|
|||
|
|
message = {
|
|||
|
|
'type': 'version_check',
|
|||
|
|
'current_version': self.current_version,
|
|||
|
|
'client_type': 'gui'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logger.debug(f"发送版本检查请求: v{self.current_version}")
|
|||
|
|
self.backend_client.send_message(message)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"发送版本检查请求失败: {e}")
|
|||
|
|
|
|||
|
|
def handle_version_response(self, response: dict):
|
|||
|
|
"""
|
|||
|
|
处理后端返回的版本检查响应
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
response: 后端响应消息
|
|||
|
|
{
|
|||
|
|
"type": "version_response",
|
|||
|
|
"has_update": true,
|
|||
|
|
"latest_version": "1.5.0",
|
|||
|
|
"current_version": "1.4.7",
|
|||
|
|
"download_url": "https://example.com/download/v1.5.0",
|
|||
|
|
"update_content": "修复了若干Bug"
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
has_update = response.get('has_update', None) # 改为 None，便于后续判断
|
|||
|
|
latest_version = response.get('latest_version', '')
|
|||
|
|
current_version = response.get('current_version', self.current_version)
|
|||
|
|
download_url = response.get('download_url', '')
|
|||
|
|
update_content = response.get('update_content', '')
|
|||
|
|
|
|||
|
|
logger.info(f"📡 收到版本检查响应: 当前v{current_version}, 最新v{latest_version}")
|
|||
|
|
|
|||
|
|
# 如果后端没有返回 has_update 字段，自动通过版本比较判断
|
|||
|
|
if has_update is None and latest_version:
|
|||
|
|
# 自动比较版本号
|
|||
|
|
compare_result = self.compare_versions(latest_version, current_version)
|
|||
|
|
has_update = compare_result > 0 # latest > current 表示有更新
|
|||
|
|
logger.info(f"🔍 后端未返回has_update字段，通过版本比较判断: {has_update}")
|
|||
|
|
|
|||
|
|
if has_update and latest_version:
|
|||
|
|
logger.info(f"🔔 发现新版本: v{latest_version}")
|
|||
|
|
if update_content:
|
|||
|
|
logger.info(f"📝 更新内容: {update_content[:100]}")
|
|||
|
|
|
|||
|
|
# 触发更新回调
|
|||
|
|
if self.update_callback:
|
|||
|
|
try:
|
|||
|
|
self.update_callback(latest_version, download_url)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"执行更新回调失败: {e}")
|
|||
|
|
else:
|
|||
|
|
logger.debug(f"✅ 当前已是最新版本: v{current_version}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"处理版本响应失败: {e}")
|
|||
|
|
|
|||
|
|
def compare_versions(self, version1: str, version2: str) -> int:
|
|||
|
|
"""
|
|||
|
|
比较两个版本号
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
version1: 版本1 (如 "1.4.7")
|
|||
|
|
version2: 版本2 (如 "1.5.0")
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
1: version1 > version2
|
|||
|
|
0: version1 == version2
|
|||
|
|
-1: version1 < version2
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
v1_parts = [int(x) for x in version1.split('.')]
|
|||
|
|
v2_parts = [int(x) for x in version2.split('.')]
|
|||
|
|
|
|||
|
|
# 补齐长度
|
|||
|
|
max_len = max(len(v1_parts), len(v2_parts))
|
|||
|
|
v1_parts.extend([0] * (max_len - len(v1_parts)))
|
|||
|
|
v2_parts.extend([0] * (max_len - len(v2_parts)))
|
|||
|
|
|
|||
|
|
# 逐位比较
|
|||
|
|
for v1, v2 in zip(v1_parts, v2_parts):
|
|||
|
|
if v1 > v2:
|
|||
|
|
return 1
|
|||
|
|
elif v1 < v2:
|
|||
|
|
return -1
|
|||
|
|
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"版本比较失败: {e}")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 使用示例
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# 配置日志
|
|||
|
|
logging.basicConfig(
|
|||
|
|
level=logging.DEBUG,
|
|||
|
|
format='%(asctime)s [%(levelname)s] %(message)s'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 模拟使用
|
|||
|
|
class MockBackendClient:
|
|||
|
|
is_connected = True
|
|||
|
|
def send_message(self, msg):
|
|||
|
|
print(f"发送消息: {msg}")
|
|||
|
|
|
|||
|
|
def on_update_available(version, url):
|
|||
|
|
print(f"🔔 发现新版本: {version}, 下载地址: {url}")
|
|||
|
|
|
|||
|
|
client = MockBackendClient()
|
|||
|
|
checker = VersionChecker(client, on_update_available)
|
|||
|
|
checker.start()
|
|||
|
|
|
|||
|
|
# 模拟接收后端响应
|
|||
|
|
time.sleep(2)
|
|||
|
|
response = {
|
|||
|
|
"type": "version_response",
|
|||
|
|
"has_update": True,
|
|||
|
|
"latest_version": "1.5.0",
|
|||
|
|
"current_version": "1.4.7",
|
|||
|
|
"download_url": "https://example.com/download",
|
|||
|
|
"update_content": "修复了若干Bug"
|
|||
|
|
}
|
|||
|
|
checker.handle_version_response(response)
|
|||
|
|
|
|||
|
|
time.sleep(2)
|
|||
|
|
checker.stop()
|
|||
|
|
|