import itertools import sqlite3 from PySide6.QtGui import QStandardItemModel, QStandardItem from PySide6.QtCore import QItemSelectionModel def main_function(ui, db_path, utf_en): # 确保 utf_en 是字节字符串 if not isinstance(utf_en, bytes): utf_en = utf_en.encode('utf-8') # 设置 QTabWidget 的可见性和标签文本 tabs_to_show = [(0, '水准测段高差计算成果表')] for index in range(ui.tabWidget.count()): visible = index in [t[0] for t in tabs_to_show] ui.tabWidget.setTabVisible(index, visible) for index, text in tabs_to_show: ui.tabWidget.setTabText(index, text) # 连接到数据库并执行查询 db1 = sqlite3.connect(db_path) db1.text_factory = lambda x: str(x, 'utf-8') cursor1 = db1.cursor() # 查询 GC_Output_Point 表中的 TableName 字段 query_table_names = "SELECT DISTINCT TableName FROM GC_Output_Point" cursor1.execute(query_table_names) query = """ SELECT New_ID, New_ResultName, New_SPName, New_EPName, New_HDiff, New_RLen, Correct_Factor, Period_Diff, Dis_Ass FROM GC_Output_Point WHERE TableName = ? """ cursor1.execute(query, (utf_en,)) result = cursor1.fetchall() cursor1.close() db1.close() # 创建 QStandardItemModel 实例 model = QStandardItemModel() model.setColumnCount(len(result[0])) # 设置表头 headers = ['序号', '结果期数', '起点', '终点', '高差', '路线长', '修正数', '期间差异', '变形判定'] model.setHorizontalHeaderLabels(headers) for row_data in result: items = [] for i, item in enumerate(row_data): if i == 4 or i == 5: # 索引4（高差） item = f"{item:.6f}" # 格式化为6位小数 elif i == 6: # 索引 6（修正数） item = f"{item:.2f}" elif i == 7: # 假设 Period_Diff 在索引 7 if item is not None and isinstance(item, (int, float)): item = f"{item:.2f}" # 格式化为2位小数 else: item = "" # 如果是 None 或非数字类型，显示空字符串 elif i == 8: # 假设 Dis_Ass 在索引 8 if item is not None and isinstance(item, str): item = item # 显示字符串内容 else: item = "" # 如果是 None，显示空字符串 items.append(QStandardItem(str(item))) model.appendRow(items) # 设置并展示表格 ui.resultTableView.setModel(model) ui.resultTableView.show() # 隐藏 QLabel ui.default_remind.hide() def search_show_function(ui, db_path, utf_en): # 确保 utf_en 是字节字符串 if not isinstance(utf_en, bytes): utf_en = utf_en.encode('utf-8') tabs_to_show = [(0, '输入数据'), (1, '水准测段高差计算成果表')] for index in range(ui.View.count()): visible = index in [t[0] for t in tabs_to_show] ui.View.setTabVisible(index, visible) for index, text in tabs_to_show: ui.View.setTabText(index, text) # 连接到数据库并执行查询 db1 = sqlite3.connect(db_path) db1.text_factory = lambda x: str(x, 'utf-8') cursor1 = db1.cursor() # 查询 GC_Input_Point 表中的数据 input_query = """ SELECT New_ID, Last_ResultName, Last_SPName, Last_EPName, Last_HDiff, Last_RLen, New_ResultName, New_SPName, New_EPName, New_HDiff, New_RLen FROM GC_Input_Point WHERE TableName = ? """ cursor1.execute(input_query, (utf_en,)) input_result = cursor1.fetchall() # 查询 GC_Input_Param 表中的数据 param_query = """ SELECT ObservationLevel, Ms_Station, Last_StationCount, Last_SumHDiff, Last_SumRLen, New_StationCount, New_SumHDiff, New_SumRLen FROM GC_Input_Param WHERE TableName = ? """ cursor1.execute(param_query, (utf_en,)) param_result = cursor1.fetchall() # 创建 QStandardItemModel 实例 input_model = QStandardItemModel() input_model.setColumnCount(len(input_result[0]) + 8) # 增加八列（2 + 6） # 设置表头 input_headers = [ '序号', '上期成果名', '上期测段起点', '上期测段终点', '上期高差', '上期路线长', '本期成果名', '本期测段起点', '本期测段终点', '本期高差', '本期路线长', '观测等级', '测站全中误差', '上次总测段数', '上期高差合计', '上期路线长合计', '本期总测段数', '本期高差合计', '本期路线长合计' ] input_model.setHorizontalHeaderLabels(input_headers) # 使用 itertools.zip_longest 处理长度不一致的情况 for row_data, param_data in itertools.zip_longest(input_result, param_result, fillvalue=(None, None, None, None, None, None, None, None)): items = [] for i, item in enumerate(row_data): if item is None: item = "" # 如果是 None，显示空字符串 else: if i in [4, 5, 10, 11]: # 索引4,5,10,11（高差和路线长） item = f"{item:.6f}" # 格式化为6位小数 items.append(QStandardItem(str(item))) # 添加新的列数据 if param_data is not None: observation_level, ms_station, last_station_count, last_sum_hdiff, last_sum_rlen, new_station_count, new_sum_hdiff, new_sum_rlen = param_data else: observation_level, ms_station, last_station_count, last_sum_hdiff, last_sum_rlen, new_station_count, new_sum_hdiff, new_sum_rlen = "", "", "", "", "", "", "", "" # 确保 observation_level 和 ms_station 是字符串 if observation_level is None: observation_level = "" if ms_station is None: ms_station = "" # 确保其他字段是字符串或格式化后的字符串 if last_station_count is None: last_station_count = "" if last_sum_hdiff is None: last_sum_hdiff = "" else: last_sum_hdiff = f"{last_sum_hdiff:.6f}" if last_sum_rlen is None: last_sum_rlen = "" else: last_sum_rlen = f"{last_sum_rlen:.6f}" if new_station_count is None: new_station_count = "" if new_sum_hdiff is None: new_sum_hdiff = "" else: new_sum_hdiff = f"{new_sum_hdiff:.6f}" if new_sum_rlen is None: new_sum_rlen = "" else: new_sum_rlen = f"{new_sum_rlen:.6f}" items.append(QStandardItem(str(observation_level))) items.append(QStandardItem(f"{ms_station:.2f}" if isinstance(ms_station, (int, float)) else "")) items.append(QStandardItem(str(last_station_count))) items.append(QStandardItem(str(last_sum_hdiff))) items.append(QStandardItem(str(last_sum_rlen))) items.append(QStandardItem(str(new_station_count))) items.append(QStandardItem(str(new_sum_hdiff))) items.append(QStandardItem(str(new_sum_rlen))) input_model.appendRow(items) # 设置并展示输入数据表格 ui.View.setTabText(0, '输入数据') ui.resultTableView1.setModel(input_model) ui.resultTableView1.show() # 隐藏 QLabel ui.default_remind1.hide() # 查询 GC_Output_Point 表中的数据 output_query = """ SELECT New_ID, New_ResultName, New_SPName, New_EPName, New_HDiff, New_RLen, Correct_Factor, Period_Diff, Dis_Ass FROM GC_Output_Point WHERE TableName = ? """ cursor1.execute(output_query, (utf_en,)) output_result = cursor1.fetchall() # 创建 QStandardItemModel 实例 output_model = QStandardItemModel() output_model.setColumnCount(len(output_result[0])) # 设置表头 output_headers = ['序号', '结果期数', '起点', '终点', '高差', '路线长', '修正数', '期间差异', '变形判定'] output_model.setHorizontalHeaderLabels(output_headers) for row_data in output_result: items = [] for i, item in enumerate(row_data): if i == 4 or i == 5: # 索引4（高差）和索引5（路线长） item = f"{item:.6f}" # 格式化为6位小数 elif i == 6: # 索引 6（修正数） item = f"{item:.2f}" elif i == 7: # 假设 Period_Diff 在索引 7 if item is not None and isinstance(item, (int, float)): item = f"{item:.2f}" # 格式化为2位小数 else: item = "" # 如果是 None 或非数字类型，显示空字符串 elif i == 8: # 假设 Dis_Ass 在索引 8 if item is not None and isinstance(item, str): item = item # 显示字符串内容 else: item = "" # 如果是 None，显示空字符串 items.append(QStandardItem(str(item))) output_model.appendRow(items) # 设置并展示输出数据表格 ui.View.setTabText(1, '水准测段高差计算成果表') ui.reconTableView1.setModel(output_model) ui.reconTableView1.show() # 返回一个选择模型 selectionModel = QItemSelectionModel(input_model) # Item选择模型 ui.resultTableView1.setSelectionModel(selectionModel) return selectionModel # 关闭数据库连接 cursor1.close() db1.close() # 隐藏 QLabel ui.default_remind1.hide()