20241218 main_new.py 能识别点击数据的扩展数据

Front/main_new.py

@@ -13,6 +13,7 @@
 # https://doc.qt.io/qtforpython/licenses.html
 #
 # ///////////////////////////////////////////////////////////////
+
 from PySide6.QtWidgets import QFileDialog,QWidget, QVBoxLayout, QTreeWidget, QApplication, QTreeWidgetItem
 from PySide6.QtCore import Signal,  Qt, Slot, QObject
 from PySide6.QtSql import QSqlTableModel,QSqlDatabase
@@ -124,6 +125,83 @@ class ElTreeData(QObject):
         # 数据改变时发出信号
         self.styleSheet_changed.emit(self.styleSheet)
 
+#全数据库
+class ElTree(QWidget):
+    """
+    Control
+    """
+    itemClicked: Signal = Signal(object)
+    itemDoubleClicked: Signal = Signal(object)
+
+    def __init__(self, treeData: ElTreeData, parent=None):
+        super(ElTree, self).__init__(parent=parent)
+
+        self.data = treeData
+        # 将按钮的点击信号绑定到当前类的点击信号
+        widgets.allTreeWidget.itemClicked.connect(lambda item: self.itemClicked.emit(item.data(0, Qt.UserRole + 1)))
+        widgets.allTreeWidget.itemDoubleClicked.connect(lambda item: self.itemDoubleClicked.emit(item.data(0, Qt.UserRole + 1)))
+        self.__render_items(True)
+
+    def __render_items(self, is_clear: bool):
+        if is_clear:
+            widgets.allTreeWidget.clear()
+            widgets.qureyTreeWidget.clear()
+        widgets.allTreeWidget.setColumnCount(1)
+        widgets.allTreeWidget.setHeaderHidden(True)
+        widgets.qureyTreeWidget.setColumnCount(1)
+        widgets.qureyTreeWidget.setHeaderHidden(True)
+        if self.data.items is not None:
+            # 转为字典
+            mapping: dict[any, TreeWidgetItem] = dict(zip([i.id for i in self.data.items], self.data.items))
+            # 树容器
+            treeWidgetItems: list[QTreeWidgetItem] = []
+            for d in self.data.items:
+                # 如果找不到父级项，则是根节点
+                parent: TreeWidgetItem = mapping.get(d.parent_id)
+                if parent is None:
+                    treeWidgetItems.append(d.treeWidgetItem)
+                else:
+                    parent.treeWidgetItem.addChild(d.treeWidgetItem)
+
+            # 挂载到树上
+            widgets.allTreeWidget.insertTopLevelItems(0, treeWidgetItems)
+
+#查询
+class ElTree1(QWidget):
+    """
+    Control
+    """
+    itemClicked: Signal = Signal(object)
+    itemDoubleClicked: Signal = Signal(object)
+
+    def __init__(self, treeData: ElTreeData, parent=None):
+        super(ElTree1, self).__init__(parent=parent)
+        self.data = treeData
+        # 将按钮的点击信号绑定到当前类的点击信号
+        widgets.qureyTreeWidget.itemClicked.connect(lambda item: self.itemClicked.emit(item.data(0, Qt.UserRole + 1)))
+        widgets.qureyTreeWidget.itemDoubleClicked.connect(lambda item: self.itemDoubleClicked.emit(item.data(0, Qt.UserRole + 1)))
+        self.__render_items(True)
+
+    def __render_items(self, is_clear: bool):
+        if is_clear:
+            widgets.qureyTreeWidget.clear()
+        widgets.qureyTreeWidget.setColumnCount(1)
+        widgets.qureyTreeWidget.setHeaderHidden(True)
+        if self.data.items is not None:
+            # 转为字典
+            mapping: dict[any, TreeWidgetItem] = dict(zip([i.id for i in self.data.items], self.data.items))
+            # 树容器
+            treeWidgetItems: list[QTreeWidgetItem] = []
+            for d in self.data.items:
+                # 如果找不到父级项，则是根节点
+                parent: TreeWidgetItem = mapping.get(d.parent_id)
+                if parent is None:
+                    treeWidgetItems.append(d.treeWidgetItem)
+                else:
+                    parent.treeWidgetItem.addChild(d.treeWidgetItem)
+
+            # 挂载到树上
+            widgets.qureyTreeWidget.insertTopLevelItems(0, treeWidgetItems)
 
 class MainWindow(QMainWindow):
     def __init__(self):
@@ -224,81 +302,19 @@ class MainWindow(QMainWindow):
         widgets.stackedWidget.setCurrentWidget(widgets.home)
         widgets.btn_home.setStyleSheet(UIFunctions.selectMenu(widgets.btn_home.styleSheet()))
 
-        #初始化树状组件
-        #///////////////////////////////////////////////////////////////
-        self.itemClicked: Signal = Signal(object)
-        self.itemDoubleClicked: Signal = Signal(object)
-        sqll = self.sql_init()
-        sqltree = ElTreeData(items=sqll)
-        self.data = sqltree
-        # 将按钮的点击信号绑定到当前类的点击信号
-        # 信号改为展示方法，这里同导出展示
-        # widgets.allTreeWidget.itemClicked.connect(lambda extend: print("单击->扩展数据为：", extend))
-        widgets.allTreeWidget.itemClicked.connect(self.test_method)
-        widgets.allTreeWidget.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
-        self.__render_items(True)
+        #初始化树状数据库
+        tree_button = self.sql_init()
+        tree_button.itemClicked.connect(lambda extend: print("单击->扩展数据为：", extend))
+        tree_button.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
 
         self.bind()
 
-    def test_method(self):
-        print('ok')
-
     #绑定组件
     def bind(self):
         # 计算
         widgets.compute.clicked.connect(self.computeClick)
         widgets.search.clicked.connect(self.searchClick)
 
-    #全部
-    def __render_items(self, is_clear: bool):
-        if is_clear:
-            widgets.allTreeWidget.clear()
-            widgets.qureyTreeWidget.clear()
-        widgets.allTreeWidget.setColumnCount(1)
-        widgets.allTreeWidget.setHeaderHidden(True)
-        widgets.qureyTreeWidget.setColumnCount(1)
-        widgets.qureyTreeWidget.setHeaderHidden(True)
-        if self.data.items is not None:
-            # 转为字典
-            mapping: dict[any, TreeWidgetItem] = dict(zip([i.id for i in self.data.items], self.data.items))
-            # 树容器
-            treeWidgetItems: list[QTreeWidgetItem] = []
-            for d in self.data.items:
-                # 如果找不到父级项，则是根节点
-                parent: TreeWidgetItem = mapping.get(d.parent_id)
-                if parent is None:
-                    treeWidgetItems.append(d.treeWidgetItem)
-                else:
-                    parent.treeWidgetItem.addChild(d.treeWidgetItem)
-
-            # 挂载到树上
-            widgets.allTreeWidget.insertTopLevelItems(0, treeWidgetItems)
-
-    #搜索
-    def __render_items1(self, is_clear: bool):
-        if is_clear:
-            widgets.qureyTreeWidget.clear()
-        widgets.qureyTreeWidget.setColumnCount(1)
-        widgets.qureyTreeWidget.setHeaderHidden(True)
-        if self.data.items is not None:
-            # 转为字典
-            mapping: dict[any, TreeWidgetItem] = dict(zip([i.id for i in self.data.items], self.data.items))
-            # 树容器
-            treeWidgetItems: list[QTreeWidgetItem] = []
-            for d in self.data.items:
-                # 如果找不到父级项，则是根节点
-                parent: TreeWidgetItem = mapping.get(d.parent_id)
-                if parent is None:
-                    treeWidgetItems.append(d.treeWidgetItem)
-                else:
-                    parent.treeWidgetItem.addChild(d.treeWidgetItem)
-
-            # 挂载到树上
-            widgets.qureyTreeWidget.insertTopLevelItems(0, treeWidgetItems)
-
-
-
-
     #删除tableview
     def delete_table_view(table_view):
         table_view.setParent(None)
@@ -582,38 +598,40 @@ class MainWindow(QMainWindow):
             result1 = cursor1.fetchall()
             for re1 in result1:
                 str1 = re1[0].decode('utf-8')
-                sqlitem.append(TreeWidgetItem(id, gcid, str1, icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png")))
                 list1 = []
-                list1.append(dbname)
+                list1.append(dbpath)
                 list1.append('水准测段高差稳定计算')
                 list1.append(str1)
                 self.dblist.append(list1)
+                sqlitem.append(TreeWidgetItem(id, gcid, str1, icon=QIcon(
+                    "D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png"),extend={'listData': list1}))
                 id = id + 1
             sqlstr2 = 'SELECT TableName FROM GS_Input_Param;'
             cursor1.execute(sqlstr2)
             result2 = cursor1.fetchall()
             for re2 in result2:
                 str2 = re2[0].decode('utf-8')
-                sqlitem.append(TreeWidgetItem(id, gsid, str2, icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png")))
                 list2 = []
-                list2.append(dbname)
+                list2.append(dbpath)
                 list2.append('控制网复测平面基准计算')
                 list2.append(str2)
                 self.dblist.append(list2)
+                sqlitem.append(TreeWidgetItem(id, gsid, str2, icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png"),extend={'listData': list2}))
                 id = id + 1
             sqlstr3 = 'SELECT TableName FROM WD_Input_Param;'
             cursor1.execute(sqlstr3)
             result3 = cursor1.fetchall()
             for re3 in result3:
                 str3 = re3[0].decode('utf-8')
-                sqlitem.append(TreeWidgetItem(id, wdid, str3, icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png")))
                 list3 = []
-                list3.append(dbname)
+                list3.append(dbpath)
                 list3.append('平面控制网稳定性计算')
                 list3.append(str3)
                 self.dblist.append(list3)
+                sqlitem.append(TreeWidgetItem(id, wdid, str3, icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png"),extend={'listData': list3}))
                 id = id + 1
-        return sqlitem
+        button = ElTree(ElTreeData(sqlitem))
+        return button
 
     #关键词查询
     def searchClick(self):
@@ -666,23 +684,16 @@ class MainWindow(QMainWindow):
             zdy.append(zdy1)
             id = id + 1
             sqlitem.append(TreeWidgetItem(id, fid, list2[2], icon=QIcon(
-                "D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png"), extend=list2[2]))
+                "D:/4work_now/20240819GS/20241211/Front/images/icons/cil-file.png"), extend={'listData': list2}))
             zdy1 = []
             zdy1.append(id)
             zdy1.append(fid)
             zdy1.append(list2[2])
             zdy.append(zdy1)
             id = id + 1
-        #展示
-        self.itemClicked: Signal = Signal(object)
-        self.itemDoubleClicked: Signal = Signal(object)
-        sqltree = ElTreeData(items=sqlitem)
-        self.data = sqltree
-        # 将按钮的点击信号绑定到当前类的点击信号
-        # 信号改为展示方法，这里同导出展示
-        self.ui.qureyTreeWidget.itemClicked.connect(lambda extend: print("单击->扩展数据为：", extend))
-        self.ui.qureyTreeWidget.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
-        self.__render_items1(True)
+        search_button = ElTree1(ElTreeData(sqlitem))
+        search_button.itemClicked.connect(lambda extend: print("单击->扩展数据为：", extend))
+        search_button.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
 
     #直接这个基础上改点选输出表格（未改）
     def computeClick(self):