20241216 main_new.py初始化全数据库树状展示（点击未实现），查询展示未写完

Front/main_new.py

@@ -14,7 +14,7 @@
 #
 # ///////////////////////////////////////////////////////////////
 from PySide6.QtWidgets import QFileDialog,QWidget, QVBoxLayout, QTreeWidget, QApplication, QTreeWidgetItem
-from PySide6.QtCore import Signal, Slot,Qt,QObject
+from PySide6.QtCore import Signal,  Qt, Slot, QObject
 from PySide6.QtSql import QSqlTableModel,QSqlDatabase
 import sqlite3
 from PySide6.QtGui import QIcon
@@ -94,8 +94,6 @@ class ElTreeData(QObject):
     """
     items_changed: Signal = Signal(str)
     styleSheet_changed: Signal = Signal(str)
-    itemClicked: Signal = Signal(object)
-    itemDoubleClicked: Signal = Signal(object)
  
     def __init__(self, items: list[TreeWidgetItem] = None, styleSheet: str = None):
         super(ElTreeData, self).__init__()
@@ -175,7 +173,7 @@ class MainWindow(QMainWindow):
         # 成果预览
         widgets.btn_new.clicked.connect(self.buttonClick)
         # 数据一览
-        
+        widgets.btn_data.clicked.connect(self.buttonClick)
         # 皮肤切换
         widgets.btn_message.clicked.connect(self.buttonClick)
         # 打开上传文件夹
@@ -194,6 +192,9 @@ class MainWindow(QMainWindow):
 
         widgets.settingsTopBtn.clicked.connect(openCloseRightBox)
 
+
+
+
         # 展示APP
         # ///////////////////////////////////////////////////////////////
         self.show()
@@ -223,13 +224,21 @@ 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.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
+        self.__render_items(True)
+
         self.bind()
 
-        #treewidget
-        self.data = ElTreeData
-        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:
@@ -257,8 +266,7 @@ class MainWindow(QMainWindow):
     def bind(self):
         # 计算
         widgets.compute.clicked.connect(self.computeClick)
-        #数据库展示（初始化）
-        widgets.btn_data.clicked.connect(self.dataClick)
+        widgets.search.clicked.connect(self.searchClick)
 
     #删除tableview
     def delete_table_view(table_view):
@@ -508,23 +516,33 @@ class MainWindow(QMainWindow):
             xx = xx + 1 
         return model
 
-    
-
-    def dataClick(self):
-        # GET BUTTON CLICKED
-        btn = self.sender()
-        btnName = btn.objectName()
+    #设一个全局变量，存数据库中所有的包含内容（数据库-三种方法-表）
+    dblist = []
+    global dblist
+    #初始化数据一览（数据库全展示）
+    def sql_init(self):
         #初始化全部数据库
         inpath = r'D:\4work_now\20240819GS\20241211\SQL'
         #读取所有的数据库名
-        list1 = []
+        sqlitem = []
+        id = 1
         for filename in os.listdir(inpath):
+            #数据库
             dbname = filename.split('.',-1)[0]
             dbpath = os.path.join(inpath,filename)
-            lista = []
-            lista.append(dbname)
-            lista.append(dbpath)
-            list1.append(lista)
+            sqlitem.append(TreeWidgetItem(id,0,dbname,icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-clone.png")))
+            pid = id
+            id = id + 1
+            #三种方法
+            sqlitem.append(TreeWidgetItem(id, pid, '水准测段高差稳定计算', icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-description.png")))
+            gcid = id
+            id = id + 1
+            sqlitem.append(TreeWidgetItem(id, pid, '控制网复测平面基准计算', icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-description.png")))
+            gsid = id
+            id = id + 1
+            sqlitem.append(TreeWidgetItem(id, pid, '平面控制网稳定性计算', icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil-description.png")))
+            wdid = id
+            id = id + 1
             #读取所有的表名（三种方式往下）
             db1 = sqlite3.connect(dbpath)
             #获取游标
@@ -532,31 +550,90 @@ class MainWindow(QMainWindow):
             sqlstr1 = 'SELECT TableName FROM GC_Input_Param;'
             cursor1.execute(sqlstr1)
             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"),extend=str1))
+                list1 = []
+                list1.append(dbname)
+                list1.append('水准测段高差稳定计算')
+                list1.append(str1)
+                dblist.append(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"),extend=str2))
+                list2 = []
+                list2.append(dbname)
+                list2.append('控制网复测平面基准计算')
+                list2.append(str2)
+                dblist.append(list2)
+                id = id + 1
             sqlstr3 = 'SELECT TableName FROM WD_Input_Param;'
             cursor1.execute(sqlstr3)
             result3 = cursor1.fetchall()
-        atw_button = ElTreeData(items=[
-        TreeWidgetItem(1, 0, "User", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 1}),
-        TreeWidgetItem(2, 1, "Child", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 2}),
-        TreeWidgetItem(3, 1, "Child", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 3}),
-        TreeWidgetItem(4, 2, "Child", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 4}),
-        TreeWidgetItem(5, 2, "Child", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 5}),
-        TreeWidgetItem(6, 5, "Child", icon=QIcon("D:/4work_now/20240819GS/20241211/Front/images/icons/cil_file.png"), extend={'id': 6}),
-        TreeWidgetItem(7, 5, "Child", extend={'id': 7}),
-        TreeWidgetItem(8, 5, "Child", extend={'id': 8}),
-        TreeWidgetItem(9, 8, "Child", extend={'id': 9}),
-        ])
-    
-        # atw_button.itemClicked.connect(lambda extend: print("单击->扩展数据为：", extend))
-        # atw_button.itemDoubleClicked.connect(lambda extend: print("双击->扩展数据为：", extend))
-        # atw_button.show()
-        widgets.stackedWidget.setCurrentWidget(widgets.datainfo)
-        UIFunctions.resetStyle(self, btnName)  # RESET ANOTHERS BUTTONS SELECTED
-        btn.setStyleSheet(UIFunctions.selectMenu(btn.styleSheet()))
-
+            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"),extend=str3))
+                list3 = []
+                list3.append(dbname)
+                list3.append('平面控制网稳定性计算')
+                list3.append(str3)
+                dblist.append(list3)
+                id = id + 1
+        return sqlitem
+
+    #关键词查询
+    def sql_search(self):
+        # GET BUTTON CLICKED
+        btn = self.sender()
+        btnName = btn.objectName()
+        #获取文本
+        text1 = self.ui.lineEdit_2.text()
+        #符合的list
+        fhlist = []
+        #查询(待优化)
+        for list1 in dblist:
+            for li in list1:
+                if text1 in li:
+                    fhlist.append(list1)
+                    break
+        #更新item值
+        sqlitem = []
+        #前两层是否一致
+        str1 = ''
+        str2 = ''
+        id = 1
+        pid = 0
+        fid = 0
+        for list2 in fhlist:
+            dbname = list2[0]
+            if dbname != str1:
+                str1 = dbname
+                sqlitem.append(TreeWidgetItem(id, 0, dbname, icon=QIcon(
+                    "D:/4work_now/20240819GS/20241211/Front/images/icons/cil-clone.png")))
+                pid = id
+            id = id + 1
+            mename = list2[1]
+            if mename != str2:
+                str2 = mename
+                sqlitem.append(TreeWidgetItem(id, pid, mename, icon=QIcon(
+                    "D:/4work_now/20240819GS/20241211/Front/images/icons/cil-description.png")))
+                fid = id
+            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]))
+        #展示
+
+
+
+
+
+
+
+    #直接这个基础上改点选输出表格（未改）
     def computeClick(self):
         # GET BUTTON CLICKED
         btn = self.sender()