mirror of
https://github.com/goldenfishs/MRobot.git
synced 2025-07-04 22:44:16 +08:00
1746 lines
69 KiB
Python
1746 lines
69 KiB
Python
import sys
|
||
import os
|
||
import serial
|
||
import serial.tools.list_ports
|
||
|
||
from PyQt5.QtCore import Qt, pyqtSignal, QThread
|
||
from PyQt5.QtGui import QTextCursor
|
||
from PyQt5.QtWidgets import (
|
||
QApplication, QWidget, QVBoxLayout, QHBoxLayout,
|
||
QSizePolicy,
|
||
QFileDialog, QMessageBox, QStackedLayout
|
||
)
|
||
|
||
from qfluentwidgets import (
|
||
Theme, setTheme, FluentIcon, SwitchButton, BodyLabel, SubtitleLabel,TitleLabel,
|
||
StrongBodyLabel, HorizontalSeparator, InfoBar, MessageDialog, Dialog,
|
||
AvatarWidget, NavigationItemPosition, FluentWindow, NavigationAvatarWidget,
|
||
PushButton, TextEdit, LineEdit, ComboBox, ImageLabel,
|
||
)
|
||
from qfluentwidgets import FluentIcon as FIF
|
||
import requests
|
||
import shutil
|
||
from PyQt5.QtCore import Qt
|
||
from PyQt5.QtWidgets import QHeaderView
|
||
|
||
from qfluentwidgets import (
|
||
TreeWidget, InfoBar, InfoBarPosition,
|
||
)
|
||
from qfluentwidgets import CheckBox
|
||
from qfluentwidgets import TreeWidget
|
||
from PyQt5.QtCore import QThread, pyqtSignal
|
||
from PyQt5.QtWidgets import QFileDialog
|
||
from qfluentwidgets import ProgressBar
|
||
import zipfile
|
||
import io
|
||
import jinja2
|
||
from PyQt5.QtWidgets import QTreeWidgetItem as TreeItem
|
||
import yaml # 确保已安装 pyyaml
|
||
import requests
|
||
from PyQt5.QtCore import Qt, QTimer
|
||
from qfluentwidgets import (
|
||
SettingCardGroup, SettingCard, ExpandSettingCard, HyperlinkButton, PushButton,
|
||
SubtitleLabel, StrongBodyLabel, BodyLabel, HorizontalSeparator, FluentIcon, InfoBar, InfoBarPosition
|
||
)
|
||
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout
|
||
|
||
from qfluentwidgets import (
|
||
SettingCardGroup, ExpandSettingCard, SubtitleLabel, BodyLabel, HorizontalSeparator, FluentIcon, InfoBar
|
||
|
||
)
|
||
# 添加quote
|
||
from urllib.parse import quote
|
||
|
||
from packaging.version import parse as vparse
|
||
__version__ = "1.0.1"
|
||
|
||
# ===================== 页面基类 =====================
|
||
class BaseInterface(QWidget):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
|
||
# ===================== 启动界面 =====================
|
||
from qfluentwidgets import ImageLabel, TitleLabel, BodyLabel, ProgressBar, PushSettingCard, HyperlinkCard
|
||
from PyQt5.QtWidgets import QDialog, QVBoxLayout, QSpacerItem, QSizePolicy
|
||
from PyQt5.QtCore import Qt
|
||
# ...existing code...
|
||
|
||
from qfluentwidgets import isDarkTheme # 加入主题判断
|
||
|
||
class SplashScreen(QDialog):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
# 自动适配主题
|
||
dark = isDarkTheme()
|
||
bg_color = "#23272e" if dark else "#f7fafd"
|
||
text_color = "#e9f6ff" if dark else "#2d7d9a"
|
||
sub_color = "#b0b8c1" if dark else "#6b7b8c"
|
||
border_color = "#3a3f4b" if dark else "#e0e6ef"
|
||
|
||
self.setWindowFlags(Qt.FramelessWindowHint | Qt.Dialog)
|
||
self.setModal(True)
|
||
self.setFixedSize(420, 260)
|
||
self.setStyleSheet(f"""
|
||
QDialog {{
|
||
background: {bg_color};
|
||
border-radius: 18px;
|
||
border: 1px solid {border_color};
|
||
}}
|
||
""")
|
||
layout = QVBoxLayout(self)
|
||
layout.setContentsMargins(36, 36, 36, 36)
|
||
layout.setSpacing(18)
|
||
|
||
layout.addSpacerItem(QSpacerItem(20, 20, QSizePolicy.Minimum, QSizePolicy.Expanding))
|
||
|
||
# Logo
|
||
self.logo = ImageLabel('img/MRobot.png')
|
||
self.logo.setFixedSize(220, 56)
|
||
self.logo.setAlignment(Qt.AlignCenter)
|
||
layout.addWidget(self.logo, alignment=Qt.AlignCenter)
|
||
|
||
# 应用名
|
||
self.title = TitleLabel("MRobot Toolbox")
|
||
self.title.setAlignment(Qt.AlignCenter)
|
||
self.title.setStyleSheet(f"font-size: 26px; font-weight: bold; color: {text_color};")
|
||
layout.addWidget(self.title)
|
||
|
||
# 状态文本
|
||
self.status = BodyLabel("正在启动...")
|
||
self.status.setAlignment(Qt.AlignCenter)
|
||
self.status.setStyleSheet(f"font-size: 15px; color: {sub_color};")
|
||
layout.addWidget(self.status)
|
||
|
||
# 进度条
|
||
self.progress = ProgressBar()
|
||
self.progress.setRange(0, 100)
|
||
self.progress.setValue(0)
|
||
self.progress.setFixedHeight(10)
|
||
layout.addWidget(self.progress)
|
||
|
||
layout.addSpacerItem(QSpacerItem(20, 20, QSizePolicy.Minimum, QSizePolicy.Expanding))
|
||
|
||
def set_status(self, text, value=None):
|
||
self.status.setText(text)
|
||
if value is not None:
|
||
self.progress.setValue(value)
|
||
|
||
# ===================== 首页界面 =====================
|
||
class HomeInterface(BaseInterface):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("homeInterface")
|
||
|
||
# 外层居中布局
|
||
outer_layout = QVBoxLayout(self)
|
||
outer_layout.setContentsMargins(0, 0, 0, 0)
|
||
outer_layout.setSpacing(0)
|
||
outer_layout.addStretch()
|
||
|
||
# 直接用布局和控件，无卡片
|
||
content_layout = QVBoxLayout()
|
||
content_layout.setSpacing(24)
|
||
content_layout.setContentsMargins(48, 48, 48, 48)
|
||
|
||
# Logo
|
||
logo = ImageLabel('img/MRobot.png')
|
||
logo.setFixedSize(320, 80)
|
||
logo.setAlignment(Qt.AlignCenter)
|
||
content_layout.addWidget(logo, alignment=Qt.AlignHCenter)
|
||
|
||
content_layout.addSpacing(8)
|
||
content_layout.addStretch()
|
||
# 主标题
|
||
title = SubtitleLabel("MRobot Toolbox")
|
||
title.setAlignment(Qt.AlignCenter)
|
||
title.setStyleSheet("font-size: 32px; font-weight: bold;")
|
||
content_layout.addWidget(title)
|
||
|
||
# 副标题
|
||
subtitle = BodyLabel("现代化，多功能机器人开发工具箱")
|
||
subtitle.setAlignment(Qt.AlignCenter)
|
||
subtitle.setStyleSheet("font-size: 18px; color: #4a6fa5;")
|
||
content_layout.addWidget(subtitle)
|
||
|
||
# 欢迎语
|
||
welcome = BodyLabel("欢迎使用 MRobot Toolbox！一站式支持代码生成、硬件管理、串口调试与零件库下载。")
|
||
welcome.setAlignment(Qt.AlignCenter)
|
||
welcome.setStyleSheet("font-size: 15px;")
|
||
content_layout.addWidget(welcome)
|
||
|
||
content_layout.addSpacing(16)
|
||
content_layout.addStretch()
|
||
|
||
# 直接加到主布局
|
||
outer_layout.addLayout(content_layout)
|
||
outer_layout.addStretch()
|
||
|
||
# 版权信息置底
|
||
copyright_label = BodyLabel("© 2025 MRobot | Powered by QUT RM&RCer")
|
||
copyright_label.setAlignment(Qt.AlignCenter)
|
||
copyright_label.setStyleSheet("font-size: 13px;")
|
||
outer_layout.addWidget(copyright_label)
|
||
outer_layout.addSpacing(18)
|
||
|
||
# ===================== 代码生成页面 =====================
|
||
class IocConfig:
|
||
def __init__(self, ioc_path):
|
||
self.ioc_path = ioc_path
|
||
self.config = {}
|
||
self._parse()
|
||
|
||
def _parse(self):
|
||
with open(self.ioc_path, encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if not line or line.startswith('#'):
|
||
continue
|
||
if '=' in line:
|
||
key, value = line.split('=', 1)
|
||
self.config[key.strip()] = value.strip()
|
||
|
||
def is_freertos_enabled(self):
|
||
# 判断是否开启FreeRTOS
|
||
ip_keys = [k for k in self.config if k.startswith('Mcu.IP')]
|
||
for k in ip_keys:
|
||
if self.config[k] == 'FREERTOS':
|
||
return True
|
||
for k in self.config:
|
||
if k.startswith('FREERTOS.'):
|
||
return True
|
||
return False
|
||
|
||
# 可扩展：添加更多参数获取方法
|
||
def get_parameter(self, key, default=None):
|
||
return self.config.get(key, default)
|
||
|
||
def get_all_with_prefix(self, prefix):
|
||
return {k: v for k, v in self.config.items() if k.startswith(prefix)}
|
||
|
||
class DataInterface(BaseInterface):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("dataInterface")
|
||
|
||
# 属性初始化
|
||
self.project_path = ""
|
||
self.project_name = ""
|
||
self.ioc_file = ""
|
||
self.freertos_enabled = False # 新增属性
|
||
|
||
# 主布局
|
||
self.stacked_layout = QStackedLayout(self)
|
||
self.setLayout(self.stacked_layout)
|
||
|
||
# --- 页面1：工程路径选择 ---
|
||
self.select_widget = QWidget()
|
||
outer_layout = QVBoxLayout(self.select_widget)
|
||
outer_layout.setContentsMargins(0, 0, 0, 0)
|
||
outer_layout.addStretch()
|
||
|
||
# 直接用布局和控件，无卡片
|
||
content_layout = QVBoxLayout()
|
||
content_layout.setSpacing(28)
|
||
content_layout.setContentsMargins(48, 48, 48, 48)
|
||
|
||
# 主标题
|
||
title = TitleLabel("MRobot 代码生成")
|
||
title.setAlignment(Qt.AlignCenter)
|
||
title.setStyleSheet("font-size: 36px; font-weight: bold; color: #2d7d9a;")
|
||
content_layout.addWidget(title)
|
||
|
||
# 副标题
|
||
subtitle = BodyLabel("请选择您的由CUBEMX生成的工程路径（.ico所在的目录），然后开启代码之旅！")
|
||
subtitle.setAlignment(Qt.AlignCenter)
|
||
subtitle.setStyleSheet("font-size: 16px; color: #4a6fa5;")
|
||
content_layout.addWidget(subtitle)
|
||
|
||
# 简要说明
|
||
desc = BodyLabel("支持自动配置和生成任务，自主选择模块代码倒入，自动识别cubemx配置！")
|
||
desc.setAlignment(Qt.AlignCenter)
|
||
desc.setStyleSheet("font-size: 14px; color: #6b7b8c;")
|
||
content_layout.addWidget(desc)
|
||
|
||
content_layout.addSpacing(18)
|
||
|
||
# 选择项目路径按钮
|
||
self.choose_btn = PushButton(FluentIcon.FOLDER, "选择项目路径")
|
||
self.choose_btn.setFixedWidth(200)
|
||
self.choose_btn.setStyleSheet("font-size: 17px;")
|
||
self.choose_btn.clicked.connect(self.choose_project_folder)
|
||
content_layout.addWidget(self.choose_btn, alignment=Qt.AlignmentFlag.AlignCenter)
|
||
|
||
# 更新代码库按钮
|
||
self.update_template_btn = PushButton(FluentIcon.SYNC, "更新代码库")
|
||
self.update_template_btn.setFixedWidth(200)
|
||
self.update_template_btn.setStyleSheet("font-size: 17px;")
|
||
self.update_template_btn.clicked.connect(self.update_user_template)
|
||
content_layout.addWidget(self.update_template_btn, alignment=Qt.AlignmentFlag.AlignCenter)
|
||
|
||
content_layout.addSpacing(10)
|
||
content_layout.addStretch()
|
||
|
||
|
||
outer_layout.addLayout(content_layout)
|
||
outer_layout.addStretch()
|
||
|
||
self.stacked_layout.addWidget(self.select_widget)
|
||
|
||
|
||
# --- 页面2：主配置页面 ---
|
||
self.config_widget = QWidget()
|
||
main_layout = QVBoxLayout(self.config_widget)
|
||
main_layout.setContentsMargins(32, 32, 32, 32)
|
||
main_layout.setSpacing(18)
|
||
|
||
# 顶部项目信息
|
||
info_layout = QHBoxLayout()
|
||
self.back_btn = PushButton(FluentIcon.SKIP_BACK, "返回")
|
||
self.back_btn.setFixedWidth(90)
|
||
self.back_btn.clicked.connect(self.back_to_select)
|
||
info_layout.addWidget(self.back_btn) # 返回按钮放最左
|
||
self.project_name_label = StrongBodyLabel()
|
||
self.project_path_label = BodyLabel()
|
||
self.ioc_file_label = BodyLabel()
|
||
self.freertos_label = BodyLabel()
|
||
info_layout.addWidget(self.project_name_label)
|
||
info_layout.addWidget(self.project_path_label)
|
||
info_layout.addWidget(self.ioc_file_label)
|
||
info_layout.addWidget(self.freertos_label)
|
||
info_layout.addStretch()
|
||
main_layout.addLayout(info_layout)
|
||
main_layout.addWidget(HorizontalSeparator())
|
||
|
||
# ======= 新增：左右分栏 =======
|
||
content_hbox = QHBoxLayout()
|
||
content_hbox.setSpacing(24)
|
||
|
||
# 左侧：文件树
|
||
left_vbox = QVBoxLayout()
|
||
left_vbox.addWidget(SubtitleLabel("用户代码模块选择"))
|
||
left_vbox.addWidget(HorizontalSeparator())
|
||
self.file_tree = TreeWidget()
|
||
self.file_tree.setHeaderLabels(["模块名"])
|
||
self.file_tree.setSelectionMode(self.file_tree.ExtendedSelection)
|
||
self.file_tree.header().setSectionResizeMode(0, QHeaderView.Stretch)
|
||
self.file_tree.setCheckedColor("#0078d4", "#2d7d9a")
|
||
self.file_tree.setBorderRadius(8)
|
||
self.file_tree.setBorderVisible(True)
|
||
left_vbox.addWidget(self.file_tree, stretch=1)
|
||
content_hbox.addLayout(left_vbox, 2)
|
||
|
||
# 右侧：操作按钮和说明
|
||
right_vbox = QVBoxLayout()
|
||
right_vbox.setSpacing(18)
|
||
right_vbox.addWidget(SubtitleLabel("操作区"))
|
||
right_vbox.addWidget(HorizontalSeparator())
|
||
|
||
# 操作按钮分组
|
||
btn_group = QVBoxLayout()
|
||
# 自动环境配置按钮
|
||
self.env_btn = PushButton("自动环境配置")
|
||
self.env_btn.setFixedWidth(200)
|
||
self.env_btn.setToolTip("自动检测并配置常用开发环境（功能开发中）")
|
||
self.env_btn.clicked.connect(self.auto_env_config)
|
||
btn_group.addWidget(self.env_btn)
|
||
# FreeRTOS相关按钮
|
||
self.freertos_task_btn = PushButton("自动生成FreeRTOS任务")
|
||
self.freertos_task_btn.setFixedWidth(200)
|
||
self.freertos_task_btn.setToolTip("自动在 freertos.c 中插入任务创建代码")
|
||
self.freertos_task_btn.clicked.connect(self.on_freertos_task_btn_clicked)
|
||
btn_group.addWidget(self.freertos_task_btn)
|
||
self.task_code_btn = PushButton("配置并生成任务代码")
|
||
self.task_code_btn.setFixedWidth(200)
|
||
self.task_code_btn.setToolTip("配置任务参数并一键生成任务代码文件")
|
||
self.task_code_btn.clicked.connect(self.on_task_code_btn_clicked)
|
||
btn_group.addWidget(self.task_code_btn)
|
||
self.generate_btn = PushButton(FluentIcon.CODE, "生成代码")
|
||
self.generate_btn.setFixedWidth(200)
|
||
self.generate_btn.setToolTip("将选中的用户模块代码复制到工程 User 目录")
|
||
self.generate_btn.clicked.connect(self.generate_code)
|
||
btn_group.addWidget(self.generate_btn)
|
||
btn_group.addSpacing(10)
|
||
right_vbox.addLayout(btn_group)
|
||
right_vbox.addStretch()
|
||
|
||
content_hbox.addLayout(right_vbox, 1)
|
||
main_layout.addLayout(content_hbox, stretch=1)
|
||
self.stacked_layout.addWidget(self.config_widget)
|
||
self.file_tree.itemChanged.connect(self.on_tree_item_changed)
|
||
|
||
def auto_env_config(self):
|
||
InfoBar.info(
|
||
title="敬请期待",
|
||
content="自动环境配置功能暂未实现，等待后续更新。",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
|
||
def choose_project_folder(self):
|
||
folder = QFileDialog.getExistingDirectory(self, "请选择代码项目文件夹")
|
||
if not folder:
|
||
return
|
||
ioc_files = [f for f in os.listdir(folder) if f.endswith('.ioc')]
|
||
if not ioc_files:
|
||
InfoBar.warning(
|
||
title="提示",
|
||
content="未找到.ioc文件，请确认项目文件夹。",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
return
|
||
self.project_path = folder
|
||
self.project_name = os.path.basename(folder)
|
||
self.ioc_file = os.path.join(folder, ioc_files[0])
|
||
self.show_config_page()
|
||
|
||
def show_config_page(self):
|
||
# 更新项目信息
|
||
self.project_name_label.setText(f"项目名称: {self.project_name}")
|
||
self.project_path_label.setText(f"项目路径: {self.project_path}")
|
||
# self.ioc_file_label.setText(f"IOC 文件: {self.ioc_file}")
|
||
try:
|
||
ioc = IocConfig(self.ioc_file)
|
||
self.freertos_enabled = ioc.is_freertos_enabled() # 记录状态
|
||
freertos_status = "已启用" if self.freertos_enabled else "未启用"
|
||
self.freertos_label.setText(f"FreeRTOS: {freertos_status}")
|
||
# self.freertos_task_btn.setEnabled(self.freertos_enabled)
|
||
except Exception as e:
|
||
self.freertos_label.setText(f"IOC解析失败: {e}")
|
||
self.freertos_task_btn.hide()
|
||
self.freertos_enabled = False
|
||
self.show_user_code_files()
|
||
self.stacked_layout.setCurrentWidget(self.config_widget)
|
||
|
||
def on_freertos_task_btn_clicked(self):
|
||
if not self.freertos_enabled:
|
||
InfoBar.warning(
|
||
title="未开启 FreeRTOS",
|
||
content="请先在 CubeMX 中开启 FreeRTOS！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
return
|
||
self.generate_freertos_task()
|
||
|
||
def on_task_code_btn_clicked(self):
|
||
if not self.freertos_enabled:
|
||
InfoBar.warning(
|
||
title="未开启 FreeRTOS",
|
||
content="请先在 CubeMX 中开启 FreeRTOS！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
return
|
||
self.open_task_config_dialog()
|
||
|
||
def back_to_select(self):
|
||
self.stacked_layout.setCurrentWidget(self.select_widget)
|
||
|
||
def update_user_template(self):
|
||
url = "http://gitea.qutrobot.top/robofish/MRobot/archive/User_code.zip"
|
||
local_dir = "User_code"
|
||
try:
|
||
resp = requests.get(url, timeout=30)
|
||
resp.raise_for_status()
|
||
z = zipfile.ZipFile(io.BytesIO(resp.content))
|
||
if os.path.exists(local_dir):
|
||
shutil.rmtree(local_dir)
|
||
for member in z.namelist():
|
||
rel_path = os.path.relpath(member, z.namelist()[0])
|
||
if rel_path == ".":
|
||
continue
|
||
target_path = os.path.join(local_dir, rel_path)
|
||
if member.endswith('/'):
|
||
os.makedirs(target_path, exist_ok=True)
|
||
else:
|
||
os.makedirs(os.path.dirname(target_path), exist_ok=True)
|
||
with open(target_path, "wb") as f:
|
||
f.write(z.read(member))
|
||
InfoBar.success(
|
||
title="更新成功",
|
||
content="用户模板已更新到最新版本！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="更新失败",
|
||
content=f"用户模板更新失败: {e}",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
def show_user_code_files(self):
|
||
self.file_tree.clear()
|
||
base_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "User_code")
|
||
user_dir = os.path.join(self.project_path, "User")
|
||
sub_dirs = ["bsp", "component", "device", "module"]
|
||
|
||
# 读取所有 describe.csv 和 dependencies.csv
|
||
describe_map = {}
|
||
dependencies_map = {}
|
||
for sub in sub_dirs:
|
||
dir_path = os.path.join(base_dir, sub)
|
||
if not os.path.isdir(dir_path):
|
||
continue
|
||
# describe
|
||
desc_path = os.path.join(dir_path, "describe.csv")
|
||
if os.path.exists(desc_path):
|
||
with open(desc_path, encoding="utf-8") as f:
|
||
for line in f:
|
||
if "," in line:
|
||
k, v = line.strip().split(",", 1)
|
||
describe_map[f"{sub}/{k.strip()}"] = v.strip()
|
||
# dependencies
|
||
dep_path = os.path.join(dir_path, "dependencies.csv")
|
||
if os.path.exists(dep_path):
|
||
with open(dep_path, encoding="utf-8") as f:
|
||
for line in f:
|
||
if "," in line:
|
||
a, b = line.strip().split(",", 1)
|
||
dependencies_map.setdefault(f"{sub}/{a.strip()}", []).append(b.strip())
|
||
|
||
self._describe_map = describe_map
|
||
self._dependencies_map = dependencies_map
|
||
|
||
self.file_tree.setHeaderLabels(["模块名", "描述"])
|
||
self.file_tree.setSelectionMode(self.file_tree.ExtendedSelection)
|
||
self.file_tree.header().setSectionResizeMode(0, QHeaderView.ResizeToContents)
|
||
self.file_tree.header().setSectionResizeMode(1, QHeaderView.Stretch) # 描述列自适应
|
||
self.file_tree.setCheckedColor("#0078d4", "#2d7d9a")
|
||
self.file_tree.setBorderRadius(8)
|
||
self.file_tree.setBorderVisible(True)
|
||
|
||
for sub in sub_dirs:
|
||
dir_path = os.path.join(base_dir, sub)
|
||
if not os.path.isdir(dir_path):
|
||
continue
|
||
group_item = TreeItem([sub, ""])
|
||
self.file_tree.addTopLevelItem(group_item)
|
||
has_file = False
|
||
for root, _, files in os.walk(dir_path):
|
||
rel_root = os.path.relpath(root, base_dir)
|
||
for f in sorted(files):
|
||
if f.endswith(".c"):
|
||
mod_name = os.path.splitext(f)[0]
|
||
rel_c = os.path.join(rel_root, f)
|
||
key = f"{rel_root}/{mod_name}".replace("\\", "/")
|
||
desc = describe_map.get(key, "")
|
||
file_item = TreeItem([mod_name, desc])
|
||
file_item.setFlags(file_item.flags() | Qt.ItemIsUserCheckable)
|
||
file_item.setData(0, Qt.UserRole, rel_c)
|
||
file_item.setData(0, Qt.UserRole + 1, key) # 存模块key
|
||
file_item.setToolTip(1, desc)
|
||
file_item.setTextAlignment(1, Qt.AlignLeft | Qt.AlignVCenter)
|
||
group_item.addChild(file_item)
|
||
dst_c = os.path.join(user_dir, rel_c)
|
||
if os.path.exists(dst_c):
|
||
file_item.setCheckState(0, Qt.Unchecked)
|
||
file_item.setText(0, f"{mod_name}（已存在）")
|
||
file_item.setForeground(0, Qt.gray)
|
||
else:
|
||
file_item.setCheckState(0, Qt.Unchecked)
|
||
group_item.addChild(file_item)
|
||
has_file = True
|
||
if not has_file:
|
||
empty_item = TreeItem(["（无 .c 文件）", ""])
|
||
group_item.addChild(empty_item)
|
||
self.file_tree.expandAll()
|
||
|
||
# 勾选依赖自动勾选
|
||
def on_tree_item_changed(self, item, column):
|
||
if column != 0:
|
||
return
|
||
if item.childCount() > 0:
|
||
return # 只处理叶子
|
||
if item.checkState(0) == Qt.Checked:
|
||
key = item.data(0, Qt.UserRole + 1)
|
||
deps = self._dependencies_map.get(key, [])
|
||
if deps:
|
||
checked = []
|
||
root = self.file_tree.invisibleRootItem()
|
||
for i in range(root.childCount()):
|
||
group = root.child(i)
|
||
for j in range(group.childCount()):
|
||
child = group.child(j)
|
||
ckey = child.data(0, Qt.UserRole + 1)
|
||
if ckey in deps and child.checkState(0) != Qt.Checked:
|
||
child.setCheckState(0, Qt.Checked)
|
||
checked.append(ckey)
|
||
if checked:
|
||
descs = [self._describe_map.get(dep, dep) for dep in checked]
|
||
InfoBar.info(
|
||
title="依赖自动勾选",
|
||
content="已自动勾选依赖模块: " + "，".join(descs),
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
|
||
|
||
def get_checked_files(self):
|
||
files = []
|
||
def _traverse(item):
|
||
for i in range(item.childCount()):
|
||
child = item.child(i)
|
||
if child.childCount() == 0 and child.checkState(0) == Qt.Checked:
|
||
files.append(child.data(0, Qt.UserRole))
|
||
_traverse(child)
|
||
root = self.file_tree.invisibleRootItem()
|
||
for i in range(root.childCount()):
|
||
_traverse(root.child(i))
|
||
return files
|
||
|
||
def generate_code(self):
|
||
import shutil
|
||
base_dir = "User_code"
|
||
user_dir = os.path.join(self.project_path, "User")
|
||
copied = []
|
||
files = self.get_checked_files()
|
||
skipped = []
|
||
for rel_c in files:
|
||
rel_h = rel_c[:-2] + ".h"
|
||
src_c = os.path.join(base_dir, rel_c)
|
||
src_h = os.path.join(base_dir, rel_h)
|
||
dst_c = os.path.join(user_dir, rel_c)
|
||
dst_h = os.path.join(user_dir, rel_h)
|
||
# 如果目标文件已存在则跳过
|
||
if os.path.exists(dst_c):
|
||
skipped.append(dst_c)
|
||
else:
|
||
os.makedirs(os.path.dirname(dst_c), exist_ok=True)
|
||
shutil.copy2(src_c, dst_c)
|
||
copied.append(dst_c)
|
||
if os.path.exists(src_h):
|
||
if os.path.exists(dst_h):
|
||
skipped.append(dst_h)
|
||
else:
|
||
os.makedirs(os.path.dirname(dst_h), exist_ok=True)
|
||
shutil.copy2(src_h, dst_h)
|
||
copied.append(dst_h)
|
||
msg = f"已拷贝 {len(copied)} 个文件到 User 目录"
|
||
if skipped:
|
||
msg += f"\n{len(skipped)} 个文件已存在，未覆盖"
|
||
InfoBar.success(
|
||
title="生成完成",
|
||
content=msg,
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
# 生成后刷新文件树，更新标记
|
||
self.show_user_code_files()
|
||
|
||
def generate_freertos_task(self):
|
||
import re
|
||
freertos_path = os.path.join(self.project_path, "Core", "Src", "freertos.c")
|
||
if not os.path.exists(freertos_path):
|
||
InfoBar.error(
|
||
title="未找到 freertos.c",
|
||
content="未找到 Core/Src/freertos.c 文件，请确认工程路径。",
|
||
parent=self,
|
||
duration=2500
|
||
)
|
||
return
|
||
with open(freertos_path, "r", encoding="utf-8") as f:
|
||
code = f.read()
|
||
|
||
changed = False
|
||
error_msgs = []
|
||
|
||
# 1. 添加 #include "task/user_task.h"
|
||
include_line = '#include "task/user_task.h"'
|
||
if include_line not in code:
|
||
# 只插入到 USER CODE BEGIN Includes 区域
|
||
include_pattern = r'(\/\* *USER CODE BEGIN Includes *\*\/\s*)'
|
||
if re.search(include_pattern, code):
|
||
code = re.sub(
|
||
include_pattern,
|
||
r'\1' + include_line + '\n',
|
||
code
|
||
)
|
||
changed = True
|
||
else:
|
||
error_msgs.append("未找到 /* USER CODE BEGIN Includes */ 区域，无法插入 include。")
|
||
|
||
# 2. 在 /* USER CODE BEGIN RTOS_THREADS */ 区域添加 osThreadNew(Task_Init, NULL, &attr_init);
|
||
rtos_threads_pattern = r'(\/\* *USER CODE BEGIN RTOS_THREADS *\*\/\s*)(.*?)(\/\* *USER CODE END RTOS_THREADS *\*\/)'
|
||
match = re.search(rtos_threads_pattern, code, re.DOTALL)
|
||
task_line = ' osThreadNew(Task_Init, NULL, &attr_init); // 创建初始化任务\n'
|
||
if match:
|
||
threads_code = match.group(2)
|
||
if 'Task_Init' not in threads_code:
|
||
# 保留原有内容，追加新行
|
||
new_threads_code = match.group(1) + threads_code + task_line + match.group(3)
|
||
code = code[:match.start()] + new_threads_code + code[match.end():]
|
||
changed = True
|
||
else:
|
||
error_msgs.append("未找到 /* USER CODE BEGIN RTOS_THREADS */ 区域，无法插入任务创建代码。")
|
||
|
||
# 3. 清空 StartDefaultTask 的 USER CODE 区域，只保留 osThreadTerminate
|
||
sdt_pattern = r'(\/\* *USER CODE BEGIN StartDefaultTask *\*\/\s*)(.*?)(\/\* *USER CODE END StartDefaultTask *\*\/)'
|
||
match = re.search(sdt_pattern, code, re.DOTALL)
|
||
if match:
|
||
if 'osThreadTerminate(osThreadGetId());' not in match.group(2):
|
||
new_sdt_code = match.group(1) + ' osThreadTerminate(osThreadGetId());\n' + match.group(3)
|
||
code = code[:match.start()] + new_sdt_code + code[match.end():]
|
||
changed = True
|
||
else:
|
||
error_msgs.append("未找到 /* USER CODE BEGIN StartDefaultTask */ 区域，无法插入终止代码。")
|
||
|
||
if changed:
|
||
with open(freertos_path, "w", encoding="utf-8") as f:
|
||
f.write(code)
|
||
InfoBar.success(
|
||
title="生成成功",
|
||
content="FreeRTOS任务代码已自动生成！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
elif error_msgs:
|
||
InfoBar.error(
|
||
title="生成失败",
|
||
content="\n".join(error_msgs),
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
else:
|
||
InfoBar.info(
|
||
title="无需修改",
|
||
content="FreeRTOS任务相关代码已存在，无需重复生成。",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
|
||
def open_task_config_dialog(self):
|
||
from PyQt5.QtWidgets import QDialog, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QSpinBox, QPushButton, QTableWidget, QTableWidgetItem, QHeaderView, QCheckBox
|
||
import yaml
|
||
import os
|
||
|
||
class TaskConfigDialog(QDialog):
|
||
def __init__(self, parent=None, config_path=None):
|
||
super().__init__(parent)
|
||
self.setWindowTitle("任务配置")
|
||
self.resize(900, 420)
|
||
layout = QVBoxLayout(self)
|
||
self.table = QTableWidget(0, 6)
|
||
self.table.setHorizontalHeaderLabels(["任务名称", "运行频率", "初始化延迟", "堆栈大小", "任务描述", "频率控制"])
|
||
self.table.horizontalHeader().setSectionResizeMode(0, QHeaderView.Stretch)
|
||
self.table.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)
|
||
self.table.horizontalHeader().setSectionResizeMode(2, QHeaderView.Stretch)
|
||
self.table.horizontalHeader().setSectionResizeMode(3, QHeaderView.Stretch)
|
||
self.table.horizontalHeader().setSectionResizeMode(4, QHeaderView.Stretch)
|
||
self.table.horizontalHeader().setSectionResizeMode(5, QHeaderView.ResizeToContents)
|
||
self.table.setColumnWidth(4, 320) # 任务描述更宽
|
||
layout.addWidget(self.table)
|
||
btn_layout = QHBoxLayout()
|
||
add_btn = QPushButton("添加任务")
|
||
del_btn = QPushButton("删除选中")
|
||
ok_btn = QPushButton("生成")
|
||
cancel_btn = QPushButton("取消")
|
||
btn_layout.addWidget(add_btn)
|
||
btn_layout.addWidget(del_btn)
|
||
btn_layout.addStretch()
|
||
btn_layout.addWidget(ok_btn)
|
||
btn_layout.addWidget(cancel_btn)
|
||
layout.addLayout(btn_layout)
|
||
add_btn.clicked.connect(self.add_row)
|
||
del_btn.clicked.connect(self.del_row)
|
||
ok_btn.clicked.connect(self.accept)
|
||
cancel_btn.clicked.connect(self.reject)
|
||
|
||
|
||
|
||
|
||
# 自动读取配置文件
|
||
if config_path and os.path.exists(config_path):
|
||
try:
|
||
with open(config_path, "r", encoding="utf-8") as f:
|
||
tasks = yaml.safe_load(f)
|
||
if tasks:
|
||
for t in tasks:
|
||
row = self.table.rowCount()
|
||
self.table.insertRow(row)
|
||
for col, key in enumerate(["name", "frequency", "delay", "stack", "description"]):
|
||
item = QTableWidgetItem(str(t.get(key, "")))
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.table.setItem(row, col, item)
|
||
# 新增频率控制复选框
|
||
freq_ctrl = QCheckBox()
|
||
freq_ctrl.setChecked(t.get("freq_control", True))
|
||
self.table.setCellWidget(row, 5, freq_ctrl)
|
||
except Exception as e:
|
||
pass # 配置文件损坏时忽略
|
||
|
||
def add_row(self):
|
||
row = self.table.rowCount()
|
||
self.table.insertRow(row)
|
||
default_values = [
|
||
f"Task{row+1}", "500", "0", "256", "不要偷懒，请写清楚每个任务的作用！（如果你看到任务上面是这句话，说明作者是个懒蛋）"
|
||
]
|
||
for col, val in enumerate(default_values):
|
||
item = QTableWidgetItem(val)
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.table.setItem(row, col, item)
|
||
freq_ctrl = QCheckBox()
|
||
freq_ctrl.setChecked(True)
|
||
self.table.setCellWidget(row, 5, freq_ctrl)
|
||
|
||
def del_row(self):
|
||
rows = set([i.row() for i in self.table.selectedItems()])
|
||
for r in sorted(rows, reverse=True):
|
||
self.table.removeRow(r)
|
||
|
||
def get_tasks(self):
|
||
tasks = []
|
||
for row in range(self.table.rowCount()):
|
||
name = self.table.item(row, 0).text().strip()
|
||
freq = self.table.item(row, 1).text()
|
||
delay = int(self.table.item(row, 2).text())
|
||
stack = int(self.table.item(row, 3).text())
|
||
desc = self.table.item(row, 4).text().strip()
|
||
freq_ctrl = self.table.cellWidget(row, 5).isChecked()
|
||
# 校验 stack 必须为 128*2^n
|
||
if stack < 128 or (stack & (stack - 1)) != 0 or stack % 128 != 0:
|
||
raise ValueError(f"第{row+1}行任务“{name}”的堆栈大小必须为128、256、512、1024等（128*2^n）")
|
||
task = {
|
||
"name": name,
|
||
"function": f"Task_{name}",
|
||
"delay": delay,
|
||
"stack": stack,
|
||
"description": desc,
|
||
"freq_control": freq_ctrl
|
||
}
|
||
if freq_ctrl:
|
||
task["frequency"] = int(freq)
|
||
tasks.append(task)
|
||
return tasks
|
||
|
||
|
||
|
||
|
||
|
||
config_path = os.path.join(self.project_path, "User", "task", "config.yaml")
|
||
dlg = TaskConfigDialog(self, config_path=config_path)
|
||
if dlg.exec() == QDialog.Accepted:
|
||
try:
|
||
tasks = dlg.get_tasks()
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="参数错误",
|
||
content=str(e),
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
return
|
||
if not tasks:
|
||
InfoBar.warning(
|
||
title="未配置任务",
|
||
content="请至少添加一个任务！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
return
|
||
try:
|
||
self.generate_task_code(tasks)
|
||
InfoBar.success(
|
||
title="生成成功",
|
||
content="任务代码已生成到 User/task 目录！",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="生成失败",
|
||
content=f"任务代码生成失败: {e}",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
def generate_task_code(self, task_list):
|
||
import os
|
||
from jinja2 import Template
|
||
import yaml
|
||
import re
|
||
import textwrap
|
||
|
||
base_dir = os.path.dirname(os.path.abspath(__file__))
|
||
template_dir = os.path.join(base_dir, "User_code", "task")
|
||
output_dir = os.path.join(self.project_path, "User", "task")
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
|
||
# 模板路径
|
||
user_task_h_tpl = os.path.join(template_dir, "user_task.h.template")
|
||
user_task_c_tpl = os.path.join(template_dir, "user_task.c.template")
|
||
init_c_tpl = os.path.join(template_dir, "init.c.template")
|
||
task_c_tpl = os.path.join(template_dir, "task.c.template")
|
||
|
||
# 只统计需要频率控制的任务
|
||
freq_tasks = [t for t in task_list if t.get("freq_control", True)]
|
||
|
||
def render_template(path, context):
|
||
with open(path, encoding="utf-8") as f:
|
||
tpl = Template(f.read())
|
||
return tpl.render(**context)
|
||
|
||
# 构造模板上下文
|
||
context_h = {
|
||
"thread_definitions": "\n".join([f" osThreadId_t {t['name']};" for t in task_list]),
|
||
"freq_definitions": "\n".join([f" float {t['name']};" for t in freq_tasks]),
|
||
"stack_definitions": "\n".join([f" UBaseType_t {t['name']};" for t in task_list]),
|
||
"last_up_time_definitions": "\n".join([f" float {t['name']};" for t in freq_tasks]),
|
||
"task_frequency_definitions": "\n".join([f"#define {t['name'].upper()}_FREQ ({t['frequency']})" for t in freq_tasks]),
|
||
"task_init_delay_definitions": "\n".join([f"#define {t['name'].upper()}_INIT_DELAY ({t['delay']})" for t in task_list]),
|
||
"task_attr_declarations": "\n".join([f"extern const osThreadAttr_t attr_{t['name']};" for t in task_list]),
|
||
"task_function_declarations": "\n".join([f"void {t['function']}(void *argument);" for t in task_list]),
|
||
}
|
||
|
||
# ----------- 用户区域保护函数 -----------
|
||
def preserve_user_region(new_code, old_code, region_name):
|
||
"""
|
||
替换 new_code 中 region_name 区域为 old_code 中的内容（如果有）
|
||
region_name: 如 'USER INCLUDE'
|
||
"""
|
||
pattern = re.compile(
|
||
rf"/\*\s*{region_name}\s*BEGIN\s*\*/(.*?)/\*\s*{region_name}\s*END\s*\*/",
|
||
re.DOTALL
|
||
)
|
||
old_match = pattern.search(old_code or "")
|
||
if not old_match:
|
||
return new_code # 旧文件没有该区域，直接返回新代码
|
||
|
||
old_content = old_match.group(1)
|
||
def repl(m):
|
||
return m.group(0).replace(m.group(1), old_content)
|
||
# 替换新代码中的该区域
|
||
return pattern.sub(repl, new_code, count=1)
|
||
|
||
# ----------- 生成 user_task.h -----------
|
||
user_task_h_path = os.path.join(output_dir, "user_task.h")
|
||
new_user_task_h = render_template(user_task_h_tpl, context_h)
|
||
|
||
# 检查并保留所有用户区域
|
||
if os.path.exists(user_task_h_path):
|
||
with open(user_task_h_path, "r", encoding="utf-8") as f:
|
||
old_code = f.read()
|
||
# 只保留有内容的用户区域
|
||
for region in ["USER INCLUDE", "USER MESSAGE", "USER CONFIG"]:
|
||
# 如果旧文件该区域有内容，则保留
|
||
pattern = re.compile(
|
||
rf"/\*\s*{region}\s*BEGIN\s*\*/(.*?)/\*\s*{region}\s*END\s*\*/",
|
||
re.DOTALL
|
||
)
|
||
old_match = pattern.search(old_code)
|
||
if old_match and old_match.group(1).strip():
|
||
new_user_task_h = preserve_user_region(new_user_task_h, old_code, region)
|
||
# 写入
|
||
with open(user_task_h_path, "w", encoding="utf-8") as f:
|
||
f.write(new_user_task_h)
|
||
|
||
# ----------- 生成 user_task.c -----------
|
||
context_c = {
|
||
"task_attr_definitions": "\n".join([
|
||
f"const osThreadAttr_t attr_{t['name']} = {{\n"
|
||
f" .name = \"{t['name']}\",\n"
|
||
f" .priority = osPriorityNormal,\n"
|
||
f" .stack_size = {t['stack']} * 4,\n"
|
||
f"}};"
|
||
for t in task_list
|
||
])
|
||
}
|
||
user_task_c = render_template(user_task_c_tpl, context_c)
|
||
with open(os.path.join(output_dir, "user_task.c"), "w", encoding="utf-8") as f:
|
||
f.write(user_task_c)
|
||
|
||
# ----------- 生成 init.c -----------
|
||
# 线程创建代码
|
||
thread_creation_code = "\n".join([
|
||
f" task_runtime.thread.{t['name']} = osThreadNew({t['function']}, NULL, &attr_{t['name']});"
|
||
for t in task_list
|
||
])
|
||
|
||
context_init = {
|
||
"thread_creation_code": thread_creation_code,
|
||
}
|
||
# 渲染模板
|
||
init_c = render_template(init_c_tpl, context_init)
|
||
|
||
# 保留 USER MESSAGE 区域
|
||
def preserve_user_region(new_code, old_code, region_name):
|
||
pattern = re.compile(
|
||
rf"/\*\s*{region_name}\s*BEGIN\s*\*/(.*?)/\*\s*{region_name}\s*END\s*\*/",
|
||
re.DOTALL
|
||
)
|
||
old_match = pattern.search(old_code or "")
|
||
if not old_match:
|
||
return new_code
|
||
old_content = old_match.group(1)
|
||
def repl(m):
|
||
return m.group(0).replace(m.group(1), old_content)
|
||
return pattern.sub(repl, new_code, count=1)
|
||
|
||
init_c_path = os.path.join(output_dir, "init.c")
|
||
if os.path.exists(init_c_path):
|
||
with open(init_c_path, "r", encoding="utf-8") as f:
|
||
old_code = f.read()
|
||
# 保留 USER MESSAGE 区域
|
||
init_c = preserve_user_region(init_c, old_code, "USER MESSAGE")
|
||
|
||
with open(init_c_path, "w", encoding="utf-8") as f:
|
||
f.write(init_c)
|
||
|
||
# ----------- 生成 task.c -----------
|
||
for t in task_list:
|
||
desc = t.get("description", "")
|
||
desc_wrapped = "\n ".join(textwrap.wrap(desc, 20))
|
||
context_task = {
|
||
"task_name": t["name"],
|
||
"task_function": t["function"],
|
||
"task_frequency": f"{t['name'].upper()}_FREQ" if t.get("freq_control", True) else None,
|
||
"task_delay": f"{t['name'].upper()}_INIT_DELAY",
|
||
"task_description": desc_wrapped,
|
||
"freq_control": t.get("freq_control", True)
|
||
}
|
||
# 渲染模板
|
||
with open(task_c_tpl, encoding="utf-8") as f:
|
||
tpl = Template(f.read())
|
||
code = tpl.render(**context_task)
|
||
# 保留USER区域
|
||
task_c_path = os.path.join(output_dir, f"{t['name']}.c")
|
||
if os.path.exists(task_c_path):
|
||
with open(task_c_path, "r", encoding="utf-8") as f:
|
||
old_code = f.read()
|
||
def preserve_user_region(new_code, old_code, region_name):
|
||
pattern = re.compile(
|
||
rf"/\*\s*{region_name}\s*BEGIN\s*\*/(.*?)/\*\s*{region_name}\s*END\s*\*/",
|
||
re.DOTALL
|
||
)
|
||
old_match = pattern.search(old_code or "")
|
||
if not old_match:
|
||
return new_code
|
||
old_content = old_match.group(1)
|
||
def repl(m):
|
||
return m.group(0).replace(m.group(1), old_content)
|
||
return pattern.sub(repl, new_code, count=1)
|
||
for region in ["USER INCLUDE", "USER STRUCT", "USER CODE", "USER INIT CODE"]:
|
||
code = preserve_user_region(code, old_code, region)
|
||
with open(task_c_path, "w", encoding="utf-8") as f:
|
||
f.write(code)
|
||
|
||
# ----------- 保存任务配置到 config.yaml -----------
|
||
config_yaml_path = os.path.join(output_dir, "config.yaml")
|
||
with open(config_yaml_path, "w", encoding="utf-8") as f:
|
||
yaml.safe_dump(task_list, f, allow_unicode=True)
|
||
# ===================== 串口终端界面 =====================
|
||
class SerialReadThread(QThread):
|
||
data_received = pyqtSignal(str)
|
||
|
||
def __init__(self, ser):
|
||
super().__init__()
|
||
self.ser = ser
|
||
self._running = True
|
||
|
||
def run(self):
|
||
while self._running:
|
||
if self.ser and self.ser.is_open and self.ser.in_waiting:
|
||
try:
|
||
data = self.ser.readline().decode(errors='ignore')
|
||
self.data_received.emit(data)
|
||
except Exception:
|
||
pass
|
||
|
||
def stop(self):
|
||
self._running = False
|
||
self.wait()
|
||
|
||
class SerialTerminalInterface(BaseInterface):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("serialTerminalInterface")
|
||
main_layout = QVBoxLayout(self)
|
||
main_layout.setSpacing(12)
|
||
|
||
# 顶部：串口设置区
|
||
top_hbox = QHBoxLayout()
|
||
top_hbox.addWidget(BodyLabel("串口:"))
|
||
self.port_combo = ComboBox()
|
||
self.refresh_ports()
|
||
top_hbox.addWidget(self.port_combo)
|
||
top_hbox.addWidget(BodyLabel("波特率:"))
|
||
self.baud_combo = ComboBox()
|
||
self.baud_combo.addItems(['9600', '115200', '57600', '38400', '19200', '4800'])
|
||
top_hbox.addWidget(self.baud_combo)
|
||
self.connect_btn = PushButton("连接")
|
||
self.connect_btn.clicked.connect(self.toggle_connection)
|
||
top_hbox.addWidget(self.connect_btn)
|
||
self.refresh_btn = PushButton(FluentIcon.SYNC, "刷新")
|
||
self.refresh_btn.clicked.connect(self.refresh_ports)
|
||
top_hbox.addWidget(self.refresh_btn)
|
||
top_hbox.addStretch()
|
||
main_layout.addLayout(top_hbox)
|
||
|
||
main_layout.addWidget(HorizontalSeparator())
|
||
|
||
# 中部：左侧预设命令，右侧显示区
|
||
center_hbox = QHBoxLayout()
|
||
# 左侧：预设命令竖排
|
||
preset_vbox = QVBoxLayout()
|
||
preset_vbox.addWidget(SubtitleLabel("快捷指令"))
|
||
#快捷指令居中
|
||
preset_vbox.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||
self.preset_commands = [
|
||
("线程监视器", "htop"),
|
||
("陀螺仪校准", "cali_gyro"),
|
||
("性能监视", "htop"),
|
||
("重启", "reset"),
|
||
("显示所有设备", "ls /dev"),
|
||
("查询id", "id"),
|
||
]
|
||
for label, cmd in self.preset_commands:
|
||
btn = PushButton(label)
|
||
btn.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred)
|
||
btn.clicked.connect(lambda _, c=cmd: self.send_preset_command(c))
|
||
preset_vbox.addWidget(btn)
|
||
preset_vbox.addStretch()
|
||
main_layout.addLayout(center_hbox, stretch=1)
|
||
|
||
|
||
# 右侧：串口数据显示区
|
||
self.text_edit = TextEdit()
|
||
self.text_edit.setReadOnly(True)
|
||
self.text_edit.setMinimumWidth(400)
|
||
center_hbox.addWidget(self.text_edit, 3)
|
||
center_hbox.addLayout(preset_vbox, 1)
|
||
|
||
main_layout.addWidget(HorizontalSeparator())
|
||
|
||
# 底部：输入区
|
||
bottom_hbox = QHBoxLayout()
|
||
self.input_line = LineEdit()
|
||
self.input_line.setPlaceholderText("输入内容，回车发送")
|
||
self.input_line.returnPressed.connect(self.send_data)
|
||
bottom_hbox.addWidget(self.input_line, 4)
|
||
send_btn = PushButton("发送")
|
||
send_btn.clicked.connect(self.send_data)
|
||
bottom_hbox.addWidget(send_btn, 1)
|
||
self.auto_enter_checkbox = CheckBox("自动回车 ")
|
||
self.auto_enter_checkbox.setChecked(True)
|
||
bottom_hbox.addWidget(self.auto_enter_checkbox)
|
||
bottom_hbox.addStretch()
|
||
main_layout.addLayout(bottom_hbox)
|
||
|
||
self.ser = None
|
||
self.read_thread = None
|
||
|
||
def send_preset_command(self, cmd):
|
||
self.input_line.setText(cmd)
|
||
self.send_data()
|
||
|
||
def refresh_ports(self):
|
||
self.port_combo.clear()
|
||
ports = serial.tools.list_ports.comports()
|
||
for port in ports:
|
||
self.port_combo.addItem(port.device)
|
||
|
||
def toggle_connection(self):
|
||
if self.ser and self.ser.is_open:
|
||
self.disconnect_serial()
|
||
else:
|
||
self.connect_serial()
|
||
|
||
def connect_serial(self):
|
||
port = self.port_combo.currentText()
|
||
baud = int(self.baud_combo.currentText())
|
||
try:
|
||
self.ser = serial.Serial(port, baud, timeout=0.1)
|
||
self.connect_btn.setText("断开")
|
||
self.text_edit.append(f"已连接到 {port} @ {baud}")
|
||
self.read_thread = SerialReadThread(self.ser)
|
||
self.read_thread.data_received.connect(self.display_data)
|
||
self.read_thread.start()
|
||
except Exception as e:
|
||
self.text_edit.append(f"连接失败: {e}")
|
||
|
||
def disconnect_serial(self):
|
||
if self.read_thread:
|
||
self.read_thread.stop()
|
||
self.read_thread = None
|
||
if self.ser:
|
||
self.ser.close()
|
||
self.ser = None
|
||
self.connect_btn.setText("连接")
|
||
self.text_edit.append("已断开连接")
|
||
|
||
def display_data(self, data):
|
||
self.text_edit.moveCursor(QTextCursor.End)
|
||
self.text_edit.insertPlainText(data)
|
||
self.text_edit.moveCursor(QTextCursor.End)
|
||
|
||
def send_data(self):
|
||
if self.ser and self.ser.is_open:
|
||
text = self.input_line.text()
|
||
try:
|
||
if not text:
|
||
self.ser.write('\n'.encode())
|
||
else:
|
||
for char in text:
|
||
self.ser.write(char.encode())
|
||
# 判断是否自动回车
|
||
if self.auto_enter_checkbox.isChecked():
|
||
self.ser.write('\n'.encode())
|
||
except Exception as e:
|
||
self.text_edit.append(f"发送失败: {e}")
|
||
self.input_line.clear()
|
||
|
||
# ===================== 零件库页面 =====================
|
||
class DownloadThread(QThread):
|
||
progressChanged = pyqtSignal(int)
|
||
finished = pyqtSignal(list, list) # success, fail
|
||
|
||
def __init__(self, files, server_url, secret_key, local_dir, parent=None):
|
||
super().__init__(parent)
|
||
self.files = files
|
||
self.server_url = server_url
|
||
self.secret_key = secret_key
|
||
self.local_dir = local_dir
|
||
|
||
def run(self):
|
||
success, fail = [], []
|
||
total = len(self.files)
|
||
max_retry = 3 # 最大重试次数
|
||
for idx, rel_path in enumerate(self.files):
|
||
retry = 0
|
||
while retry < max_retry:
|
||
try:
|
||
# 先统一分隔符，再编码
|
||
rel_path_unix = rel_path.replace("\\", "/")
|
||
encoded_path = quote(rel_path_unix)
|
||
url = f"{self.server_url}/download/{encoded_path}"
|
||
params = {"key": self.secret_key}
|
||
resp = requests.get(url, params=params, stream=True, timeout=10)
|
||
if resp.status_code == 200:
|
||
local_path = os.path.join(self.local_dir, rel_path)
|
||
os.makedirs(os.path.dirname(local_path), exist_ok=True)
|
||
with open(local_path, "wb") as f:
|
||
shutil.copyfileobj(resp.raw, f)
|
||
success.append(rel_path)
|
||
break # 下载成功，跳出重试循环
|
||
else:
|
||
print(f"下载失败({resp.status_code}): {rel_path}，第{retry+1}次尝试")
|
||
retry += 1
|
||
except Exception as e:
|
||
print(f"下载异常: {rel_path}，第{retry+1}次尝试，错误: {e}")
|
||
retry += 1
|
||
else:
|
||
fail.append(rel_path)
|
||
self.progressChanged.emit(int((idx + 1) / total * 100))
|
||
self.finished.emit(success, fail)
|
||
class PartLibraryInterface(BaseInterface):
|
||
SERVER_URL = "http://154.37.215.220:5000"
|
||
SECRET_KEY = "MRobot_Download"
|
||
LOCAL_LIB_DIR = "mech_lib"
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("partLibraryInterface")
|
||
layout = QVBoxLayout(self)
|
||
layout.setSpacing(16)
|
||
|
||
layout.addWidget(SubtitleLabel("零件库（在线bate版）"))
|
||
layout.addWidget(HorizontalSeparator())
|
||
layout.addWidget(BodyLabel("感谢重庆邮电大学整理的零件库，选择需要的文件下载到本地。（如无法使用或者下载失败，请尝试重新下载或检查网络连接）"))
|
||
|
||
btn_layout = QHBoxLayout()
|
||
refresh_btn = PushButton(FluentIcon.SYNC, "刷新列表")
|
||
refresh_btn.clicked.connect(self.refresh_list)
|
||
btn_layout.addWidget(refresh_btn)
|
||
|
||
# 新增：打开本地零件库按钮
|
||
open_local_btn = PushButton(FluentIcon.FOLDER, "打开本地零件库")
|
||
open_local_btn.clicked.connect(self.open_local_lib)
|
||
btn_layout.addWidget(open_local_btn)
|
||
btn_layout.addStretch()
|
||
layout.addLayout(btn_layout)
|
||
|
||
self.tree = TreeWidget(self)
|
||
|
||
self.tree.setHeaderLabels(["名称", "类型"])
|
||
self.tree.setSelectionMode(self.tree.ExtendedSelection)
|
||
self.tree.header().setSectionResizeMode(0, QHeaderView.Stretch)
|
||
self.tree.header().setSectionResizeMode(1, QHeaderView.ResizeToContents)
|
||
self.tree.setCheckedColor("#0078d4", "#2d7d9a")
|
||
self.tree.setBorderRadius(8)
|
||
self.tree.setBorderVisible(True)
|
||
layout.addWidget(self.tree, stretch=1)
|
||
|
||
download_btn = PushButton(FluentIcon.DOWNLOAD, "下载选中文件")
|
||
download_btn.clicked.connect(self.download_selected_files)
|
||
layout.addWidget(download_btn)
|
||
|
||
self.refresh_list(first=True)
|
||
|
||
|
||
|
||
def refresh_list(self, first=False):
|
||
self.tree.clear()
|
||
try:
|
||
resp = requests.get(
|
||
f"{self.SERVER_URL}/list",
|
||
params={"key": self.SECRET_KEY},
|
||
timeout=5
|
||
)
|
||
resp.raise_for_status()
|
||
tree = resp.json()
|
||
self.populate_tree(self.tree, tree, "")
|
||
if not first:
|
||
InfoBar.success(
|
||
title="刷新成功",
|
||
content="零件库已经是最新的！",
|
||
parent=self,
|
||
position=InfoBarPosition.TOP,
|
||
duration=2000
|
||
)
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="刷新失败",
|
||
content=f"获取零件库失败: {e}",
|
||
parent=self,
|
||
position=InfoBarPosition.TOP,
|
||
duration=3000
|
||
)
|
||
|
||
def populate_tree(self, parent, node, path_prefix):
|
||
from PyQt5.QtWidgets import QTreeWidgetItem
|
||
for dname, dnode in node.get("dirs", {}).items():
|
||
item = QTreeWidgetItem([dname, "文件夹"])
|
||
if isinstance(parent, TreeWidget):
|
||
parent.addTopLevelItem(item)
|
||
else:
|
||
parent.addChild(item)
|
||
self.populate_tree(item, dnode, os.path.join(path_prefix, dname))
|
||
for fname in node.get("files", []):
|
||
item = QTreeWidgetItem([fname, "文件"])
|
||
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
|
||
item.setCheckState(0, Qt.Unchecked)
|
||
item.setData(0, Qt.UserRole, os.path.join(path_prefix, fname))
|
||
if isinstance(parent, TreeWidget):
|
||
parent.addTopLevelItem(item)
|
||
else:
|
||
parent.addChild(item)
|
||
|
||
def get_checked_files(self):
|
||
files = []
|
||
def _traverse(item):
|
||
for i in range(item.childCount()):
|
||
child = item.child(i)
|
||
if child.text(1) == "文件" and child.checkState(0) == Qt.Checked:
|
||
files.append(child.data(0, Qt.UserRole))
|
||
_traverse(child)
|
||
root = self.tree.invisibleRootItem()
|
||
for i in range(root.childCount()):
|
||
_traverse(root.child(i))
|
||
return files
|
||
|
||
def download_selected_files(self):
|
||
files = self.get_checked_files()
|
||
if not files:
|
||
InfoBar.info(
|
||
title="提示",
|
||
content="请先勾选要下载的文件。",
|
||
parent=self,
|
||
position=InfoBarPosition.TOP,
|
||
duration=2000
|
||
)
|
||
return
|
||
|
||
# 进度条对话框
|
||
self.progress_dialog = Dialog(
|
||
title="正在下载",
|
||
content="正在下载选中文件，请稍候...",
|
||
parent=self
|
||
)
|
||
self.progress_bar = ProgressBar()
|
||
self.progress_bar.setValue(0)
|
||
# 插入进度条到内容布局
|
||
self.progress_dialog.textLayout.addWidget(self.progress_bar)
|
||
self.progress_dialog.show()
|
||
|
||
# 启动下载线程
|
||
self.download_thread = DownloadThread(
|
||
files, self.SERVER_URL, self.SECRET_KEY, self.LOCAL_LIB_DIR
|
||
)
|
||
self.download_thread.progressChanged.connect(self.progress_bar.setValue)
|
||
self.download_thread.finished.connect(self.on_download_finished)
|
||
self.download_thread.finished.connect(self.download_thread.deleteLater)
|
||
self.download_thread.start()
|
||
|
||
def on_download_finished(self, success, fail):
|
||
self.progress_dialog.close()
|
||
msg = f"成功下载: {len(success)} 个文件\n失败: {len(fail)} 个文件"
|
||
dialog = Dialog(
|
||
title="下载结果",
|
||
content=msg,
|
||
parent=self
|
||
)
|
||
# 添加“打开文件夹”按钮
|
||
open_btn = PushButton("打开文件夹")
|
||
def open_folder():
|
||
folder = os.path.abspath(self.LOCAL_LIB_DIR)
|
||
# 打开文件夹（macOS用open，Windows用explorer，Linux用xdg-open）
|
||
import platform, subprocess
|
||
if platform.system() == "Darwin":
|
||
subprocess.call(["open", folder])
|
||
elif platform.system() == "Windows":
|
||
subprocess.call(["explorer", folder])
|
||
else:
|
||
subprocess.call(["xdg-open", folder])
|
||
dialog.close()
|
||
open_btn.clicked.connect(open_folder)
|
||
# 添加按钮到Dialog布局
|
||
dialog.textLayout.addWidget(open_btn)
|
||
dialog.exec()
|
||
|
||
def open_local_lib(self):
|
||
folder = os.path.abspath(self.LOCAL_LIB_DIR)
|
||
import platform, subprocess
|
||
if platform.system() == "Darwin":
|
||
subprocess.call(["open", folder])
|
||
elif platform.system() == "Windows":
|
||
subprocess.call(["explorer", folder])
|
||
else:
|
||
subprocess.call(["xdg-open", folder])
|
||
|
||
# ===================== 设置界面 =====================
|
||
class SettingInterface(BaseInterface):
|
||
themeSwitchRequested = pyqtSignal()
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("settingInterface")
|
||
layout = QVBoxLayout()
|
||
self.setLayout(layout)
|
||
|
||
# 标题
|
||
layout.addSpacing(10)
|
||
layout.addWidget(SubtitleLabel("设置中心"))
|
||
layout.addSpacing(10)
|
||
layout.addWidget(HorizontalSeparator())
|
||
|
||
# 主题切换区域
|
||
theme_title = StrongBodyLabel("外观设置")
|
||
theme_desc = BodyLabel("切换夜间/白天模式，适应不同环境。")
|
||
theme_desc.setWordWrap(True)
|
||
layout.addSpacing(10)
|
||
layout.addWidget(theme_title)
|
||
layout.addWidget(theme_desc)
|
||
|
||
theme_box = QHBoxLayout()
|
||
self.theme_label = BodyLabel("夜间模式")
|
||
self.theme_switch = SwitchButton()
|
||
self.theme_switch.setChecked(Theme.DARK == Theme.DARK)
|
||
self.theme_switch.checkedChanged.connect(self.on_theme_switch)
|
||
theme_box.addWidget(self.theme_label)
|
||
theme_box.addWidget(self.theme_switch)
|
||
theme_box.addStretch()
|
||
layout.addLayout(theme_box)
|
||
|
||
layout.addSpacing(15)
|
||
layout.addWidget(HorizontalSeparator())
|
||
|
||
# 其它设置区域（示例）
|
||
other_title = StrongBodyLabel("其它设置")
|
||
other_desc = BodyLabel("更多功能正在开发中，敬请期待。")
|
||
other_desc.setWordWrap(True)
|
||
layout.addSpacing(10)
|
||
layout.addWidget(other_title)
|
||
layout.addWidget(other_desc)
|
||
|
||
# 版权信息
|
||
layout.addStretch()
|
||
copyright_label = BodyLabel("© 2025 MRobot Toolbox")
|
||
copyright_label.setAlignment(Qt.AlignmentFlag.AlignHCenter)
|
||
layout.addWidget(copyright_label)
|
||
layout.addSpacing(10)
|
||
|
||
def on_theme_switch(self, checked):
|
||
self.themeSwitchRequested.emit()
|
||
|
||
# ===================== 帮助与关于界面 =====================
|
||
|
||
# ...existing code...
|
||
|
||
|
||
# 注意：PushSettingCard、HyperlinkCard 已由你的 SettingCard 文件定义
|
||
|
||
# ...existing code...
|
||
|
||
from PyQt5.QtWidgets import QScrollArea, QWidget, QVBoxLayout, QMessageBox
|
||
from qfluentwidgets import ScrollArea, VBoxLayout
|
||
from qfluentwidgets import VBoxLayout
|
||
class HelpInterface(BaseInterface):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("helpInterface")
|
||
|
||
layout = QVBoxLayout()
|
||
layout.setContentsMargins(0, 0, 0, 0)
|
||
layout.setSpacing(0)
|
||
self.setLayout(layout)
|
||
|
||
content_widget = QWidget()
|
||
main_layout = VBoxLayout(content_widget)
|
||
main_layout.setContentsMargins(32, 32, 32, 32)
|
||
main_layout.setSpacing(18)
|
||
# 标题
|
||
main_layout.addWidget(SubtitleLabel("帮助中心"))
|
||
main_layout.addWidget(HorizontalSeparator())
|
||
|
||
# 版本与更新
|
||
version_card = PushSettingCard(
|
||
"检查更新",
|
||
FluentIcon.INFO,
|
||
f"当前版本：MRobot Toolbox v{__version__}",
|
||
"点击按钮检查是否有新版本。",
|
||
parent=self
|
||
)
|
||
version_card.clicked.connect(self.check_update)
|
||
main_layout.addWidget(version_card)
|
||
|
||
# FAQ分组
|
||
faq_group = SettingCardGroup("常见问题", self)
|
||
faq_card1 = PushSettingCard(
|
||
"查看解决方法",
|
||
FluentIcon.HELP,
|
||
"启动报错/界面异常怎么办？",
|
||
"遇到启动问题请尝试重启、检查依赖，或加入交流群获取帮助。",
|
||
parent=self
|
||
)
|
||
faq_card1.clicked.connect(lambda: self.show_info(
|
||
"启动报错/界面异常解决方法",
|
||
"1. 尝试重启软件。\n2. 检查Python和依赖库版本。\n3. 如仍有问题，请在GitHub提交Issue。"
|
||
))
|
||
faq_group.addSettingCard(faq_card1)
|
||
|
||
faq_card2 = PushSettingCard(
|
||
"查看解决方法",
|
||
FluentIcon.LIBRARY,
|
||
"零件库无法下载怎么办？",
|
||
"如遇网络问题或下载失败，请多次尝试或联系管理员。",
|
||
parent=self
|
||
)
|
||
faq_card2.clicked.connect(lambda: self.show_info(
|
||
"零件库无法下载解决方法",
|
||
"1. 检查网络连接。\n2. 多次刷新或重启软件。\n3. 若仍无法下载，请加入QQ群：857466609 反馈。"
|
||
))
|
||
faq_group.addSettingCard(faq_card2)
|
||
|
||
# faq_card3 = PushSettingCard(
|
||
# "获取下载链接",
|
||
# FluentIcon.DOWNLOAD,
|
||
# "如何下载最新版？",
|
||
# "点击按钮获取最新版下载地址。",
|
||
# parent=self
|
||
# )
|
||
# faq_card3.clicked.connect(lambda: self.show_info(
|
||
# "最新版下载地址",
|
||
# "GitHub发布页：https://github.com/goldenfishs/MRobot/releases\n如遇下载问题，请联系QQ群：857466609"
|
||
# ))
|
||
# faq_group.addSettingCard(faq_card3)
|
||
main_layout.addWidget(faq_group)
|
||
|
||
# 联系方式
|
||
contact_group = SettingCardGroup("联系方式", self)
|
||
contact_card = PushSettingCard(
|
||
"复制邮箱",
|
||
FluentIcon.MESSAGE,
|
||
"联系开发团队",
|
||
"点击按钮复制邮箱地址：support@mrobot.com",
|
||
parent=self
|
||
)
|
||
contact_card.clicked.connect(lambda: self.copy_text("support@mrobot.com", "邮箱已复制：1683502971@qq.com"))
|
||
contact_group.addSettingCard(contact_card)
|
||
main_layout.addWidget(contact_group)
|
||
|
||
main_layout.addStretch()
|
||
# 不使用滚动区，直接添加内容区
|
||
layout.addWidget(content_widget)
|
||
|
||
def copy_text(self, text, message):
|
||
clipboard = QApplication.clipboard()
|
||
clipboard.setText(text)
|
||
InfoBar.info(
|
||
title="已复制",
|
||
content=message,
|
||
parent=self,
|
||
position=InfoBarPosition.TOP,
|
||
duration=2000
|
||
)
|
||
|
||
def check_update(self):
|
||
latest = check_update()
|
||
if latest:
|
||
self.show_info(
|
||
"发现新版本",
|
||
f"检测到新版本 {latest}，请前往 GitHub 下载：\nhttps://github.com/goldenfishs/MRobot/releases"
|
||
)
|
||
else:
|
||
InfoBar.info(
|
||
title="已是最新版",
|
||
content="当前已是最新版本。",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
|
||
def show_info(self, title, content):
|
||
dialog = Dialog(
|
||
title=title,
|
||
content=content,
|
||
parent=self
|
||
)
|
||
dialog.exec()
|
||
|
||
class AboutInterface(BaseInterface):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("aboutInterface")
|
||
layout = QVBoxLayout()
|
||
self.setLayout(layout)
|
||
|
||
# ===================== 主窗口与导航 =====================
|
||
class MainWindow(FluentWindow):
|
||
themeChanged = pyqtSignal(Theme)
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.setWindowTitle("MR_ToolBox")
|
||
self.resize(800, 600)
|
||
self.setMinimumSize(640, 480)
|
||
self.setWindowFlag(Qt.Window)
|
||
# 记录当前主题
|
||
self.current_theme = Theme.DARK
|
||
latest = check_update()
|
||
if latest:
|
||
InfoBar.info(
|
||
title="发现新版本",
|
||
content=f"检测到新版本 {latest}，请前往 GitHub 下载更新。",
|
||
parent=self,
|
||
duration=5000
|
||
)
|
||
# 创建页面实例
|
||
self.setting_page = SettingInterface(self)
|
||
self.setting_page.themeSwitchRequested.connect(self.toggle_theme)
|
||
|
||
self.page_registry = [
|
||
(HomeInterface(self), FluentIcon.HOME, "首页", NavigationItemPosition.TOP),
|
||
(DataInterface(self), FluentIcon.LIBRARY, "MRobot代码生成", NavigationItemPosition.SCROLL),
|
||
(SerialTerminalInterface(self), FluentIcon.COMMAND_PROMPT, "Mini_Shell", NavigationItemPosition.SCROLL),
|
||
(PartLibraryInterface(self), FluentIcon.DOWNLOAD, "零件库", NavigationItemPosition.SCROLL), # ← 加上这一行
|
||
(self.setting_page, FluentIcon.SETTING, "设置", NavigationItemPosition.BOTTOM),
|
||
(HelpInterface(self), FluentIcon.HELP, "帮助", NavigationItemPosition.BOTTOM),
|
||
# (AboutInterface(self), FluentIcon.INFO, "关于", NavigationItemPosition.BOTTOM),
|
||
]
|
||
self.initNavigation()
|
||
|
||
def initNavigation(self):
|
||
for page, icon, name, position in self.page_registry:
|
||
self.addSubInterface(page, icon, name, position)
|
||
self.navigationInterface.addSeparator()
|
||
avatar = NavigationAvatarWidget('用户', ':/qfluentwidgets/images/avatar.png')
|
||
self.navigationInterface.addWidget(
|
||
routeKey='avatar',
|
||
widget=avatar,
|
||
onClick=self.show_user_info, # 这里改为 self.show_user_info
|
||
position=NavigationItemPosition.BOTTOM
|
||
)
|
||
|
||
def toggle_theme(self):
|
||
# 切换主题
|
||
if self.current_theme == Theme.DARK:
|
||
self.current_theme = Theme.LIGHT
|
||
else:
|
||
self.current_theme = Theme.DARK
|
||
setTheme(self.current_theme)
|
||
# 同步设置界面按钮状态
|
||
self.setting_page.theme_switch.setChecked(self.current_theme == Theme.DARK)
|
||
|
||
def show_user_info(self):
|
||
dialog = Dialog(
|
||
title="MRobot",
|
||
content="账号：VIP内测版",
|
||
parent=self
|
||
)
|
||
dialog.exec()
|
||
|
||
def check_update():
|
||
try:
|
||
repo = "goldenfishs/MRobot"
|
||
url = f"https://api.github.com/repos/{repo}/releases/latest"
|
||
resp = requests.get(url, timeout=5)
|
||
if resp.status_code == 200:
|
||
latest = resp.json()["tag_name"].lstrip("v")
|
||
print(f"本地版本: {__version__}, 最新版本: {latest}") # 调试用
|
||
if vparse(latest) > vparse(__version__):
|
||
return latest
|
||
except Exception as e:
|
||
print(f"检查更新失败: {e}")
|
||
return None
|
||
|
||
# ===================== 程序入口 =====================
|
||
def main():
|
||
from PyQt5.QtWidgets import QApplication # <-- 移到这里，所有平台都能用
|
||
import platform
|
||
if platform.system() == "Windows":
|
||
try:
|
||
from PyQt5.QtCore import Qt
|
||
import ctypes
|
||
ctypes.windll.shcore.SetProcessDpiAwareness(1)
|
||
except Exception:
|
||
pass
|
||
QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
|
||
QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)
|
||
|
||
app = QApplication(sys.argv)
|
||
# 跟随系统主题
|
||
setTheme(Theme.DARK)
|
||
splash = SplashScreen()
|
||
setTheme(Theme.DARK)
|
||
splash.show()
|
||
setTheme(Theme.DARK)
|
||
app.processEvents()
|
||
|
||
# 步骤1：获取零件库
|
||
splash.set_status("正在获取零件仓库...", 20)
|
||
try:
|
||
import requests
|
||
resp = requests.get("http://154.37.215.220:5000/list", params={"key": "MRobot_Download"}, timeout=5)
|
||
resp.raise_for_status()
|
||
except Exception:
|
||
pass
|
||
app.processEvents()
|
||
|
||
# 步骤2：检查更新
|
||
splash.set_status("正在检查软件更新...", 60)
|
||
latest = check_update()
|
||
app.processEvents()
|
||
|
||
# 步骤3：加载主窗口
|
||
splash.set_status("正在加载主界面...", 90)
|
||
window = MainWindow()
|
||
window.show()
|
||
setTheme(Theme.DARK)
|
||
splash.set_status("启动完成", 100)
|
||
from PyQt5.QtCore import QTimer
|
||
QTimer.singleShot(500, splash.close)
|
||
|
||
# 有新版本弹窗
|
||
if latest:
|
||
InfoBar.info(
|
||
title="发现新版本",
|
||
content=f"检测到新版本 {latest}，请前往帮助页面下载新版。",
|
||
parent=window,
|
||
duration=5000
|
||
)
|
||
|
||
sys.exit(app.exec_())
|
||
|
||
if __name__ == '__main__':
|
||
main() |