20250217 添加删除库功能

Front/main.py

@@ -38,6 +38,7 @@ from Front.modules import Settings
 from Front.widgets import *
 # main.py
 from Front.back.Program_Run.utils import resource_path  # 从 utils.py 导入 resource_path
+import gc
 
 # IMPORT / GUI AND MODULES AND WIDGETS
 # ///////////////////////////////////////////////////////////////
@@ -190,20 +191,75 @@ class ElTree(QWidget):
             widgets.allTreeWidget.insertTopLevelItems(0, treeWidgetItems)
 
     def contextMenuEvent(self, pos):
-        # 创建右键菜单
-        context_menu = QMenu(self)
-
-        # 添加删除选项
-        delete_action = QAction("删除", self)
-        delete_action.triggered.connect(self.delete_item)
-        context_menu.addAction(delete_action)
-
-        # 添加导出选项
-        export_action = QAction("导出", self)
-        export_action.triggered.connect(self.export_item)
-        context_menu.addAction(export_action)
+        try:
+            # 获取当前选中的项目
+            items = widgets.allTreeWidget.selectedItems()
+            item = items[0]
+            if item:
+                # 创建右键菜单
+                context_menu = QMenu(self)
+                if item.parent() is None:
+                    # 根节点菜单
+                    action = QAction("删除库", self)
+                    action.triggered.connect(self.delete_db)
+                    context_menu.addAction(action)
+                elif item.parent().parent() is None:
+                    # 父节点菜单
+                    action1 = QAction("删除", self)
+                    action1.triggered.connect(lambda: self.on_parent_action(item))
+                    context_menu.addAction(action1)
+                else:
+                    # 子节点菜单
+                    action = QAction("删除表", self)
+                    action.triggered.connect(self.delete_item)
+                    context_menu.addAction(action)
+                    action2 = QAction("导出", self)
+                    action2.triggered.connect(self.export_item)
+                    context_menu.addAction(action2)
+                # # 创建右键菜单
+                # context_menu = QMenu(self)
+                #
+                # # 添加删除选项
+                # delete_action = QAction("删除", self)
+                # delete_action.triggered.connect(self.delete_item)
+                # context_menu.addAction(delete_action)
+                #
+                # # 添加导出选项
+                # export_action = QAction("导出", self)
+                # export_action.triggered.connect(self.export_item)
+                # context_menu.addAction(export_action)
+                context_menu.exec(widgets.allTreeWidget.mapToGlobal(pos))
+        except:
+            pass
 
-        context_menu.exec(widgets.allTreeWidget.mapToGlobal(pos))
+    def delete_db(self):
+        # 获取当前选中的项目
+        selected_items = widgets.allTreeWidget.selectedItems()
+        item = selected_items[0]
+        dbname = item.text(0)
+        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+        # 确认删除操作
+        reply = QMessageBox.question(self, '确认删除', f'确定要删除项目 {dbname} 吗？',
+                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
+        if reply == QMessageBox.No:
+            return
+        try:
+            # 从 QTreeWidget 中移除节点
+            widgets.allTreeWidget.takeTopLevelItem(widgets.allTreeWidget.indexOfTopLevelItem(item))
+            try:
+                widgets.qureyTreeWidget.takeTopLevelItem(widgets.allTreeWidget.indexOfTopLevelItem(item))
+            except:
+                pass
+            # 强制垃圾回收
+            gc.collect()
+            os.remove(dbpath)
+            QMessageBox.information(self, '成功', f'项目 {dbname} 中的数据已删除')
+            # 刷新树状图
+            self.refresh_tree()
+            # 模拟点击 Enter 键  实现程序自动刷新的效果
+            QTest.keyClick(widgets.allTreeWidget, Qt.Key_Return)
+        except Exception as e:
+            QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
 
     def delete_item(self):
         # 获取当前选中的项目
@@ -364,20 +420,65 @@ class ElTree1(QWidget):
             widgets.qureyTreeWidget.insertTopLevelItems(0, treeWidgetItems)
 
     def contextMenuEvent(self, pos):
-        # 创建右键菜单
-        context_menu = QMenu(self)
-
-        # 添加删除选项
-        delete_action = QAction("删除", self)
-        delete_action.triggered.connect(self.delete_item)
-        context_menu.addAction(delete_action)
-
-        # 添加导出选项
-        export_action = QAction("导出", self)
-        export_action.triggered.connect(self.export_item)
-        context_menu.addAction(export_action)
+        # 获取当前选中的项目
+        try:
+            items = widgets.qureyTreeWidget.selectedItems()
+            item = items[0]
+            if item:
+                # 创建右键菜单
+                context_menu = QMenu(self)
+                if item.parent() is None:
+                    # 根节点菜单
+                    action = QAction("删除库", self)
+                    action.triggered.connect(self.delete_db)
+                    context_menu.addAction(action)
+                elif item.parent().parent() is None:
+                    # 父节点菜单
+                    action1 = QAction("删除", self)
+                    action1.triggered.connect(lambda: self.on_parent_action(item))
+                    context_menu.addAction(action1)
+                else:
+                    # 子节点菜单
+                    action = QAction("删除表", self)
+                    action.triggered.connect(self.delete_item)
+                    context_menu.addAction(action)
+                    action2 = QAction("导出", self)
+                    action2.triggered.connect(self.export_item)
+                    context_menu.addAction(action2)
+                context_menu.exec(widgets.qureyTreeWidget.mapToGlobal(pos))
+        except:
+            pass
 
-        context_menu.exec(widgets.qureyTreeWidget.mapToGlobal(pos))
+    def delete_db(self):
+        # 获取当前选中的项目
+        selected_items = widgets.qureyTreeWidget.selectedItems()
+        item = selected_items[0]
+        dbname = item.text(0)
+        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+        # 确认删除操作
+        reply = QMessageBox.question(self, '确认删除', f'确定要删除项目 {dbname} 吗？',
+                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
+        if reply == QMessageBox.No:
+            return
+        try:
+            # 从 QTreeWidget 中移除节点
+            widgets.allTreeWidget.takeTopLevelItem(widgets.allTreeWidget.indexOfTopLevelItem(item))
+            try:
+                widgets.qureyTreeWidget.takeTopLevelItem(widgets.allTreeWidget.indexOfTopLevelItem(item))
+            except:
+                pass
+            # 强制垃圾回收
+            gc.collect()
+            #删除路径
+            os.remove(dbpath)
+            QMessageBox.information(self, '成功', f'项目 {dbname} 中的数据已删除')
+            # 清空树状图
+            widgets.qureyTreeWidget.clear()
+            self.refresh_tree()
+            # 模拟点击 Enter 键  实现程序自动刷新的效果
+            QTest.keyClick(widgets.qureyTreeWidget, Qt.Key_Return)
+        except Exception as e:
+            QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
 
     def delete_item(self):
         # 获取当前选中的项目