mirror of
https://github.com/goldenfishs/MRobot.git
synced 2026-02-04 18:00:19 +08:00
346 lines
14 KiB
Python
346 lines
14 KiB
Python
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QStackedWidget, QSizePolicy
|
||
from PyQt5.QtCore import Qt
|
||
from qfluentwidgets import PushSettingCard, FluentIcon, TabBar
|
||
from qfluentwidgets import TitleLabel, BodyLabel, PushButton, FluentIcon
|
||
from PyQt5.QtWidgets import QFileDialog, QDialog, QHBoxLayout
|
||
from qfluentwidgets import ComboBox, PrimaryPushButton, SubtitleLabel
|
||
import os
|
||
import shutil
|
||
import tempfile
|
||
from .function_fit_interface import FunctionFitInterface
|
||
from .ai_interface import AIInterface
|
||
from qfluentwidgets import InfoBar
|
||
from .tools.update_code import update_code
|
||
from .code_generate_interface import CodeGenerateInterface
|
||
|
||
class CodeConfigurationInterface(QWidget):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.setObjectName("CodeConfigurationInterface")
|
||
self.vBoxLayout = QVBoxLayout(self)
|
||
self.vBoxLayout.setAlignment(Qt.AlignTop)
|
||
self.vBoxLayout.setContentsMargins(10, 0, 10, 10) # 设置外边距
|
||
|
||
# 顶部标签栏，横向拉伸
|
||
self.tabBar = TabBar(self)
|
||
self.tabBar.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
|
||
self.vBoxLayout.addWidget(self.tabBar)
|
||
|
||
self.stackedWidget = QStackedWidget(self)
|
||
self.vBoxLayout.addWidget(self.stackedWidget)
|
||
|
||
# 初始主页面
|
||
self.mainPage = QWidget(self)
|
||
mainLayout = QVBoxLayout(self.mainPage)
|
||
mainLayout.setAlignment(Qt.AlignTop)
|
||
mainLayout.setSpacing(28) # 设置间距
|
||
mainLayout.setContentsMargins(48, 48, 48, 48) # 设置内容边距
|
||
|
||
#添加空行
|
||
title = TitleLabel("MRobot 代码生成")
|
||
title.setAlignment(Qt.AlignCenter)
|
||
mainLayout.addWidget(title)
|
||
|
||
subtitle = BodyLabel("请选择您的由CUBEMX生成的工程路径（.ico所在的目录），然后开启代码之旅！")
|
||
subtitle.setAlignment(Qt.AlignCenter)
|
||
mainLayout.addWidget(subtitle)
|
||
|
||
desc = BodyLabel("支持自动配置和生成任务，自主选择模块代码倒入，自动识别cubemx配置！")
|
||
desc.setAlignment(Qt.AlignCenter)
|
||
mainLayout.addWidget(desc)
|
||
|
||
mainLayout.addSpacing(18)
|
||
|
||
self.choose_btn = PushButton(FluentIcon.FOLDER, "选择项目路径")
|
||
self.choose_btn.setFixedWidth(200)
|
||
mainLayout.addWidget(self.choose_btn, alignment=Qt.AlignmentFlag.AlignCenter)
|
||
|
||
self.update_template_btn = PushButton(FluentIcon.SYNC, "更新代码库")
|
||
self.update_template_btn.setFixedWidth(200)
|
||
mainLayout.addWidget(self.update_template_btn, alignment=Qt.AlignmentFlag.AlignCenter)
|
||
|
||
self.preset_ioc_btn = PushButton(FluentIcon.LIBRARY, "获取预设IOC")
|
||
self.preset_ioc_btn.setFixedWidth(200)
|
||
mainLayout.addWidget(self.preset_ioc_btn, alignment=Qt.AlignmentFlag.AlignCenter)
|
||
|
||
mainLayout.addSpacing(10)
|
||
mainLayout.addStretch()
|
||
|
||
# 添加主页面到堆叠窗口
|
||
self.addSubInterface(self.mainPage, "mainPage", "代码生成主页")
|
||
|
||
self.setLayout(self.vBoxLayout)
|
||
|
||
# 信号连接
|
||
self.stackedWidget.currentChanged.connect(self.onCurrentIndexChanged)
|
||
self.tabBar.tabCloseRequested.connect(self.onCloseTab)
|
||
self.choose_btn.clicked.connect(self.choose_project_folder) # 启用选择项目路径按钮
|
||
self.update_template_btn.clicked.connect(self.on_update_template)
|
||
self.preset_ioc_btn.clicked.connect(self.use_preset_ioc) # 启用预设工程按钮
|
||
|
||
|
||
def on_update_template(self):
|
||
def info(parent):
|
||
InfoBar.success(
|
||
title="更新成功",
|
||
content="用户模板已更新到最新版本！",
|
||
parent=parent,
|
||
duration=2000
|
||
)
|
||
def error(parent, msg):
|
||
InfoBar.error(
|
||
title="更新失败",
|
||
content=f"用户模板更新失败: {msg}",
|
||
parent=parent,
|
||
duration=3000
|
||
)
|
||
update_code(parent=self, info_callback=info, error_callback=error)
|
||
|
||
def get_preset_ioc_files(self):
|
||
"""获取预设的ioc文件列表"""
|
||
try:
|
||
preset_ioc_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), "assets", "User_code", "ioc")
|
||
if not os.path.exists(preset_ioc_dir):
|
||
return []
|
||
|
||
ioc_files = []
|
||
for filename in os.listdir(preset_ioc_dir):
|
||
if filename.endswith('.ioc'):
|
||
ioc_files.append({
|
||
'name': os.path.splitext(filename)[0],
|
||
'filename': filename,
|
||
'path': os.path.join(preset_ioc_dir, filename)
|
||
})
|
||
return ioc_files
|
||
except Exception as e:
|
||
print(f"获取预设ioc文件失败: {e}")
|
||
return []
|
||
|
||
def use_preset_ioc(self):
|
||
"""使用预设ioc文件"""
|
||
preset_files = self.get_preset_ioc_files()
|
||
if not preset_files:
|
||
InfoBar.warning(
|
||
title="无预设工程",
|
||
content="未找到可用的预设工程文件",
|
||
parent=self,
|
||
duration=2000
|
||
)
|
||
return
|
||
|
||
# 创建选择对话框
|
||
dialog = QDialog(self)
|
||
dialog.setWindowTitle("获取预设IOC")
|
||
dialog.resize(400, 200)
|
||
dialog.setModal(True)
|
||
|
||
layout = QVBoxLayout(dialog)
|
||
layout.setContentsMargins(24, 24, 24, 24)
|
||
layout.setSpacing(16)
|
||
|
||
# 标题
|
||
title_label = SubtitleLabel("选择要使用的IOC模版")
|
||
layout.addWidget(title_label)
|
||
|
||
# 选择下拉框
|
||
select_layout = QHBoxLayout()
|
||
select_label = BodyLabel("预设IOC：")
|
||
preset_combo = ComboBox()
|
||
|
||
# 修复ComboBox数据问题
|
||
for i, preset in enumerate(preset_files):
|
||
preset_combo.addItem(preset['name'])
|
||
|
||
select_layout.addWidget(select_label)
|
||
select_layout.addWidget(preset_combo)
|
||
layout.addLayout(select_layout)
|
||
|
||
layout.addSpacing(16)
|
||
|
||
# 按钮区域
|
||
btn_layout = QHBoxLayout()
|
||
btn_layout.addStretch()
|
||
|
||
cancel_btn = PushButton("取消")
|
||
ok_btn = PrimaryPushButton("保存到")
|
||
|
||
cancel_btn.clicked.connect(dialog.reject)
|
||
ok_btn.clicked.connect(dialog.accept)
|
||
|
||
btn_layout.addWidget(cancel_btn)
|
||
btn_layout.addWidget(ok_btn)
|
||
layout.addLayout(btn_layout)
|
||
|
||
# 显示对话框
|
||
if dialog.exec() == QDialog.Accepted:
|
||
selected_index = preset_combo.currentIndex()
|
||
if selected_index >= 0 and selected_index < len(preset_files):
|
||
selected_preset = preset_files[selected_index]
|
||
self.save_preset_template(selected_preset)
|
||
|
||
def save_preset_template(self, preset_info):
|
||
"""保存预设模板到用户指定位置"""
|
||
try:
|
||
# 让用户选择保存位置
|
||
save_dir = QFileDialog.getExistingDirectory(
|
||
self,
|
||
f"选择保存 {preset_info['name']} 模板的位置",
|
||
os.path.expanduser("~")
|
||
)
|
||
|
||
if not save_dir:
|
||
return
|
||
|
||
# 复制ioc文件到用户选择的目录
|
||
target_path = os.path.join(save_dir, preset_info['filename'])
|
||
|
||
# 检查目标文件是否已存在
|
||
if os.path.exists(target_path):
|
||
from qfluentwidgets import Dialog
|
||
dialog = Dialog("文件已存在", f"目标位置已存在 {preset_info['filename']}，是否覆盖？", self)
|
||
if dialog.exec() != Dialog.Accepted:
|
||
return
|
||
|
||
# 复制文件
|
||
shutil.copy2(preset_info['path'], target_path)
|
||
|
||
InfoBar.success(
|
||
title="模板保存成功",
|
||
content=f"预设模板 {preset_info['name']} 已保存到:\n{target_path}",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="保存失败",
|
||
content=f"保存预设模板失败: {str(e)}",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
def open_project_from_path(self, folder_path):
|
||
"""从指定路径打开工程"""
|
||
try:
|
||
if not os.path.exists(folder_path):
|
||
return
|
||
|
||
ioc_files = [f for f in os.listdir(folder_path) if f.endswith('.ioc')]
|
||
if ioc_files:
|
||
# 检查是否已存在 codeGenPage 标签页
|
||
for i in range(self.stackedWidget.count()):
|
||
widget = self.stackedWidget.widget(i)
|
||
if widget is not None and widget.objectName() == "codeGenPage":
|
||
# 如果已存在，则切换到该标签页，并更新路径显示
|
||
if hasattr(widget, "project_path"):
|
||
widget.project_path = folder_path
|
||
if hasattr(widget, "refresh"):
|
||
widget.refresh()
|
||
self.stackedWidget.setCurrentWidget(widget)
|
||
self.tabBar.setCurrentTab("codeGenPage")
|
||
return
|
||
|
||
# 不存在则新建
|
||
code_gen_page = CodeGenerateInterface(folder_path, self)
|
||
self.addSubInterface(code_gen_page, "codeGenPage", "代码生成")
|
||
self.stackedWidget.setCurrentWidget(code_gen_page)
|
||
self.tabBar.setCurrentTab("codeGenPage")
|
||
else:
|
||
InfoBar.error(
|
||
title="未找到.ioc文件",
|
||
content="所选文件夹不是有效的CUBEMX工程目录",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
except Exception as e:
|
||
InfoBar.error(
|
||
title="打开工程失败",
|
||
content=f"打开工程失败: {str(e)}",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
def choose_project_folder(self):
|
||
folder = QFileDialog.getExistingDirectory(self, "选择CUBEMX工程目录")
|
||
if not folder:
|
||
return
|
||
ioc_files = [f for f in os.listdir(folder) if f.endswith('.ioc')]
|
||
if ioc_files:
|
||
# 检查是否已存在 codeGenPage 标签页
|
||
for i in range(self.stackedWidget.count()):
|
||
widget = self.stackedWidget.widget(i)
|
||
if widget is not None and widget.objectName() == "codeGenPage":
|
||
# 如果已存在，则切换到该标签页，并更新路径显示
|
||
if hasattr(widget, "project_path"):
|
||
widget.project_path = folder
|
||
if hasattr(widget, "refresh"):
|
||
widget.refresh()
|
||
self.stackedWidget.setCurrentWidget(widget)
|
||
self.tabBar.setCurrentTab("codeGenPage")
|
||
return
|
||
# 不存在则新建
|
||
code_gen_page = CodeGenerateInterface(folder, self)
|
||
self.addSubInterface(code_gen_page, "codeGenPage", "代码生成")
|
||
self.stackedWidget.setCurrentWidget(code_gen_page)
|
||
self.tabBar.setCurrentTab("codeGenPage")
|
||
else:
|
||
InfoBar.error(
|
||
title="未找到.ioc文件",
|
||
content="所选文件夹不是有效的CUBEMX工程目录，请重新选择。",
|
||
parent=self,
|
||
duration=3000
|
||
)
|
||
|
||
|
||
def addSubInterface(self, widget: QWidget, objectName: str, text: str):
|
||
widget.setObjectName(objectName)
|
||
self.stackedWidget.addWidget(widget)
|
||
self.tabBar.addTab(
|
||
routeKey=objectName,
|
||
text=text,
|
||
onClick=lambda: self.stackedWidget.setCurrentWidget(widget)
|
||
)
|
||
|
||
def onCurrentIndexChanged(self, index):
|
||
widget = self.stackedWidget.widget(index)
|
||
self.tabBar.setCurrentTab(widget.objectName())
|
||
|
||
def onAddNewTab(self):
|
||
pass # 可自定义添加新标签页逻辑
|
||
|
||
def onCloseTab(self, index: int):
|
||
item = self.tabBar.tabItem(index)
|
||
widget = self.findChild(QWidget, item.routeKey())
|
||
# 禁止关闭主页
|
||
if widget.objectName() == "mainPage":
|
||
return
|
||
self.stackedWidget.removeWidget(widget)
|
||
self.tabBar.removeTab(index)
|
||
widget.deleteLater()
|
||
|
||
def open_fit_tab(self):
|
||
# 检查是否已存在标签页，避免重复添加
|
||
for i in range(self.stackedWidget.count()):
|
||
widget = self.stackedWidget.widget(i)
|
||
if widget.objectName() == "fitPage":
|
||
self.stackedWidget.setCurrentWidget(widget)
|
||
self.tabBar.setCurrentTab("fitPage")
|
||
return
|
||
fit_page = FunctionFitInterface(self)
|
||
self.addSubInterface(fit_page, "fitPage", "曲线拟合")
|
||
self.stackedWidget.setCurrentWidget(fit_page)
|
||
self.tabBar.setCurrentTab("fitPage")
|
||
|
||
def open_ai_tab(self):
|
||
# 检查是否已存在标签页，避免重复添加
|
||
for i in range(self.stackedWidget.count()):
|
||
widget = self.stackedWidget.widget(i)
|
||
if widget.objectName() == "aiPage":
|
||
self.stackedWidget.setCurrentWidget(widget)
|
||
self.tabBar.setCurrentTab("aiPage")
|
||
return
|
||
ai_page = AIInterface(self)
|
||
self.addSubInterface(ai_page, "aiPage", "AI问答")
|
||
self.stackedWidget.setCurrentWidget(ai_page)
|
||
self.tabBar.setCurrentTab("aiPage")
|