20241227 实现行删除

Front/main.py

@@ -956,12 +956,6 @@ class MainWindow(QMainWindow):
             # 关联自定义的右键菜单方法到customContextMenuRequested信号
             self.ui.resultTableView1.customContextMenuRequested.connect(self.show_custom_context_menu)
 
-    # 右键对应行可删除对应行（只是视图上）
-    def seleceModel_itemclicked(self):
-        curIndex = self.selectModel.currentIndex()
-        self.ui.resultTableView1.model().removeRow(curIndex.row())
-        kk = 1
-
     # 搜索的item展示
     def itembuttonClick1(self):
         # 判定是否获取的是根节点
@@ -1009,6 +1003,28 @@ class MainWindow(QMainWindow):
             # 调用默认的 keyPressEvent 处理其他按键事件
             super(widgets.datainfo.__class__, widgets.datainfo).keyPressEvent(event)
 
+    # 右键对应行可删除对应行（所有的，搜索的再写一份，免得弄错）
+    def seleceModel_itemclicked(self):
+        selected_items = widgets.allTreeWidget.selectedItems()
+        item = selected_items[0]
+        data = item.data(0, Qt.UserRole + 1)
+        # 获取数据库路径和表名
+        dbname = data['listData'][0]
+        tablename = data['listData'][2]
+        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+        tablename_utf8 = tablename.encode('utf-8')
+        # 获取父节点
+        parent_item = item.parent()
+        parent_name = parent_item.text(0)
+        curIndex = self.selectModel.currentIndex()
+        # self.ui.resultTableView1.model().removeRow(curIndex.row())
+        col = curIndex.column()
+        row = curIndex.row()
+        # 重新组织表的内容，方便重新计算
+        UIFunctions.del_row_from_db(self, row, parent_name, tablename_utf8, dbpath)
+        # 1秒后自动跳转
+        QTimer.singleShot(1000, lambda: self.simulateButtonClick("btn_new"))
+
 
 if __name__ == "__main__":
     app = QApplication(sys.argv)

Front/modules/ui_functions.py

@@ -20,6 +20,7 @@ from PySide6.QtWidgets import QMessageBox
 
 from main import *
 import importlib
+import sqlite3
 
 from select import select
 
@@ -377,4 +378,89 @@ class UIFunctions(MainWindow):
         elif current_text == "平面控制网稳定性计算":
             selectModel = WDshow.search_show_function(self.ui, file_path, utf_en)
             return selectModel
+
+    # 删除行数据
+    def del_row_from_db(self, row, methodName, tableName_utf8, dbPath):
+        db1 = sqlite3.connect(dbPath)
+        cursor1 = db1.cursor()
+        # 通过行锁定对应的字段
+        if methodName == '水准测段高差稳定计算':
+            sqlstr1 = """SELECT * FROM GC_Input_Point WHERE TableName = ? LIMIT 1 OFFSET ?"""
+            cursor1.execute(sqlstr1, (tableName_utf8, row,))
+            result1 = cursor1.fetchall()
+            list1 = result1[0]
+            # 100%准确查询
+            # sqlstr3 = """SELECT * FROM GC_Input_Point  WHERE TableName = ? and New_SPName = ? and New_EPName = ? and New_HDiff = ? and New_RLen = ?"""
+            # cursor1.execute(sqlstr3, (tableName_utf8, list1[7], list1[8], list1[11], list1[12],))
+            # result3 = cursor1.fetchall()
+            sqlstr2 = """DELETE FROM GC_Input_Point  WHERE TableName = ? and New_SPName = ? and New_EPName = ? and New_HDiff = ? and New_RLen = ?"""
+            cursor1.execute(sqlstr2, (tableName_utf8, list1[7], list1[8], list1[11], list1[12],))
+            # # 获取删除的记录数量
+            # deleted_count = cursor1.rowcount
+            # 直接使用计算方式
+            tableName = tableName_utf8.decode('utf-8')
+            # 先把结果相关的点数据删一下
+            sqlstr3 = """DELETE FROM GC_Output_Point  WHERE TableName = ?"""
+            cursor1.execute(sqlstr3, (tableName_utf8,))
+            # 提交事务
+            db1.commit()
+            # 关闭db
+            db1.close()
+            GCcompute.main_function(tableName, dbPath)
+            GCshow.main_function(self.ui, dbPath, tableName_utf8)
+        elif methodName == '控制网复测平面基准计算':
+            sqlstr1 = """SELECT * FROM GS_Input_Point WHERE TableName = ? LIMIT 1 OFFSET ?"""
+            cursor1.execute(sqlstr1, (tableName_utf8, row,))
+            result1 = cursor1.fetchall()
+            list1 = result1[0]
+            # 100%准确查询
+            # sqlstr3 = """SELECT * FROM GC_Input_Point  WHERE TableName = ? and New_SPName = ? and New_EPName = ? and New_HDiff = ? and New_RLen = ?"""
+            # cursor1.execute(sqlstr3, (tableName_utf8, list1[7], list1[8], list1[11], list1[12],))
+            # result3 = cursor1.fetchall()
+            sqlstr2 = """DELETE FROM GS_Input_Point  WHERE TableName = ? and PointName = ? and New_X = ? and New_Y = ?"""
+            cursor1.execute(sqlstr2, (tableName_utf8, list1[7], list1[5], list1[6],))
+            # # 获取删除的记录数量
+            # deleted_count = cursor1.rowcount
+            # 直接使用计算方式
+            tableName = tableName_utf8.decode('utf-8')
+            # 先把结果相关的点和参数删一下
+            sqlstr3 = """DELETE FROM GS_Result_Point  WHERE TableName = ?"""
+            cursor1.execute(sqlstr3, (tableName_utf8,))
+            sqlstr4 = """DELETE FROM GS_Trans_Point  WHERE TableName = ?"""
+            cursor1.execute(sqlstr4, (tableName_utf8,))
+            sqlstr5 = """DELETE FROM GS_Trans_Param  WHERE TableName = ?"""
+            cursor1.execute(sqlstr5, (tableName_utf8,))
+            # 提交事务
+            db1.commit()
+            # 关闭db
+            db1.close()
+            GScompute.main_function(tableName, dbPath)
+            GSshow.main_function(self.ui, dbPath, tableName_utf8)
+        else:
+            # 平面控制网稳定性计算
+            sqlstr1 = """SELECT * FROM WD_Input_Point WHERE TableName = ? LIMIT 1 OFFSET ?"""
+            cursor1.execute(sqlstr1, (tableName_utf8, row,))
+            result1 = cursor1.fetchall()
+            list1 = result1[0]
+            # 100%准确查询
+            # sqlstr3 = """SELECT * FROM GC_Input_Point  WHERE TableName = ? and New_SPName = ? and New_EPName = ? and New_HDiff = ? and New_RLen = ?"""
+            # cursor1.execute(sqlstr3, (tableName_utf8, list1[7], list1[8], list1[11], list1[12],))
+            # result3 = cursor1.fetchall()
+            sqlstr2 = """DELETE FROM WD_Input_Point  WHERE TableName = ? and PointName = ? and New_X = ? and New_Y = ?"""
+            cursor1.execute(sqlstr2, (tableName_utf8, list1[11], list1[8], list1[9],))
+            # # 获取删除的记录数量
+            # deleted_count = cursor1.rowcount
+            # 直接使用计算方式
+            tableName = tableName_utf8.decode('utf-8')
+            # 先把结果相关的点和参数删一下
+            sqlstr3 = """DELETE FROM WD_Result_Point  WHERE TableName = ?"""
+            cursor1.execute(sqlstr3, (tableName_utf8,))
+            sqlstr4 = """DELETE FROM WD_Result_Param  WHERE TableName = ?"""
+            cursor1.execute(sqlstr4, (tableName_utf8,))
+            # 提交事务
+            db1.commit()
+            # 关闭db
+            db1.close()
+            WDcompute.main_function(tableName, dbPath)
+            WDshow.main_function(self.ui, dbPath, tableName_utf8)
     # END - GUI DEFINITIONS