添加ioc和自动任务

app/code_configuration_interface.py

@@ -2,8 +2,11 @@ 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
+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
@@ -56,6 +59,10 @@ class CodeConfigurationInterface(QWidget):
         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()
 
@@ -69,6 +76,7 @@ class CodeConfigurationInterface(QWidget):
         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):
@@ -88,6 +96,170 @@ class CodeConfigurationInterface(QWidget):
             )
         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: