完成搜索查询树中的右键单击事件，完成GC初始数据中大部分数据的导出

Back/GC/GCExport.py

@@ -7,22 +7,14 @@ import os
 from PySide6.QtWidgets import QMessageBox
 from openpyxl.styles import Font, NamedStyle, Alignment
 from openpyxl.utils.dataframe import dataframe_to_rows
+from openpyxl import Workbook
 
 
-def main_function(ui, utf_en, db_path):
-    # 获取应用的安装目录
-    app_install_dir = os.path.dirname(os.path.abspath(__file__))  # 假设脚本文件位于应用的安装目录下
-    export_folder = os.path.join(app_install_dir, 'Export')
-
-    # 如果 Export 文件夹不存在，则创建它
-    if not os.path.exists(export_folder):
-        os.makedirs(export_folder)
-
-
-    # 连接数据库
-    conn = sqlite3.connect(db_path)
-
+def export_result_data(ui, utf_en, db_path, export_folder):
     try:
+        # 连接数据库
+        conn = sqlite3.connect(db_path)
+
         # 查询数据库表为DataFrame，添加 TableName 条件
         query = "SELECT * FROM GC_Output_Point WHERE TableName=?"
         df = pd.read_sql_query(query, conn, params=(utf_en,))
@@ -76,12 +68,10 @@ def main_function(ui, utf_en, db_path):
         correct_factor_df = pd.read_sql_query(query_param, conn, params=(utf_en,))
 
         correct_factor = correct_factor_df.iloc[0]['Correct_Factor']
+
         # 创建一个新的 Excel 工作簿
-        from openpyxl import Workbook
         wb = Workbook()
         ws = wb.active
-        # # 在最上面插入一行
-        # ws.insert_rows(1)
         # 在E1单元格添加内容“修正系数”
         ws.cell(row=1, column=1, value="序号")
         ws.cell(row=1, column=2, value="结果期数")
@@ -93,8 +83,7 @@ def main_function(ui, utf_en, db_path):
         ws.cell(row=1, column=8, value="期间差异")
         ws.cell(row=1, column=9, value="变形判定")
 
-
-# 将 DataFrame 写入工作表
+        # 将 DataFrame 写入工作表
         for r in dataframe_to_rows(df, index=False, header=True):
             ws.append(r)
 
@@ -151,3 +140,116 @@ def main_function(ui, utf_en, db_path):
     finally:
         # 关闭数据库连接
         conn.close()
+
+
+def export_initial_data(ui, utf_en, db_path, export_folder):
+    try:
+        # 解码 utf_en
+        decoded_utf_en = utf_en.decode('utf-8') if isinstance(utf_en, bytes) else utf_en
+
+        # 创建一个新的 Excel 工作簿
+        wb = Workbook()
+        ws = wb.active
+
+        # 合并指定的单元格
+        merge_cells = ['A1:B1', 'D1:E1', 'F1:G1', 'I1:J1']
+        for cell_range in merge_cells:
+            ws.merge_cells(cell_range)
+
+        # 设置合并单元格后的居中对齐
+        for cell_range in merge_cells:
+            cell = ws[cell_range.split(':')[0]]
+            cell.alignment = Alignment(horizontal='center', vertical='center')
+
+        # 填写表头信息
+        ws['A1'] = '观测等级'
+        ws['A2'] = '序号'
+        ws['B2'] = '起点'
+        ws['C2'] = '终点'
+        ws['D2'] = '高差（m）'
+        ws['E2'] = '路线长（km）'
+        ws['F1'] = '测站全中误差（mm）'
+        ws['F2'] = '序号'
+        ws['G2'] = '起点'
+        ws['H2'] = '终点'
+        ws['I2'] = '高差（m）'
+        ws['J2'] = '路线长（km）'
+
+        # 连接数据库
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+
+        # 查询数据库表为DataFrame，添加 TableName 条件
+        query = "SELECT * FROM GC_Input_Point WHERE TableName=?"
+        cursor.execute(query, (utf_en,))
+        rows = cursor.fetchall()
+
+        # 获取列名
+        column_names = [description[0] for description in cursor.description]
+
+        # 定义需要的列索引
+        last_id_index = column_names.index('Last_ID')
+        last_spname_index = column_names.index('Last_SPName')
+        last_epname_index = column_names.index('Last_EPName')
+        last_hdiff_index = column_names.index('Last_HDiff')
+        last_rlen_index = column_names.index('Last_RLen')
+        new_id_index = column_names.index('New_ID')
+        new_spname_index = column_names.index('New_SPName')
+        new_epname_index = column_names.index('New_EPName')
+        new_hdiff_index = column_names.index('New_HDiff')
+        new_rlen_index = column_names.index('New_RLen')
+
+        # 填充数据到Excel
+        for idx, row in enumerate(rows, start=3):
+            ws.cell(row=idx, column=1, value=row[last_id_index])
+            ws.cell(row=idx, column=2, value=row[last_spname_index])
+            ws.cell(row=idx, column=3, value=row[last_epname_index])
+            ws.cell(row=idx, column=4, value=row[last_hdiff_index])
+            ws.cell(row=idx, column=5, value=row[last_rlen_index])
+            ws.cell(row=idx, column=6, value=row[new_id_index])
+            ws.cell(row=idx, column=7, value=row[new_spname_index])
+            ws.cell(row=idx, column=8, value=row[new_epname_index])
+            ws.cell(row=idx, column=9, value=row[new_hdiff_index])
+            ws.cell(row=idx, column=10, value=row[new_rlen_index])
+
+            # 设置 D, E, I, J 列为强制保留六位小数
+            ws.cell(row=idx, column=4).number_format = '0.000000'
+            ws.cell(row=idx, column=5).number_format = '0.000000'
+            ws.cell(row=idx, column=9).number_format = '0.000000'
+            ws.cell(row=idx, column=10).number_format = '0.000000'
+        # 设置B,C,G,H的列宽为9.25
+        ws.column_dimensions['B'].width = 9.25
+        ws.column_dimensions['C'].width = 9.25
+        ws.column_dimensions['G'].width = 9.25
+        ws.column_dimensions['H'].width = 9.25
+
+        # 设置 D, E, I, J 列的列宽为11.5
+        ws.column_dimensions['D'].width = 11.5
+        ws.column_dimensions['E'].width = 11.5
+        ws.column_dimensions['I'].width = 11.5
+        ws.column_dimensions['J'].width = 11.5
+        # 设置文件名
+        excel_filename = f"{os.path.splitext(decoded_utf_en)[0]}-初始数据.xlsx"
+        excel_filepath = os.path.join(export_folder, excel_filename)
+
+        # 保存 Excel 文件
+        wb.save(excel_filepath)
+
+        QMessageBox.information(ui, '成功', f'初始数据文件已成功导出到 {export_folder}')
+    except Exception as e:
+        QMessageBox.critical(ui, '错误', f'导出初始数据过程中发生错误: {str(e)}')
+
+
+def main_function(ui, utf_en, db_path):
+    # 获取应用的安装目录
+    app_install_dir = os.path.dirname(os.path.abspath(__file__))  # 假设脚本文件位于应用的安装目录下
+    export_folder = os.path.join(app_install_dir, 'Export')
+
+    # 如果 Export 文件夹不存在，则创建它
+    if not os.path.exists(export_folder):
+        os.makedirs(export_folder)
+
+    # 调用导出初始数据函数
+    export_initial_data(ui, utf_en, db_path, export_folder)
+    # 调用导出结果函数
+    export_result_data(ui, utf_en, db_path, export_folder)

Front/main.py

@@ -15,6 +15,7 @@
 # ///////////////////////////////////////////////////////////////
 import time
 
+from PySide6.QtTest import QTest
 from PySide6.QtWidgets import QFileDialog, QWidget, QVBoxLayout, QTreeWidget, QApplication, QTreeWidgetItem
 from PySide6.QtCore import Signal, Qt, Slot, QObject
 from PySide6.QtSql import QSqlTableModel, QSqlDatabase
@@ -304,6 +305,7 @@ class ElTree(QWidget):
             main_window.refresh_tree()
 
 
+# 查询
 # 查询
 class ElTree1(QWidget):
     """
@@ -321,6 +323,11 @@ class ElTree1(QWidget):
             lambda item: self.itemDoubleClicked.emit(item.data(0, Qt.UserRole + 1)))
         self.__render_items(True)
 
+        # 设置 QTreeWidget 的 contextMenuPolicy 为 Qt.CustomContextMenu
+        widgets.qureyTreeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
+        # 连接 customContextMenuRequested 信号到 contextMenuEvent 方法
+        widgets.qureyTreeWidget.customContextMenuRequested.connect(self.contextMenuEvent)
+
     def __render_items(self, is_clear: bool):
         if is_clear:
             widgets.qureyTreeWidget.clear()
@@ -342,6 +349,135 @@ class ElTree1(QWidget):
             # 挂载到树上
             widgets.qureyTreeWidget.insertTopLevelItems(0, treeWidgetItems)
 
+    def contextMenuEvent(self, pos):
+        # 创建右键菜单
+        context_menu = QMenu(self)
+
+        # 添加删除选项
+        delete_action = QAction("删除", self)
+        delete_action.triggered.connect(self.delete_item)
+        context_menu.addAction(delete_action)
+
+        # 添加导出选项
+        export_action = QAction("导出", self)
+        export_action.triggered.connect(self.export_item)
+        context_menu.addAction(export_action)
+
+        context_menu.exec(widgets.qureyTreeWidget.mapToGlobal(pos))
+
+    def delete_item(self):
+        # 获取当前选中的项目
+        selected_items = widgets.qureyTreeWidget.selectedItems()
+        if not selected_items:
+            QMessageBox.warning(self, '警告', '请先选择一个项目')
+            return
+
+        item = selected_items[0]
+        data = item.data(0, Qt.UserRole + 1)
+        if not data:
+            self.warning = QMessageBox.warning(self, '警告', '所选项目没有关联的数据')
+            return
+
+        # 获取数据库路径和表名
+        dbname = data['listData'][0]
+        tablename = data['listData'][2]
+        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+
+        # 确保 tablename 是 UTF-8 编码的字符串
+        try:
+            tablename_utf8 = tablename.encode('utf-8')
+        except Exception as e:
+            QMessageBox.critical(self, '错误', f'表名编码转换错误: {str(e)}')
+            return
+        # 获取父节点
+        parent_item = item.parent()
+        if not parent_item:
+            QMessageBox.warning(self, '警告', '所选项目没有父节点')
+            return
+
+        parent_name = parent_item.text(0)
+
+        # 确定要删除的表
+        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'确定要删除项目表 {tablename} 吗？',
+                                     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} WHERE TableName = ?", (tablename_utf8,))
+            conn.commit()
+            conn.close()
+            QMessageBox.information(self, '成功', f'表 {tablename} 中的数据已从相关表中删除')
+
+            # 刷新树状图
+            self.refresh_tree()
+            # 模拟点击 Enter 键  实现程序自动刷新的效果
+            QTest.keyClick(widgets.qureyTreeWidget, Qt.Key_Return)
+        except Exception as e:
+            QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
+
+    def export_item(self):
+        # 获取当前选中的项目
+        selected_items = widgets.qureyTreeWidget.selectedItems()
+        if not selected_items:
+            QMessageBox.warning(self, '警告', '请先选择一个项目')
+            return
+
+        item = selected_items[0]
+        data = item.data(0, Qt.UserRole + 1)
+        if not data:
+            QMessageBox.warning(self, '警告', '所选项目没有关联的数据')
+            return
+
+        # 获取数据库路径和表名
+        dbname = data['listData'][0]
+        tablename = data['listData'][2]
+        tablename_utf8 = tablename.encode('utf-8')
+        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+
+        # 获取父节点
+        parent_item = item.parent()
+        if not parent_item:
+            QMessageBox.warning(self, '警告', '所选项目没有父节点')
+            return
+
+        parent_name = parent_item.text(0)
+
+        # 根据父节点名称确定导出操作
+        if parent_name == '水准测段高差稳定计算':
+            GCExport.main_function(self, tablename_utf8, dbpath)
+        elif parent_name == '控制网复测平面基准计算':
+            GSExport.main_function(self, dbpath, tablename)
+        elif parent_name == '平面控制网稳定性计算':
+            WDExport.main_function(self, dbpath, tablename)
+        else:
+            QMessageBox.warning(self, '警告', '未知的父节点')
+            return
+
+    def refresh_tree(self):
+        # 调用 MainWindow 的 refresh_tree 方法
+        main_window = QApplication.instance().activeWindow()
+        if isinstance(main_window, MainWindow):
+            main_window.refresh_tree()
+
+
 
 # class DatabaseChangeHandler(FileSystemEventHandler):
 #     def __init__(self, main_window):