完成了在最后一个页面上的树状图中单击右键实现导出的功能（输入文件的导出还未完成）

Back/GC/GCExport.py

@@ -9,11 +9,16 @@ from openpyxl.styles import Font, NamedStyle, Alignment
 from openpyxl.utils.dataframe import dataframe_to_rows
 
 
-def main_function(ui, file_path, utf_en, db_path):
-    export_folder = os.path.dirname(file_path)
+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)
 

Back/GS/GSExport.py

@@ -234,14 +234,16 @@ def openpyxl_write(file_name, dbpath, filename):
 
 
 # 主函数 写入excel文件
-def main_function(ui, file_path, dbpath, excelname):
-    # dbpath = r"D:\4work_now\20240819GS\ControlNetwork\ControlNetwork\UI\SQL\DataBase.db"
-    # outpath = r'D:\4work_now\20240819GS\JPG\GS成果表.xlsx'
-    # file_path = r'D:\4work_now\20240819GS\二滩_20240816.xls'
-    file_name = os.path.dirname(file_path)
+def main_function(ui, dbpath, excelname):
+    # 获取应用的安装目录
+    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)
     try:
-        openpyxl_write(file_name, dbpath, excelname)
-        QMessageBox.information(ui, '成功', f'成果文件已成功导出到 {file_name}')
+        openpyxl_write(export_folder, dbpath, excelname)
+        QMessageBox.information(ui, '成功', f'成果文件已成功导出到 {export_folder}')
     except Exception as e:
         QMessageBox.critical(ui, '错误', f'导出过程中发生错误: {str(e)}')
-

Back/WD/WDExport.py

@@ -194,14 +194,20 @@ def openpyxl_write(file_name, dbpath, filename):
 
 
 # 主函数 写入excel文件
-def main_function(ui, file_path, dbpath, excelname):
+def main_function(ui, dbpath, excelname):
     # dbpath = r"D:\4work_now\20240819GS\ControlNetwork\ControlNetwork\UI\SQL\DataBase.db"
     # outpath = r'D:\4work_now\20240819GS\JPG\WD成果表.xlsx'
     # file_path = r'D:\4work_now\20240819GS\test_wd.xls'
-    file_name = os.path.dirname(file_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)
     try:
-        openpyxl_write(file_name, dbpath, excelname)
-        QMessageBox.information(ui, '成功', f'成果文件已成功导出到 {file_name}')
+        openpyxl_write(export_folder, dbpath, excelname)
+        QMessageBox.information(ui, '成功', f'成果文件已成功导出到 {export_folder}')
     except Exception as e:
         QMessageBox.critical(ui, '错误', f'导出过程中发生错误: {str(e)}')
 

Front/main.py

@@ -178,12 +178,19 @@ class ElTree(QWidget):
             widgets.allTreeWidget.insertTopLevelItems(0, treeWidgetItems)
 
     def contextMenuEvent(self, pos):
-        print("contextMenuEvent called")  # 调试信息
         # 创建右键菜单
         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.allTreeWidget.mapToGlobal(pos))
 
     def delete_item(self):
@@ -196,15 +203,13 @@ class ElTree(QWidget):
         item = selected_items[0]
         data = item.data(0, Qt.UserRole + 1)
         if not data:
-            QMessageBox.warning(self, '警告', '所选项目没有关联的数据')
+            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")
-        print(dbpath)
-        print(tablename)
 
         # 确保 tablename 是 UTF-8 编码的字符串
         try:
@@ -212,7 +217,6 @@ class ElTree(QWidget):
         except Exception as e:
             QMessageBox.critical(self, '错误', f'表名编码转换错误: {str(e)}')
             return
-        print(tablename_utf8)
         # 获取父节点
         parent_item = item.parent()
         if not parent_item:
@@ -220,7 +224,6 @@ class ElTree(QWidget):
             return
 
         parent_name = parent_item.text(0)
-        print(f"Parent Name: {parent_name}")
 
         # 确定要删除的表
         tables_to_delete = []
@@ -255,7 +258,43 @@ class ElTree(QWidget):
             self.refresh_tree()
         except Exception as e:
             QMessageBox.critical(self, '错误', f'删除数据时出错: {str(e)}')
-            print(e)
+    def export_item(self):
+        # 获取当前选中的项目
+        selected_items = widgets.allTreeWidget.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 方法
@@ -309,7 +348,6 @@ class DatabaseChangeHandler(FileSystemEventHandler):
 
     def on_modified(self, event):
         if event.src_path.endswith('.db'):
-            print(f"Database file {event.src_path} has been modified.")
             self.main_window.refresh_tree()
 
 

Front/modules/ui_functions.py

@@ -354,11 +354,11 @@ class UIFunctions(MainWindow):
         excelname = os.path.basename(file_path)  # 文件名
         utf_en = excelname.encode('utf-8')  # 转换文件名为utf-8编码形式
         if current_text == "水准测段高差稳定计算":
-            GCExport.main_function(self, file_path, utf_en, db_path)
+            GCExport.main_function(self, utf_en, db_path)
         elif current_text == "控制网复测平面基准计算":
-            GSExport.main_function(self, file_path, db_path, excelname)
+            GSExport.main_function(self, db_path, excelname)
         elif current_text == "平面控制网稳定性计算":
-            WDExport.main_function(self, file_path, db_path, excelname)
+            WDExport.main_function(self, db_path, excelname)
         else:
             QMessageBox.warning(self, '警告', '请选择有效的计算类型')