完成了GC的全部导出功能

Back/GC/GCExport.py

@@ -12,6 +12,8 @@ from openpyxl import Workbook
 
 def export_result_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
         # 连接数据库
         conn = sqlite3.connect(db_path)
 
@@ -130,7 +132,7 @@ def export_result_data(ui, utf_en, db_path, export_folder):
             cell = ws[cell_range.split(':')[0]]
             cell.alignment = Alignment(horizontal='center', vertical='center')
         # 保存 Excel 文件
-        excel_filename = f"水准测段高差计算成果表{time.strftime('%Y%m%d_%H%M%S')}.xlsx"
+        excel_filename = f"{os.path.splitext(decoded_utf_en)[0]}-成果数据.xlsx"
         excel_filepath = os.path.join(export_folder, excel_filename)
         wb.save(excel_filepath)
 
@@ -160,9 +162,25 @@ def export_initial_data(ui, utf_en, db_path, export_folder):
         for cell_range in merge_cells:
             cell = ws[cell_range.split(':')[0]]
             cell.alignment = Alignment(horizontal='center', vertical='center')
+        # 连接数据库
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+
+        # 查询 GC_Input_Param 表获取 ObservationLevel, Last_ResultName, New_ResultName, Ms_Station 字段
+        query_observation_level = "SELECT ObservationLevel, Last_ResultName, New_ResultName, Ms_Station FROM GC_Input_Param WHERE TableName=?"
+        cursor.execute(query_observation_level, (utf_en,))
+        result = cursor.fetchone()
+        if result:
+            observation_level, last_result_name, new_result_name, ms_station = result
+        else:
+            observation_level = "未找到"
+            last_result_name = "未找到"
+            new_result_name = "未找到"
+            ms_station = "未找到"
 
         # 填写表头信息
         ws['A1'] = '观测等级'
+        ws['C1'] = observation_level  # 将 ObservationLevel 写入 C1 单元格
         ws['A2'] = '序号'
         ws['B2'] = '起点'
         ws['C2'] = '终点'
@@ -175,6 +193,14 @@ def export_initial_data(ui, utf_en, db_path, export_folder):
         ws['I2'] = '高差（m）'
         ws['J2'] = '路线长（km）'
 
+        # 写入 Last_ResultName, New_ResultName, Ms_Station 到相应的单元格
+        ws['D1'] = last_result_name
+        ws['I1'] = new_result_name
+        ws['H1'] = ms_station
+
+        # 设置 H1 单元格的数值格式为保留两位小数
+        ws['H1'].number_format = '0.00'
+
         # 连接数据库
         conn = sqlite3.connect(db_path)
         cursor = conn.cursor()
@@ -240,7 +266,7 @@ def export_initial_data(ui, utf_en, db_path, export_folder):
         QMessageBox.critical(ui, '错误', f'导出初始数据过程中发生错误: {str(e)}')
 
 
-def main_function(ui, utf_en, db_path):
+def main_function_initial(ui, utf_en, db_path):
     # 获取应用的安装目录
     app_install_dir = os.path.dirname(os.path.abspath(__file__))  # 假设脚本文件位于应用的安装目录下
     export_folder = os.path.join(app_install_dir, 'Export')
@@ -251,5 +277,15 @@ def main_function(ui, utf_en, db_path):
 
     # 调用导出初始数据函数
     export_initial_data(ui, utf_en, db_path, export_folder)
+
+
+def main_function_result(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_result_data(ui, utf_en, db_path, export_folder)

Front/main.py

@@ -289,7 +289,8 @@ class ElTree(QWidget):
 
         # 根据父节点名称确定导出操作
         if parent_name == '水准测段高差稳定计算':
-            GCExport.main_function(self, tablename_utf8, dbpath)
+            GCExport.main_function_initial(self, tablename_utf8, dbpath)
+            GCExport.main_function_result(self, tablename_utf8, dbpath)
         elif parent_name == '控制网复测平面基准计算':
             GSExport.main_function(self, dbpath, tablename)
         elif parent_name == '平面控制网稳定性计算':

Front/modules/ui_functions.py

@@ -357,7 +357,7 @@ class UIFunctions(MainWindow):
         excelname = os.path.basename(file_path)  # 文件名
         utf_en = excelname.encode('utf-8')  # 转换文件名为utf-8编码形式
         if current_text == "水准测段高差稳定计算":
-            GCExport.main_function(self, utf_en, db_path)
+            GCExport.main_function_result(self, utf_en, db_path)
         elif current_text == "控制网复测平面基准计算":
             GSExport.main_function(self, db_path, excelname)
         elif current_text == "平面控制网稳定性计算":