20250217 添加父节点删除功能

Front/main.py

@@ -206,7 +206,7 @@ class ElTree(QWidget):
                 elif item.parent().parent() is None:
                     # 父节点菜单
                     action1 = QAction("删除", self)
-                    action1.triggered.connect(lambda: self.on_parent_action(item))
+                    action1.triggered.connect(self.delete_js)
                     context_menu.addAction(action1)
                 else:
                     # 子节点菜单
@@ -232,12 +232,55 @@ class ElTree(QWidget):
         except:
             pass
 
+    def delete_js(self):
+        #选中当前计算
+        selected_items = widgets.allTreeWidget.selectedItems()
+        item = selected_items[0]
+        parent_name = item.text(0)
+        # 获取父节点
+        db_item = item.parent()
+        if not db_item:
+            QMessageBox.warning(self, '警告', '所选项目没有父节点')
+            return
+        dbname = db_item.text(0)
+        dbpath = os.path.join(os.path.abspath('./SQL'), f"{dbname}.db")
+        # 确定要清空的表
+        tables_to_delete = []
+        if parent_name == '水准测段高差稳定计算':
+            tables_to_delete = ['GC_Input_Param', 'GC_Input_Point', 'GC_Output_Point']
+        elif parent_name == '控制网复测平面基准计算':
+            tables_to_delete = ['GS_Input_Param', 'GS_Input_Point', 'GS_Result_Point', 'GS_Trans_Param',
+                                'GS_Trans_Point']
+        elif parent_name == '平面控制网稳定性计算':
+            tables_to_delete = ['WD_Input_Param', 'WD_Input_Point', 'WD_Result_Param', 'WD_Result_Point']
+        else:
+            QMessageBox.warning(self, '警告', '未知的父节点')
+            return
+        # 确认删除操作
+        reply = QMessageBox.question(self, '确认删除', f'确定要删除所有 {parent_name} 的计算数据吗？',
+                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
+        if reply == QMessageBox.No:
+            return
+            # 清空表
+        try:
+            conn = sqlite3.connect(dbpath)
+            cursor = conn.cursor()
+            for table in tables_to_delete:
+                cursor.execute(f"DELETE FROM {table}")
+            conn.commit()
+            conn.close()
+            QMessageBox.information(self, '成功', f'计算 {parent_name} 中的数据已清空')
+            # 刷新树状图
+            self.refresh_tree()
+        except Exception as e:
+            QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
+
     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")
+        dbpath = os.path.join(os.path.abspath('./SQL'), f"{dbname}.db")
         # 确认删除操作
         reply = QMessageBox.question(self, '确认删除', f'确定要删除项目 {dbname} 吗？',
                                      QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
@@ -256,8 +299,8 @@ class ElTree(QWidget):
             QMessageBox.information(self, '成功', f'项目 {dbname} 中的数据已删除')
             # 刷新树状图
             self.refresh_tree()
-            # 模拟点击 Enter 键  实现程序自动刷新的效果
-            QTest.keyClick(widgets.allTreeWidget, Qt.Key_Return)
+            # # 模拟点击 Enter 键  实现程序自动刷新的效果
+            # QTest.keyClick(widgets.allTreeWidget, Qt.Key_Return)
         except Exception as e:
             QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
 
@@ -435,7 +478,7 @@ class ElTree1(QWidget):
                 elif item.parent().parent() is None:
                     # 父节点菜单
                     action1 = QAction("删除", self)
-                    action1.triggered.connect(lambda: self.on_parent_action(item))
+                    action1.triggered.connect(self.delete_js)
                     context_menu.addAction(action1)
                 else:
                     # 子节点菜单
@@ -449,12 +492,59 @@ class ElTree1(QWidget):
         except:
             pass
 
+    def delete_js(self):
+        #选中当前计算
+        selected_items = widgets.qureyTreeWidget.selectedItems()
+        item = selected_items[0]
+        parent_name = item.text(0)
+        # 获取父节点
+        db_item = item.parent()
+        if not db_item:
+            QMessageBox.warning(self, '警告', '所选项目没有父节点')
+            return
+        dbname = db_item.text(0)
+        dbpath = os.path.join(os.path.abspath('./SQL'), f"{dbname}.db")
+        # 确定要清空的表
+        tables_to_delete = []
+        if parent_name == '水准测段高差稳定计算':
+            tables_to_delete = ['GC_Input_Param', 'GC_Input_Point', 'GC_Output_Point']
+        elif parent_name == '控制网复测平面基准计算':
+            tables_to_delete = ['GS_Input_Param', 'GS_Input_Point', 'GS_Result_Point', 'GS_Trans_Param',
+                                'GS_Trans_Point']
+        elif parent_name == '平面控制网稳定性计算':
+            tables_to_delete = ['WD_Input_Param', 'WD_Input_Point', 'WD_Result_Param', 'WD_Result_Point']
+        else:
+            QMessageBox.warning(self, '警告', '未知的父节点')
+            return
+        # 确认删除操作
+        reply = QMessageBox.question(self, '确认删除', f'确定要删除所有 {parent_name} 的计算数据吗？',
+                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
+        if reply == QMessageBox.No:
+            return
+            # 清空表
+        try:
+            conn = sqlite3.connect(dbpath)
+            cursor = conn.cursor()
+            for table in tables_to_delete:
+                cursor.execute(f"DELETE FROM {table}")
+            conn.commit()
+            conn.close()
+            QMessageBox.information(self, '成功', f'计算 {parent_name} 中的数据已清空')
+            # 清空树状图
+            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_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")
+        dbpath = os.path.join(os.path.abspath('./SQL'), f"{dbname}.db")
         # 确认删除操作
         reply = QMessageBox.question(self, '确认删除', f'确定要删除项目 {dbname} 吗？',
                                      QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
@@ -475,8 +565,8 @@ class ElTree1(QWidget):
             # 清空树状图
             widgets.qureyTreeWidget.clear()
             self.refresh_tree()
-            # 模拟点击 Enter 键  实现程序自动刷新的效果
-            QTest.keyClick(widgets.qureyTreeWidget, Qt.Key_Return)
+            # # 模拟点击 Enter 键  实现程序自动刷新的效果
+            # QTest.keyClick(widgets.qureyTreeWidget, Qt.Key_Return)
         except Exception as e:
             QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
 
@@ -927,18 +1017,6 @@ class MainWindow(QMainWindow):
                 pid = id
                 id = id + 1
                 # 三种方法
-                sqlitem.append(TreeWidgetItem(id, pid, '水准测段高差稳定计算', icon=QIcon(
-                    os.path.join(self.absPath, "images/icons/cil-description.png"))))
-                gcid = id
-                id = id + 1
-                sqlitem.append(TreeWidgetItem(id, pid, '控制网复测平面基准计算', icon=QIcon(
-                    os.path.join(self.absPath, "images/icons/cil-description.png"))))
-                gsid = id
-                id = id + 1
-                sqlitem.append(TreeWidgetItem(id, pid, '平面控制网稳定性计算', icon=QIcon(
-                    os.path.join(self.absPath, "images/icons/cil-description.png"))))
-                wdid = id
-                id = id + 1
                 # 读取所有的表名（三种方式往下）
                 db1 = sqlite3.connect(dbpath)
                 # 获取游标
@@ -946,6 +1024,11 @@ class MainWindow(QMainWindow):
                 sqlstr1 = 'SELECT TableName FROM GC_Input_Param;'
                 cursor1.execute(sqlstr1)
                 result1 = cursor1.fetchall()
+                if len(result1) != 0:
+                    sqlitem.append(TreeWidgetItem(id, pid, '水准测段高差稳定计算', icon=QIcon(
+                        os.path.join(self.absPath, "images/icons/cil-description.png"))))
+                    gcid = id
+                    id = id + 1
                 for re1 in result1:
                     str1 = re1[0].decode('utf-8')
                     list1 = []
@@ -959,6 +1042,11 @@ class MainWindow(QMainWindow):
                 sqlstr2 = 'SELECT TableName FROM GS_Input_Param;'
                 cursor1.execute(sqlstr2)
                 result2 = cursor1.fetchall()
+                if len(result2) != 0:
+                    sqlitem.append(TreeWidgetItem(id, pid, '控制网复测平面基准计算', icon=QIcon(
+                        os.path.join(self.absPath, "images/icons/cil-description.png"))))
+                    gsid = id
+                    id = id + 1
                 for re2 in result2:
                     str2 = re2[0].decode('utf-8')
                     list2 = []
@@ -973,6 +1061,11 @@ class MainWindow(QMainWindow):
                 sqlstr3 = 'SELECT TableName FROM WD_Input_Param;'
                 cursor1.execute(sqlstr3)
                 result3 = cursor1.fetchall()
+                if len(result3) != 0:
+                    sqlitem.append(TreeWidgetItem(id, pid, '平面控制网稳定性计算', icon=QIcon(
+                        os.path.join(self.absPath, "images/icons/cil-description.png"))))
+                    wdid = id
+                    id = id + 1
                 for re3 in result3:
                     str3 = re3[0].decode('utf-8')
                     list3 = []