ControlNetwork/Front/main.py

# ///////////////////////////////////////////////////////////////
#
# BY: WANDERSON M.PIMENTA
# PROJECT MADE WITH: Qt Designer and PySide6
# V: 1.0.0
#
# This project can be used freely for all uses, as long as they maintain the
# respective credits only in the Python scripts, any information in the visual
# interface (GUI) can be modified without any implication.
#
# There are limitations on Qt licenses if you want to use your products
# commercially, I recommend reading them on the official website:
# https://doc.qt.io/qtforpython/licenses.html
#
# ///////////////////////////////////////////////////////////////
import time

from PySide6.QtWidgets import QFileDialog, QWidget, QVBoxLayout, QTreeWidget, QApplication, QTreeWidgetItem
from PySide6.QtCore import Signal, Qt, Slot, QObject
from PySide6.QtSql import QSqlTableModel, QSqlDatabase
import sqlite3
from PySide6.QtGui import QIcon

import sys
import os
import platform

import Back.Program_Run.database_operations
# IMPORT / GUI AND MODULES AND WIDGETS
# ///////////////////////////////////////////////////////////////
from modules import *
from widgets import *

os.environ["QT_FONT_DPI"] = "96"  # FIX Problem for High DPI and Scale above 100%

# SET AS GLOBAL WIDGETS
# ///////////////////////////////////////////////////////////////
widgets = None
excelname = ''

# 获取项目根目录
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
# 将项目根目录添加到 sys.path
sys.path.append(project_root)


# 表格的模型
class MyTableModel(QAbstractTableModel):
    def __init__(self, data):
        super().__init__()
        self._data = data

    # 重新定义行数
    def rowCount(self, parent=QModelIndex()):
        return len(self._data)

    # 重新定义列数
    def columnCount(self, parent=QModelIndex()):
        return len(self._data[0]) if self._data else 0

    # 重新定义数据
    def data(self, index, role=Qt.DisplayRole):
        if not index.isValid():
            return None
        if role == Qt.DisplayRole:
            return self._data[index.row()][index.column()]
        return None


# 数据一览树状
class TreeWidgetItem:
    def __init__(self, id: any, parent_id: any, name: str, icon: QIcon = None, extend: object = None):
        """
        菜单数据接口
        :param id: ID
        :param parent_id: 父ID
        :param name: 菜单名称
        :param icon: 图标
        :param extend: 扩展数据
        """
        self.id: any = id
        self.parent_id: any = parent_id
        self.name: str = name
        self.extend: object = extend
        # 实例化
        self.treeWidgetItem = QTreeWidgetItem([self.name])
        # 存储相关数据
        self.treeWidgetItem.setData(0, Qt.UserRole + 1, extend)
        self.treeWidgetItem.setIcon(0, QIcon(':/icons/default.png'))
        if icon is not None:
            self.treeWidgetItem.setIcon(0, icon)


class ElTreeData(QObject):
    """
    Data Model
    """
    items_changed: Signal = Signal(str)
    styleSheet_changed: Signal = Signal(str)

    def __init__(self, items: list[TreeWidgetItem] = None, styleSheet: str = None):
        super(ElTreeData, self).__init__()
        # 定义数据
        self._items: list[TreeWidgetItem]
        self._styleSheet: str
        # 初始化数据
        self.items = items
        self.styleSheet = styleSheet

    @property
    def items(self):
        return self._items

    @items.setter
    def items(self, value):
        self._items = value
        # 数据改变时发出信号
        self.items_changed.emit(self.items)

    @property
    def styleSheet(self):
        return self._styleSheet

    @styleSheet.setter
    def styleSheet(self, value):
        self._styleSheet = value
        # 数据改变时发出信号
        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):
        QMainWindow.__init__(self)
        self.file_path = None  # 初始化时文件路径设置为None
        # super().__init__()
        # SET AS GLOBAL WIDGETS
        # ///////////////////////////////////////////////////////////////
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        # 连接信号与槽函数，实现点击操作
        self.ui.createFile.clicked.connect(self.on_create_file_clicked)
        self.ui.export_2.clicked.connect(self.on_export_2_clicked)
        # self.comboBox_2 = QComboBox(self)
        # ...此处为省略代码...
        global widgets
        widgets = self.ui
        widgets.search.clicked.connect(self.buttonClick)
        # 设置 datainfo 页面的焦点策略，使其能够捕获键盘事件
        widgets.datainfo.setFocusPolicy(Qt.StrongFocus)
        widgets.datainfo.keyPressEvent = self.datainfo_keyPressEvent
        # 是否使用系统自带标题栏  True是不使用，False使用
        # ///////////////////////////////////////////////////////////////
        Settings.ENABLE_CUSTOM_TITLE_BAR = True

        # APP名称
        # ///////////////////////////////////////////////////////////////
        title = "控制网复测平面基准归算程序"
        description = "控制网复测平面基准归算程序"
        # APPLY TEXTS
        self.setWindowTitle(title)
        widgets.titleRightInfo.setText(description)

        # TOGGLE MENU
        # ///////////////////////////////////////////////////////////////
        widgets.toggleButton.clicked.connect(lambda: UIFunctions.toggleMenu(self, True))

        # SET UI DEFINITIONS
        # ///////////////////////////////////////////////////////////////
        UIFunctions.uiDefinitions(self)

        # QTableWidget PARAMETERS
        # ///////////////////////////////////////////////////////////////
        # widgets.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        # 点击事件
        # ///////////////////////////////////////////////////////////////

        # 左边菜单
        # 首页
        widgets.btn_home.clicked.connect(self.buttonClick)
        # 输入表格
        widgets.btn_widgets.clicked.connect(self.buttonClick)
        # 成果预览
        widgets.btn_new.clicked.connect(self.buttonClick)
        # 数据一览
        widgets.btn_data.clicked.connect(self.buttonClick)
        # 皮肤切换
        widgets.btn_message.clicked.connect(self.buttonClick)
        # 打开上传文件夹
        widgets.upload.clicked.connect(self.buttonClick)
        # 文件计算
        widgets.compute.clicked.connect(self.buttonClick)
        # 数据一览的搜索键
        widgets.search.clicked.connect(self.buttonClick)

        # 拓展左侧栏
        # def openCloseLeftBox():
        #     UIFunctions.toggleLeftBox(self, True)
        # widgets.toggleLeftBox.clicked.connect(openCloseLeftBox)
        # widgets.extraCloseColumnBtn.clicked.connect(openCloseLeftBox)

        # EXTRA RIGHT BOX
        def openCloseRightBox():
            UIFunctions.toggleRightBox(self, True)

        widgets.settingsTopBtn.clicked.connect(openCloseRightBox)

        # 展示APP
        # ///////////////////////////////////////////////////////////////
        self.show()

        # 设置主题
        # ///////////////////////////////////////////////////////////////
        if getattr(sys, 'frozen', False):
            absPath = os.path.dirname(os.path.abspath(sys.executable))
        elif __file__:
            absPath = os.path.dirname(os.path.abspath(__file__))
        useCustomTheme = True
        self.useCustomTheme = useCustomTheme
        self.absPath = absPath
        themeFile = r"themes\py_dracula_light.qss"

        # 设置主题和HACKS
        if useCustomTheme:
            # LOAD AND APPLY STYLE
            UIFunctions.theme(self, themeFile, True)

            # SET HACKS
            AppFunctions.setThemeHack(self)

        # 设置主页和选择菜单
        # ///////////////////////////////////////////////////////////////
        widgets.stackedWidget.setCurrentWidget(widgets.home)
        widgets.btn_home.setStyleSheet(UIFunctions.selectMenu(widgets.btn_home.styleSheet()))
        # self.bind()

        # 初始化树状数据库
        tree_button = self.sql_init()
        tree_button.itemClicked.connect(self.itembuttonClick)
        tree_button.itemDoubleClicked.connect(self.itembuttonClick)

    # 绑定组件
    def bind(self):
        # 计算
        widgets.compute.clicked.connect(self.buttonClick)

    # 删除tableview
    def delete_table_view(table_view):
        table_view.setParent(None)
        table_view.deleteLater()

    # ///////////////////////////////////////////////////////////////
    def buttonClick(self):
        # GET BUTTON CLICKED
        btn = self.sender()
        btnName = btn.objectName()

        # 首页
        if btnName == "btn_home":
            widgets.stackedWidget.setCurrentWidget(widgets.home)
            UIFunctions.resetStyle(self, btnName)
            btn.setStyleSheet(UIFunctions.selectMenu(btn.styleSheet()))

        # 输入表格
        if btnName == "btn_widgets":
            widgets.stackedWidget.setCurrentWidget(widgets.widgets)
            UIFunctions.resetStyle(self, btnName)
            btn.setStyleSheet(UIFunctions.selectMenu(btn.styleSheet()))

        # 成果预览
        if btnName == "btn_new":
            widgets.stackedWidget.setCurrentWidget(widgets.new_page)  # SET PAGE
            UIFunctions.resetStyle(self, btnName)  # RESET ANOTHERS BUTTONS SELECTED
            btn.setStyleSheet(UIFunctions.selectMenu(btn.styleSheet()))  # SELECT MENU

        # 数据一览
        if btnName == "btn_data":
            widgets.stackedWidget.setCurrentWidget(widgets.datainfo)  # SET PAGE
            UIFunctions.resetStyle(self, btnName)  # RESET ANOTHERS BUTTONS SELECTED
            btn.setStyleSheet(UIFunctions.selectMenu(btn.styleSheet()))  # SELECT MENU

        # 皮肤切换
        if btnName == "btn_message":
            if self.useCustomTheme:
                themeFile = os.path.abspath(os.path.join(self.absPath, "themes/py_dracula_dark.qss"))
                UIFunctions.theme(self, themeFile, True)
                # SET HACKS
                AppFunctions.setThemeHack(self)
                self.useCustomTheme = False
            else:
                themeFile = os.path.abspath(os.path.join(self.absPath, "themes/py_dracula_light.qss"))
                UIFunctions.theme(self, themeFile, True)
                # SET HACKS
                AppFunctions.setThemeHack(self)
                self.useCustomTheme = True
        # 文件上传
        if btnName == "upload":
            # 使用 QFileDialog 获取文件路径
            file_path, _ = QFileDialog.getOpenFileName(
                self,
                "选择文件",
                "",
                "表格文件 (*.xls *.xlsx)"
                #  水准后期添加支持.xls文件
            )
            if file_path:
                # 处理选中的文件
                self.file_path = file_path  # 将获取的路径保存到类的属性中
                UIFunctions.execute_script_based_on_selection(self, file_path)

        # 文件计算
        if btnName == "compute":
            if self.file_path:  # 确保有文件路径后再进行处理
                # 计算与展示的业务代码
                UIFunctions.compute_show_process_excel_file(self, self.file_path)
                # 1秒后自动跳转
                QTimer.singleShot(1000, lambda: self.simulateButtonClick("btn_new"))
            else:
                QMessageBox.warning(self, '警告', '请先选择项目并上传文件')

        # 树状查询
        if btnName == "search":
            self.searchClick()

        # 输出点击回馈
        print(f'Button "{btnName}" pressed!')

    # 辅助函数，点击计算后自动跳转页面
    def simulateButtonClick(self, btnName):
        # 根据按钮名称找到对应的按钮
        btn = self.findChild(QObject, btnName)
        if btn:
            # 触发按钮点击事件
            btn.click()

    # 新增槽函数来调用 create_database_and_tables
    def on_create_file_clicked(self):
        Back.Program_Run.database_operations.create_database_and_tables(self, self.ui)

    # 定义导出数据库的槽函数
    def on_export_2_clicked(self):
        if self.file_path:
            UIFunctions.export_database_to_excel(self, self.file_path)
        else:
            QMessageBox.warning(self, '警告', '请先选择项目并上传文件')

    # RESIZE EVENTS
    # ///////////////////////////////////////////////////////////////
    def resizeEvent(self, event):
        # Update Size Grips
        UIFunctions.resize_grips(self)

    # 鼠标点击事件
    # ///////////////////////////////////////////////////////////////
    def mousePressEvent(self, event):
        # SET DRAG POS WINDOW
        # self.dragPos = event.globalPos()
        self.dragPos = event.globalPosition().toPoint()

        # 输出鼠标事件
        if event.buttons() == Qt.LeftButton:
            print('Mouse click: LEFT CLICK')
        if event.buttons() == Qt.RightButton:
            print('Mouse click: RIGHT CLICK')

    # 设一个全局变量，存数据库中所有的包含内容（数据库-三种方法-表）
    dblist = []

    # 初始化数据一览（数据库全展示）
    def sql_init(self):
        # 初始化全部数据库
        # inpath = r'D:\4work_now\20240819GS\20241211\SQL'
        inpath = os.path.abspath('../SQL')
        # 读取所有的数据库名
        sqlitem = []
        id = 1
        for filename in os.listdir(inpath):
            # 数据库
            dbname = filename.split('.', -1)[0]
            dbpath = os.path.join(inpath, filename)
            sqlitem.append(TreeWidgetItem(id, 0, dbname, icon=QIcon(
                os.path.abspath(os.path.join(self.absPath, "images/icons/cil-clone.png")))))
            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)
            # 获取游标
            cursor1 = db1.cursor()
            sqlstr1 = 'SELECT TableName FROM GC_Input_Param;'
            cursor1.execute(sqlstr1)
            result1 = cursor1.fetchall()
            for re1 in result1:
                str1 = re1[0].decode('utf-8')
                list1 = []
                list1.append(dbname)
                list1.append('水准测段高差稳定计算')
                list1.append(str1)
                self.dblist.append(list1)
                sqlitem.append(TreeWidgetItem(id, gcid, str1, icon=QIcon(
                    os.path.join(self.absPath, "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')
                list2 = []
                list2.append(dbname)
                list2.append('控制网复测平面基准计算')
                list2.append(str2)
                self.dblist.append(list2)
                sqlitem.append(TreeWidgetItem(id, gsid, str2,
                                              icon=QIcon(os.path.join(self.absPath, "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')
                list3 = []
                list3.append(dbname)
                list3.append('平面控制网稳定性计算')
                list3.append(str3)
                self.dblist.append(list3)
                sqlitem.append(TreeWidgetItem(id, wdid, str3,
                                              icon=QIcon(os.path.join(self.absPath, "images/icons/cil-file.png")),
                                              extend={'listData': list3}))
                id = id + 1
        button = ElTree(ElTreeData(sqlitem))
        return button

    # 关键词查询
    def searchClick(self):
        # GET BUTTON CLICKED
        btn = self.sender()
        btnName = btn.objectName()
        # 获取文本
        text1 = self.ui.lineEdit_2.text()
        # 符合的list
        fhlist = []
        # 查询(待优化)
        for list1 in self.dblist:
            for li in list1:
                if text1 in li:
                    fhlist.append(list1)
                    break
        # 更新item值
        sqlitem = []
        # 前两层是否一致
        str1 = ''
        str2 = ''
        id = 1
        pid = 0
        fid = 0
        zdy = []
        for list2 in fhlist:
            dbname = list2[0]
            if dbname != str1:
                str1 = dbname
                pid = id
                str2 = ''
                sqlitem.append(TreeWidgetItem(id, 0, dbname, icon=QIcon(
                    os.path.join(self.absPath, "images/icons/cil-clone.png"))))
            zdy1 = []
            zdy1.append(id)
            zdy1.append(0)
            zdy1.append(dbname)
            zdy.append(zdy1)
            id = id + 1
            mename = list2[1]
            if mename != str2:
                str2 = mename
                fid = id
                sqlitem.append(TreeWidgetItem(id, pid, mename, icon=QIcon(
                    os.path.join(self.absPath, "images/icons/cil-description.png"))))
            zdy1 = []
            zdy1.append(id)
            zdy1.append(pid)
            zdy1.append(mename)
            zdy.append(zdy1)
            id = id + 1
            sqlitem.append(TreeWidgetItem(id, fid, list2[2], icon=QIcon(
                os.path.join(self.absPath, "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
        search_button = ElTree1(ElTreeData(sqlitem))
        search_button.itemClicked.connect(self.itembuttonClick1)
        search_button.itemDoubleClicked.connect(self.itembuttonClick1)
        # 隐藏默认label
        self.ui.defaultLabel.setVisible(False)
    # 全树的item展示
    def itembuttonClick(self):
        # 判定是否获取的是根节点
        pd = 1
        select_item = self.ui.allTreeWidget.currentItem()
        # 获取点击的item的text和它对应的上两个节点
        str1 = select_item.text(0)
        current_text = ''  # 初始化 current_text
        str3 = ''  # 初始化 str3
        try:
            pa_item = select_item.parent()
            current_text = pa_item.text(0)
            z_item = pa_item.parent()
            str3 = z_item.text(0)
        except:
            pd = -1
        if current_text == '':
            pd = -1
        if str3 == '':
            pd = -1
        if pd == -1:
            pass
        else:
            # 数据库目录
            inpath = os.path.abspath('../SQL')
            file_path = inpath + '/' + str3 + '.db'
            # 数据库路径，哪种方法，表名
            UIFunctions.search_data_to_show(self, file_path, current_text, str1)

    # 搜索的item展示
    def itembuttonClick1(self):
        # 判定是否获取的是根节点
        pd = 1
        select_item = self.ui.qureyTreeWidget.currentItem()
        # 获取点击的item的text和它对应的上两个节点
        str1 = select_item.text(0)
        current_text = ''  # 初始化 current_text
        str3 = ''  # 初始化 str3
        try:
            pa_item = select_item.parent()
            current_text = pa_item.text(0)
            z_item = pa_item.parent()
            str3 = z_item.text(0)
        except:
            pd = -1
        if current_text == '':
            pd = -1
        if str3 == '':
            pd = -1
        if pd == -1:
            pass
        else:
            # 数据库目录
            inpath = os.path.abspath('../SQL')
            file_path = inpath + '/' + str3 + '.db'
            # 数据库路径，哪种方法，表名
            UIFunctions.search_data_to_show(self, file_path, current_text, str1)

# 键盘回车事件，目前用于搜索按钮
    def datainfo_keyPressEvent(self, event):
        if event.key() == Qt.Key_Return or event.key() == Qt.Key_Enter:
            # 检查 lineEdit_2 中是否有内容
            if widgets.lineEdit_2.text():
                # 模拟 search 按钮点击事件
                self.ui.search.click()
        else:
            # 调用默认的 keyPressEvent 处理其他按键事件
            super(widgets.datainfo.__class__, widgets.datainfo).keyPressEvent(event)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    app.setWindowIcon(QIcon("icon.ico"))
    window = MainWindow()
    # window.resize(1440, 960)  # 高宽
    sys.exit(app.exec())