jiateng_ws/widgets/loading_dialog_widget.py

import logging
from PySide6.QtWidgets import QMessageBox
from PySide6.QtCore import Qt, Signal
from ui.loading_dialog_ui import LoadingDialogUI
from widgets.order_query_dialog import OrderQueryDialog
from utils.app_mode import AppMode
from utils.pallet_type_manager import PalletTypeManager
from apis.gc_api import GcApi

class LoadingDialog(LoadingDialogUI):
    """上料操作对话框"""
    
    # 定义信号，用于向主窗口传递托盘号
    tray_code_signal = Signal(str, str, str, str)
    # 定义信号，用于向主窗口传递订单号
    order_code_signal = Signal(str)
    
    def __init__(self, parent=None, user_id=None, user_name=None, corp_id=None):
        super().__init__()
        self.parent = parent
        self.user_id = user_id,
        self.user_name = user_name
        self.corp_id = corp_id
        
        # 存储订单数据
        self.order_data = None
        
        # 彻底禁用对话框的回车键关闭功能
        self.setModal(True)
        # 禁用所有按钮的默认行为
        self.confirm_button.setAutoDefault(False)
        self.confirm_button.setDefault(False)
        self.cancel_button.setAutoDefault(False)
        self.cancel_button.setDefault(False)
        
        # 设置对话框特性，按下Escape键才能关闭
        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)
        
        # 确保焦点跟踪器能够跟踪此对话框中的输入框
        from utils.focus_tracker import FocusTracker
        self.focus_tracker = FocusTracker.get_instance()
        
        # 连接信号和槽
        self.connect_signals()
    
    def connect_signals(self):
        """连接信号和槽"""
        # 订单查询按钮点击事件
        self.order_query_btn.clicked.connect(self.show_order_query_dialog)
        
        # 订单号输入框回车事件触发查询
        self.order_input.returnPressed.connect(self.handle_order_return_pressed)
        
        # 确认按钮点击事件
        self.confirm_button.clicked.connect(self.accept)
        
        # 取消按钮点击事件
        self.cancel_button.clicked.connect(self.reject)
        
        # 托盘号输入框回车事件
        self.tray_input.returnPressed.connect(self.on_tray_entered)
    
    def handle_order_return_pressed(self):
        """处理订单输入框的回车事件"""
        logging.info("订单输入框回车事件触发")
        # 获取当前订单号值
        order_code = self.order_input.text().strip()
        if not order_code:
            QMessageBox.warning(self, "提示", "请输入订单号")
            return
        
        # 发送订单号到主窗口
        from widgets.main_window import MainWindow
        main_window = self.parent
        if main_window and isinstance(main_window, MainWindow):
            logging.info(f"发送订单号到主窗口: {order_code}")
            self.order_code_signal.emit(order_code)
        
        # 查询订单信息
        self.on_order_query(order_code)
        
        # 阻止事件继续传播
        return True
    
    def show_order_query_dialog(self):
        """显示订单查询对话框"""
        try:
            # 创建订单查询对话框
            dialog = OrderQueryDialog(self)
            
            # 连接订单选择信号
            dialog.order_selected.connect(self.on_order_selected)
            
            # 显示对话框
            dialog.exec()
            
        except Exception as e:
            logging.error(f"显示订单查询对话框失败: {e}")
            QMessageBox.critical(self, "错误", f"显示订单查询对话框失败: {str(e)}")
    
    def on_order_selected(self, order_data):
        """处理订单选择事件
        
        Args:
            order_data: 订单数据字典
        """
        try:
            # 存储订单数据
            self.order_data = order_data
            logging.info(f"已存储订单数据: {order_data.get('mo', '')}")
            
            # 更新UI
            # 注意：此时order_data["note"]已经被修改为order_data["mo"]的值
            self.order_input.setText(order_data.get("mo", ""))
            
            # 更新轴型
            self.axis_value.setText(order_data.get("zx_name", "--"))
            
            # 更新数量
            self.quantity_value.setText(str(order_data.get("sl", "--")))
            
            # 更新重量
            self.weight_value.setText(str(order_data.get("zx_zl", "--")))
            
            # 自动填充托盘号
            if "xpack" in order_data:
                self.tray_input.setText(order_data["xpack"])
            
            # 发送订单号到主窗口
            from widgets.main_window import MainWindow
            main_window = self.parent
            if main_window and isinstance(main_window, MainWindow):
                # 使用note字段作为订单号（已经被修改为mo的值）
                order_code = order_data.get("mo", "")
                logging.info(f"发送订单号到主窗口: {order_code}")
                self.order_code_signal.emit(order_code)
            
            # 设置焦点到托盘号输入框
            self.tray_input.setFocus()
            
        except Exception as e:
            logging.error(f"处理订单选择事件失败: {e}")
            QMessageBox.critical(self, "错误", f"处理订单选择事件失败: {str(e)}")
    
    def on_order_query(self, order_code):
        """查询订单信息，同时生成托盘号，并且回显到页面上"""
        try:
            # 构建查询参数
            query_params = {
                "srch_mo": order_code,
                "srch_note": order_code,
                "corp_id": self.corp_id
            }
            
            # 创建订单查询对话框（仅用于执行查询）
            dialog = OrderQueryDialog(self)
            
            # 执行查询
            results = dialog.query_orders(query_params)
            
            # 处理查询结果
            if results:
                # 如果只有一个结果，直接使用
                if len(results) == 1:
                    self.on_order_selected(results[0])
                    return
                else:
                    # 如果有多个结果，显示查询对话框让用户选择
                    dialog.update_result_table(results)
                    dialog.order_selected.connect(self.on_order_selected)
                    dialog.exec()
                    return
            
            # 如果没有结果，并且不是API模式，显示提示
            if not AppMode.is_api():
                QMessageBox.warning(self, "提示", f"未找到订单: {order_code}")
                return
                
            # 如果是API模式，尝试使用旧的API方式查询
            # 调用接口
            gc_api = GcApi()
            # 防止response为None导致异常
            # 获取工程号信息，并且初始化数据
            order_response = gc_api.get_order_info(order_code, self.corp_id)
            # 生成托盘号
            if order_response and order_response.get("status", False):
                # 将接口数据保存到数据库，用于后续的关联使用
                from dao.inspection_dao import InspectionDAO
                inspection_dao = InspectionDAO()
                order_info = order_response.get("data", {})[0]
                # 设置轴数
                order_info['user_id'] = self.user_id[0]
                order_info['user_name'] = self.user_name
                order_info['data_corp'] = self.corp_id
                inspection_dao.save_order_info(order_code, order_info)
                self.axis_value.setText(order_info['zx_name'])
                self.quantity_value.setText(str(order_info['sl']))
                self.weight_value.setText(str(order_info['zx_zl']))
                
                # 存储订单数据
                self.order_data = order_info
                logging.info(f"已存储订单数据: {order_code}")
            else:
                QMessageBox.warning(self, "提示", f"未找到订单: {order_code}")
                return
            
            xpack_response = gc_api.get_xpack(order_code, self.corp_id)
            if xpack_response and xpack_response.get("status", False):
                xpack = xpack_response['xpack']
                spack = xpack_response['spack']
                self.tray_input.setText(xpack)
                self.pallet_tier_value.setText('3')
                
                # 将spack保存到order_data中，以便后续使用
                if hasattr(self, 'order_data') and self.order_data:
                    self.order_data['spack'] = spack
                    logging.info(f"已将spack: {spack}保存到order_data中")
            
                # 发送托盘号到主窗口
                from widgets.main_window import MainWindow
                main_window = self.parent
                if main_window and isinstance(main_window, MainWindow):
                    # 检查托盘号是否已存在
                    # 由于 tray_edit 现在是 QLineEdit，不再需要检查是否存在
                    # 直接设置文本
                    logging.info(f"设置主窗口当前托盘号: {xpack}")
                    main_window.tray_edit.setText(xpack)
                    
                    # 手动触发托盘号变更事件
                    main_window.handle_tray_changed(xpack)
                    
                    # spack值已经获取，可以在后续处理中使用
                    logging.info(f"已获取spack: {spack}，将与xpack一起使用")
            
            return order_info
                
        except Exception as e:
            logging.error(f"查询订单信息失败: {e}")
            QMessageBox.critical(self, "错误", f"查询订单信息失败: {str(e)}")
    
    def on_xpack_query(self, tray_id):
        """通过托盘号查询订单信息，并回显到页面（完全仿照on_order_query）"""
        try:
            query_params = {
                "srch_spack": tray_id,
                "corp_id": self.corp_id
            }
            dialog = OrderQueryDialog(self)
            results = dialog.query_orders_xpack(query_params)
            if results:
                if len(results) == 1:
                    self.on_order_selected(results[0])
                    return
                else:
                    dialog.update_result_table(results)
                    dialog.order_selected.connect(self.on_order_selected)
                    dialog.exec()
                    return
            QMessageBox.warning(self, "提示", "未找到该托盘号对应的订单信息")
        except Exception as e:
            logging.error(f"托盘号查订单接口异常: {e}")
            QMessageBox.warning(self, "错误", f"托盘号查订单接口异常: {str(e)}")

    def on_tray_entered(self):
        tray_id = self.tray_input.text().strip()
        if not tray_id:
            QMessageBox.warning(self, "提示", "请输入托盘号")
            return
        self.on_xpack_query(tray_id)
        self.pallet_tier_value.setFocus()
    
    def keyPressEvent(self, event):
        """重写键盘事件处理，防止回车关闭对话框"""
        # 如果按下回车键
        if event.key() == Qt.Key_Return or event.key() == Qt.Key_Enter:
            logging.info(f"捕获到回车键事件，当前焦点部件: {self.focusWidget()}")
            
            # 如果焦点在托盘输入框上，触发查询
            if self.focusWidget() == self.tray_input:
                event.accept()  # 消费掉这个事件
                return
            
            # 如果焦点在确认按钮上，则允许默认行为(确认)
            if self.focusWidget() == self.confirm_button:
                return super().keyPressEvent(event)
            
            # 其他情况下，阻止回车事件传播
            event.accept()  # 消费掉这个事件
            return
        
        # 其他键位事件交给父类处理
        super().keyPressEvent(event)
    
    def accept(self):
        """重写接受方法，确保在确认前所有数据都已处理"""
        logging.info("确认按钮被点击或回车触发确认")
        
        # 检查托盘层数是否有值
        tier_value = self.pallet_tier_value.text().strip()
        if not tier_value:
            QMessageBox.warning(self, "提示", "请输入托盘层数")
            return
        
        # 获取包装号
        pallet_code = self.tray_input.text().strip()
        if not pallet_code:
            QMessageBox.warning(self, "提示", "请输入托盘号")
            return
        
        # 检查订单号
        order_id = self.order_input.text().strip()
        if not order_id:
            QMessageBox.warning(self, "提示", "请输入订单号")
            self.order_input.setFocus()
            return
        
        # if not self.order_data:
        #     QMessageBox.warning(self, "提示", "请先查询订单信息")
        #     return
        
        try:
            # 保存托盘档案信息
            pallet_manager = PalletTypeManager.get_instance()
            success = pallet_manager.save_pallet_archives(
                pallet_code=pallet_code,
                tier=int(tier_value),
                user_id=self.user_id[0],
                user_name=self.user_name
            )
            
            if not success:
                QMessageBox.warning(self, "提示", "保存托盘档案失败")
                return
            
            logging.info(f"已保存托盘档案：托盘号={pallet_code}, 层数={tier_value}")
        except Exception as e:
            logging.error(f"保存托盘档案时发生错误: {str(e)}")
            QMessageBox.warning(self, "错误", f"保存托盘档案失败: {str(e)}")
            return
        
        # 确保主窗口启动了监听
        from widgets.main_window import MainWindow
        main_window = self.parent
        if main_window and isinstance(main_window, MainWindow):
            # 确保主窗口已启动监听
            try:
                if not hasattr(main_window, 'modbus_monitor') or not main_window.modbus_monitor.is_running():
                    main_window.setup_modbus_monitor()
                    logging.info("已在LoadingDialog确认时启动Modbus监控")
                
                # 启动串口监听
                main_window.serial_manager.auto_open_configured_ports()
                
                # 启动键盘监听器
                main_window.serial_manager.start_keyboard_listener()
                logging.info("已在LoadingDialog确认时启动键盘监听器")
            except Exception as e:
                logging.error(f"LoadingDialog确认时启动监听失败: {str(e)}")
        
        # 调用父类的accept方法关闭对话框
        super().accept()
    
    def reject(self):
        """重写拒绝方法"""
        logging.info("取消按钮被点击或ESC触发取消")
        # 调用父类的reject方法关闭对话框
        super().reject()
    
    def get_loading_data(self):
        """获取上料数据
        
        Returns:
            dict: 上料数据字典
        """
        # 获取spack值，如果order_data中有则使用，否则默认使用tray_id
        spack = ""
        if self.order_data and isinstance(self.order_data, dict) and 'spack' in self.order_data:
            spack = self.order_data['spack']
        else:
            spack = self.tray_input.text().strip()
            
        return {
            "order_id": self.order_input.text().strip(),
            "tray_id": self.tray_input.text().strip(),
            "spack": spack,  # 添加spack字段
            "pallet_tier": self.pallet_tier_value.text().strip(),
            "order_data": self.order_data if self.order_data else {}
        }