import time import openpyxl import pandas as pd import sqlite3 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 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) # 查询数据库表为DataFrame，添加 TableName 条件 query = "SELECT * FROM GC_Output_Point WHERE TableName=?" df = pd.read_sql_query(query, conn, params=(utf_en,)) # 检查是否有匹配的数据 if df.empty: QMessageBox.warning(ui, '警告', '没有找到匹配的数据进行导出') conn.close() return # 假设 TableName 字段是以字节序列存储的 UTF-8 编码字符串 if 'TableName' in df.columns: try: df['TableName'] = df['TableName'].apply(lambda x: x.decode('utf-8') if isinstance(x, bytes) else x) except Exception as e: QMessageBox.critical(ui, '错误', f'TableName 字段解码失败: {str(e)}') conn.close() return # 删除 TableName 列 if 'TableName' in df.columns: df.drop(columns=['TableName'], inplace=True) # new HDiff 保留小数点后6位 if 'New_HDiff' in df.columns: df['New_HDiff'] = df['New_HDiff'].round(6) # New_RLen 保留小数点后6位 if 'New_RLen' in df.columns: df['New_RLen'] = df['New_RLen'].round(6) # Correct_Factor 保留小数点后2位 if 'Correct_Factor' in df.columns: df['Correct_Factor'] = df['Correct_Factor'].round(2) # Period_Diff 保留小数点后2位 if 'Period_Diff' in df.columns: df['Period_Diff'] = df['Period_Diff'].round(2) # 重命名指定的列 column_mapping = { 'New_ID': '序号', 'New_ResultName': '结果期数', 'New_SPName': '起点', 'New_EPName': '终点', 'New_HDiff': '高差', 'New_RLen': '路线长', 'Correct_Factor': '修正数', 'Period_Diff': '期间差异', 'Dis_Ass': '变形判定' } df.rename(columns=column_mapping, inplace=True) # 查询 GC_Input_Param 表获取 Correct_Factor 字段 query_param = "SELECT Correct_Factor FROM GC_Input_Param WHERE TableName=?" correct_factor_df = pd.read_sql_query(query_param, conn, params=(utf_en,)) correct_factor = correct_factor_df.iloc[0]['Correct_Factor'] # 创建一个新的 Excel 工作簿 wb = Workbook() ws = wb.active # 在E1单元格添加内容“修正系数” ws.cell(row=1, column=1, value="序号") ws.cell(row=1, column=2, value="结果期数") ws.cell(row=1, column=3, value="起点") ws.cell(row=1, column=4, value="终点") ws.cell(row=1, column=5, value="修正系数") ws.cell(row=1, column=6, value=correct_factor) ws.cell(row=1, column=7, value="修正数") ws.cell(row=1, column=8, value="期间差异") ws.cell(row=1, column=9, value="变形判定") # 将 DataFrame 写入工作表 for r in dataframe_to_rows(df, index=False, header=True): ws.append(r) # 定义自定义样式 style_0_000000 = NamedStyle(name="style_0_000000", number_format='0.000000') style_0_00 = NamedStyle(name="style_0_00", number_format='0.00') # 添加样式到工作簿 wb.add_named_style(style_0_000000) wb.add_named_style(style_0_00) # 应用样式到相应列 hdiff_col_index = df.columns.get_loc('高差') + 1 # 获取“高差”列的索引（+1 因为 Excel 列索引从 1 开始） rlen_col_index = df.columns.get_loc('路线长') + 1 # 获取“路线长”列的索引 correct_factor_col_index = df.columns.get_loc('修正数') + 1 # 获取“修正数”列的索引 period_diff_col_index = df.columns.get_loc('期间差异') + 1 # 获取“期间差异”列的索引 for row in range(2, ws.max_row + 1): # 从第二行开始，因为第一行为标题行 ws.cell(row=row, column=hdiff_col_index).style = style_0_000000 ws.cell(row=row, column=rlen_col_index).style = style_0_000000 ws.cell(row=row, column=correct_factor_col_index).style = style_0_00 ws.cell(row=row, column=period_diff_col_index).style = style_0_00 # 设置 Dis_Ass 列为“变形”的行的字体颜色为红色 red_font = Font(color="FF0000") dis_ass_col_index = df.columns.get_loc('变形判定') + 1 # 获取“变形判定”列的索引 for row in range(2, ws.max_row + 1): # 从第二行开始，因为第一行为标题行 cell_value = ws.cell(row=row, column=dis_ass_col_index).value if cell_value == '变形': for col in range(1, ws.max_column + 1): ws.cell(row=row, column=col).font = red_font # 设置列宽 ws.column_dimensions['E'].width = 12 ws.column_dimensions['F'].width = 12 # 保存 Excel 文件之前添加单元格合并操作 merge_cells = ['A1:A2', 'B1:B2', 'C1:C2', 'D1:D2', 'G1:G2', 'H1:H2', 'I1:I2'] 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') # 保存 Excel 文件 excel_filename = f"{os.path.splitext(decoded_utf_en)[0]}-成果数据.xlsx" excel_filepath = os.path.join(export_folder, excel_filename) wb.save(excel_filepath) QMessageBox.information(ui, '成功', f'成果文件已成功导出到 {export_folder}') except Exception as e: QMessageBox.critical(ui, '错误', f'导出过程中发生错误: {str(e)}') 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') # 连接数据库 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'] = '终点' ws['D2'] = '高差（m）' ws['E2'] = '路线长（km）' ws['F1'] = '测站全中误差（mm）' ws['F2'] = '序号' ws['G2'] = '起点' ws['H2'] = '终点' 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() # 查询数据库表为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_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') # 如果 Export 文件夹不存在，则创建它 if not os.path.exists(export_folder): os.makedirs(export_folder) # 调用导出初始数据函数 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)