from PySide6.QtWidgets import ( QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QMessageBox, QDialog, QFileDialog ) from PySide6.QtCore import Qt, QSettings import os import glob from src.db_manager import DatabaseManager from src.xml_parser import XmlParser from utils.readMTP import MTPHandler from src.session_manager import SessionManager from utils.logger import get_logger, log_timing_context logger = get_logger("base_page") class BaseInspectionPage(QDialog): def __init__(self, title, parent=None): super().__init__(parent) self.setWindowTitle(title) self.setWindowFlags(self.windowFlags() | Qt.Window | Qt.WindowMaximizeButtonHint) self.settings = QSettings("Tengzhi", "InspectionApp") self.mtp_handler = MTPHandler() self.session = SessionManager() self.page_type = title # 保存页面类型（"入检" 或 "手检"） self.main_layout = QVBoxLayout(self) self.main_layout.setContentsMargins(0, 0, 0, 0) self.main_layout.setSpacing(0) self.create_toolbar() # Content Area (to be filled by subclasses) self.content_widget = QWidget() self.content_layout = QVBoxLayout(self.content_widget) self.content_layout.setContentsMargins(20, 20, 20, 20) self.content_layout.setSpacing(15) self.content_layout.setAlignment(Qt.AlignTop) self.main_layout.addWidget(self.content_widget, 1) # Stretch = 1 to fill space # Bottom Button Bar self.create_bottom_bar() def create_toolbar(self): toolbar = QWidget() toolbar.setStyleSheet("background-color: #f8f8f8; border-bottom: 1px solid #ddd;") layout = QHBoxLayout(toolbar) layout.setContentsMargins(10, 5, 10, 5) title_label = QPushButton("⚙ 设置读取路径") title_label.setStyleSheet(""" QPushButton { border: none; background-color: transparent; font-family: 'Microsoft YaHei'; font-size: 13px; color: #666; text-align: left; } QPushButton:hover { color: #0086fa; } """) title_label.clicked.connect(self.on_settings) layout.addWidget(title_label) layout.addStretch() self.main_layout.addWidget(toolbar) def on_settings(self): from PySide6.QtWidgets import QFormLayout, QLineEdit, QRadioButton, QButtonGroup dialog = QDialog(self) dialog.setWindowTitle("同步设置") dialog.setFixedWidth(400) layout = QVBoxLayout(dialog) form = QFormLayout() # Mode Selection mode_group = QButtonGroup(dialog) current_mode = self.settings.value("sync_mode", "local") rb_local = QRadioButton("本地路径 (本地硬盘)") rb_mtp = QRadioButton("MTP设备 (连接的PDA)") mode_group.addButton(rb_local) mode_group.addButton(rb_mtp) if current_mode == "local": rb_local.setChecked(True) else: rb_mtp.setChecked(True) form.addRow("同步模式:", rb_local) form.addRow("", rb_mtp) self.edit_local_path = QLineEdit(self.settings.value("xml_path", os.getcwd())) btn_browse = QPushButton("浏览...") btn_browse.clicked.connect(self.browse_local_path) local_path_layout = QHBoxLayout() local_path_layout.addWidget(self.edit_local_path) local_path_layout.addWidget(btn_browse) form.addRow("本地文件夹:", local_path_layout) form.addRow(QWidget()) # Spacer # MTP Settings self.edit_mtp_device = QLineEdit(self.settings.value("mtp_device", "Trizeps VII")) form.addRow("MTP设备名称:", self.edit_mtp_device) self.edit_mtp_path = QLineEdit(self.settings.value("mtp_path", "Storage Card/Program Files/xSort/Results/Analyses")) form.addRow("MTP 内部路径:", self.edit_mtp_path) layout.addLayout(form) # Save Button btn_save = QPushButton("保存配置") btn_save.setStyleSheet("background-color: #0086fa; color: white; padding: 10px; font-weight: bold;") btn_save.clicked.connect(lambda: self.save_settings(dialog, rb_local.isChecked())) layout.addWidget(btn_save) dialog.exec() def browse_local_path(self): directory = QFileDialog.getExistingDirectory(self, "选择本地文件夹", self.edit_local_path.text()) if directory: self.edit_local_path.setText(directory) def save_settings(self, dialog, is_local): self.settings.setValue("sync_mode", "local" if is_local else "mtp") self.settings.setValue("xml_path", self.edit_local_path.text()) self.settings.setValue("mtp_device", self.edit_mtp_device.text()) self.settings.setValue("mtp_path", self.edit_mtp_path.text()) self.show_info("设置成功", "配置已更新。") dialog.accept() def create_bottom_bar(self): button_bar = QWidget() button_bar.setStyleSheet("background-color: #e0e0e0; border-top: 1px solid #999;") button_layout = QHBoxLayout(button_bar) button_layout.setContentsMargins(10, 10, 10, 10) button_layout.setSpacing(10) btn_style = """ QPushButton { background-color: #0086fa; color: white; border-radius: 5px; padding: 15px; font-family: 'Microsoft YaHei'; font-size: 19px; font-weight: bold; } QPushButton:pressed { background-color: #006bbd; } """ # Buttons: [Sync, Submit, Print, Close] self.btn_sync = QPushButton("同步") self.btn_sync.setStyleSheet(btn_style) self.btn_sync.clicked.connect(self.on_sync) self.btn_submit = QPushButton("提交") self.btn_submit.setStyleSheet(btn_style) self.btn_submit.clicked.connect(self.on_submit) self.btn_close = QPushButton("关闭") self.btn_close.setStyleSheet(btn_style) self.btn_close.clicked.connect(self.close) button_layout.addWidget(self.btn_sync) button_layout.addWidget(self.btn_submit) button_layout.addWidget(self.btn_close) self.main_layout.addWidget(button_bar) def on_sync(self): logger.info(f"[{self.page_type}] 开始同步操作") sync_mode = self.settings.value("sync_mode", "local") xml_path = None target_xml_name = "" if sync_mode == "local": xml_dir = self.settings.value("xml_path", "xml_data") if not os.path.exists(xml_dir): self.show_info("同步失败", f"本地路径不存在: {xml_dir}") return xml_files = glob.glob(os.path.join(xml_dir, "*.xml")) if not xml_files: self.show_info("同步失败", "本地目录下未找到 .xml 文件") return latest_xml = max(xml_files, key=os.path.getmtime) xml_path = latest_xml target_xml_name = os.path.basename(latest_xml) logger.info(f"本地模式，选定文件: {target_xml_name}") else: device_name = self.settings.value("mtp_device", "Trizeps VII") remote_path = self.settings.value("mtp_path", "") self.show_info("正在同步", f"正在从设备 [{device_name}] 读取最新数据...") with log_timing_context(f"MTP同步 设备={device_name}", "base_page"): local_cache, filename = self.mtp_handler.get_latest_xml_windows(device_name, remote_path) if not local_cache: self.show_info("同步失败", f"MTP同步错误: {filename}") return xml_path = local_cache target_xml_name = filename # 1.解析xml parser = XmlParser() with log_timing_context("XML解析", "base_page"): parsed_data, msg = parser.parse_file(xml_path) if not parsed_data: self.show_info("同步失败", f"XML解析错误: {msg}") return # 2. 写入数据 db = DatabaseManager() with log_timing_context("DB写入分析数据", "base_page"): success, db_msg = db.insert_analysis_data(parsed_data) if not success: self.show_info("数据库错误", f"插入失败: {db_msg}\n(请检查数据库连接)") return with log_timing_context("DB读取最新样本数据", "base_page"): latest_data = db.get_latest_sample_data() if latest_data: self.update_ui_with_data(latest_data) self.show_info("同步成功", f"已同步 {len(parsed_data)} 条元素数据。\n源文件: {target_xml_name}") else: self.show_info("同步异常", "数据插入成功但无法读取回显。") logger.info(f"[{self.page_type}] 同步操作完成") def update_ui_with_data(self, data): ### 更新UI界面，如果不符合，对不符合的元素字体标红，并且 pass def clear_data(self): """清空界面所有数据（子类实现）""" pass def on_submit(self): """提交检验数据到 e_pz_hjqy_import 表""" logger.info(f"[{self.page_type}] 开始提交检验数据") try: # 1. 收集表单数据 gch = self.inputs.get("batch_no").text().strip() if self.inputs.get("batch_no") else "" heat_number = self.inputs.get("material1").text().strip() if self.inputs.get("material1") else "" heat_number2 = self.inputs.get("info1").text().strip() if self.inputs.get("info1") else "" cz = self.inputs.get("spec1").text().strip() if self.inputs.get("spec1") else "" cz2 = self.inputs.get("spec2").text().strip() if self.inputs.get("spec2") else "" check_information = self.inputs.get("insp_info_1").text().strip() if self.inputs.get("insp_info_1") else "" check_information2 = self.inputs.get("insp_info_2").text().strip() if self.inputs.get("insp_info_2") else "" # 产地/材质 拼接显示时，提交只需要产地 (cd) display_cd = self.inputs.get("material").text().strip() if self.inputs.get("material") else "" cd = display_cd.split('/')[0] if '/' in display_cd else display_cd size = self.inputs.get("spec").text().strip() if self.inputs.get("spec") else "" # 2. 验证必填字段 if not gch: self.show_info("提交失败", "工程号不能为空") return # 3. 收集所有元素数据 element_names = ["C", "Si", "Mn", "P", "S", "Cr", "Ni", "Mo", "Cu", "Ti", "Nb", "V", "Al", "W", "Co", "N", "Fe", "Se"] elements = {} for elem in element_names: if elem in self.inputs: value_str = self.inputs[elem].text().strip() if value_str: try: elements[elem] = float(value_str) except ValueError: pass # 4. 取当前用户信息 user_id = self.session.get_user_id() dept_id = self.session.get_dept_id() data_corp = self.session.get_data_corp() if not user_id: self.show_info("提交失败", "未获取到用户信息，请重新登录") return # 5. 构建提交数据 inspection_data = { "gch": gch, "heat_number": heat_number, "heat_number2": heat_number2, "cz": cz, "cz2": cz2, "check_information": check_information, "check_information2": check_information2, "cd": cd, "size": size, "elements": elements, "user_id": user_id, "dept_id": dept_id, "data_corp": data_corp } # 6. 插入数据库 db = DatabaseManager() with log_timing_context(f"DB提交检验数据 gch={gch}", "base_page"): success, message = db.insert_inspection_data(inspection_data) if success: logger.info(f"[{self.page_type}] 提交成功: {message}") self.show_info("提交成功", message) self.clear_data() else: logger.warning(f"[{self.page_type}] 提交失败: {message}") self.show_info("提交失败", message) except Exception as e: logger.error(f"[{self.page_type}] 提交异常: {e}", exc_info=True) self.show_info("提交异常", f"发生错误: {str(e)}") def show_info(self, title, message): QMessageBox.information(self, title, message)