simple-dibang/ui/cargo_tab.py

"""
货物维护Tab页面
"""
from PyQt5.QtWidgets import (
    QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QPushButton,
    QTableWidget, QTableWidgetItem, QHeaderView, QMessageBox, QDialog,
    QFormLayout
)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QFont
from database import Database


class CargoTab(QWidget):
    """货物维护Tab"""
    
    def __init__(self, db: Database):
        super().__init__()
        self.db = db
        self.init_ui()
        self.refresh_list()
    
    def init_ui(self):
        """初始化界面"""
        layout = QVBoxLayout()
        layout.setSpacing(10)
        layout.setContentsMargins(15, 15, 15, 15)
        
        # 标题
        title = QLabel("货物名称维护")
        title_font = QFont("微软雅黑", 18, QFont.Bold)
        title.setFont(title_font)
        layout.addWidget(title)
        
        # 按钮区域
        button_layout = QHBoxLayout()
        button_layout.setSpacing(10)
        
        self.add_btn = QPushButton("新增")
        self.add_btn.setMinimumHeight(45)
        self.add_btn.setMinimumWidth(120)
        self.add_btn.setFont(QFont("微软雅黑", 16))
        self.add_btn.setStyleSheet("background-color: #2196F3; color: white; border-radius: 5px;")
        self.add_btn.clicked.connect(self.add_cargo)
        # 添加快捷键
        from PyQt5.QtGui import QKeySequence
        self.add_btn.setShortcut(QKeySequence("Ctrl+N"))
        
        self.edit_btn = QPushButton("修改")
        self.edit_btn.setMinimumHeight(45)
        self.edit_btn.setMinimumWidth(120)
        self.edit_btn.setFont(QFont("微软雅黑", 16))
        self.edit_btn.clicked.connect(self.edit_cargo)
        
        self.delete_btn = QPushButton("删除")
        self.delete_btn.setMinimumHeight(45)
        self.delete_btn.setMinimumWidth(120)
        self.delete_btn.setFont(QFont("微软雅黑", 16))
        self.delete_btn.setStyleSheet("background-color: #f44336; color: white; border-radius: 5px;")
        self.delete_btn.clicked.connect(self.delete_cargo)
        # 添加快捷键
        from PyQt5.QtGui import QKeySequence
        self.delete_btn.setShortcut(QKeySequence("Delete"))
        
        self.refresh_btn = QPushButton("刷新")
        self.refresh_btn.setMinimumHeight(45)
        self.refresh_btn.setMinimumWidth(120)
        self.refresh_btn.setFont(QFont("微软雅黑", 16))
        self.refresh_btn.setStyleSheet("""
            QPushButton {
                background-color: #607D8B; 
                color: white; 
                border-radius: 5px;
                border: none;
            }
            QPushButton:hover {
                background-color: #455A64;
            }
            QPushButton:pressed {
                background-color: #37474F;
            }
            QPushButton:disabled {
                background-color: #BDBDBD;
                color: #757575;
            }
        """)
        self.refresh_btn.setEnabled(True)  # 确保按钮是启用状态
        self.refresh_btn.clicked.connect(self.refresh_list)
        self.refresh_btn.setShortcut(QKeySequence("F5"))
        
        button_layout.addWidget(self.add_btn)
        button_layout.addWidget(self.edit_btn)
        button_layout.addWidget(self.delete_btn)
        button_layout.addWidget(self.refresh_btn)
        button_layout.addStretch()
        
        layout.addLayout(button_layout)
        
        # 表格
        self.table = QTableWidget()
        self.table.setColumnCount(2)
        self.table.setHorizontalHeaderLabels(["序号", "货物名称"])
        self.table.horizontalHeader().setFont(QFont("微软雅黑", 15, QFont.Bold))
        self.table.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)
        self.table.setFont(QFont("微软雅黑", 15))
        self.table.setAlternatingRowColors(True)
        self.table.setSelectionBehavior(QTableWidget.SelectRows)
        self.table.setSelectionMode(QTableWidget.SingleSelection)
        
        layout.addWidget(self.table)
        self.setLayout(layout)
    
    def refresh_list(self):
        """刷新列表"""
        cargos = self.db.get_all_cargos()
        self.table.setRowCount(len(cargos))
        
        for i, cargo in enumerate(cargos):
            # 序号
            self.table.setItem(i, 0, QTableWidgetItem(str(i + 1)))
            
            # 货物名称
            self.table.setItem(i, 1, QTableWidgetItem(cargo['name']))
        
        self.table.resizeRowsToContents()
    
    def get_selected_cargo_id(self) -> int:
        """获取选中的货物ID"""
        current_row = self.table.currentRow()
        if current_row < 0:
            return -1
        
        cargos = self.db.get_all_cargos()
        if current_row >= len(cargos):
            return -1
        
        return cargos[current_row]['id']
    
    def add_cargo(self):
        """新增货物"""
        dialog = CargoDialog(self.db, parent=self)
        if dialog.exec_() == QDialog.Accepted:
            self.refresh_list()
    
    def edit_cargo(self):
        """修改货物"""
        cargo_id = self.get_selected_cargo_id()
        if cargo_id < 0:
            QMessageBox.warning(self, "提示", "请先选择要修改的货物")
            return
        
        cargos = self.db.get_all_cargos()
        cargo = next((c for c in cargos if c['id'] == cargo_id), None)
        if not cargo:
            QMessageBox.warning(self, "错误", "未找到该货物")
            return
        
        dialog = CargoDialog(self.db, cargo=cargo, parent=self)
        if dialog.exec_() == QDialog.Accepted:
            self.refresh_list()
    
    def delete_cargo(self):
        """删除货物"""
        cargo_id = self.get_selected_cargo_id()
        if cargo_id < 0:
            QMessageBox.warning(self, "提示", "请先选择要删除的货物")
            return
        
        # 检查是否被引用
        if self.db.is_cargo_used(cargo_id):
            QMessageBox.warning(self, "错误", "该货物已被地磅单引用，无法删除")
            return
        
        # 确认删除
        reply = QMessageBox.question(
            self, "确认", "确定要删除该货物吗？",
            QMessageBox.Yes | QMessageBox.No,
            QMessageBox.No
        )
        
        if reply == QMessageBox.Yes:
            if self.db.delete_cargo(cargo_id):
                QMessageBox.information(self, "成功", "删除成功！")
                self.refresh_list()
            else:
                QMessageBox.warning(self, "错误", "删除失败")


class CargoDialog(QDialog):
    """货物编辑对话框"""
    
    def __init__(self, db: Database, cargo=None, parent=None):
        super().__init__(parent)
        self.db = db
        self.cargo = cargo
        self.init_ui()
    
    def init_ui(self):
        """初始化对话框界面"""
        if self.cargo:
            self.setWindowTitle("修改货物名称")
        else:
            self.setWindowTitle("新增货物名称")
        
        self.setMinimumWidth(400)
        
        layout = QVBoxLayout()
        layout.setSpacing(15)
        layout.setContentsMargins(20, 20, 20, 20)
        
        # 表单
        form_layout = QFormLayout()
        form_layout.setSpacing(15)
        
        name_label = QLabel("货物名称 *:")
        name_label.setFont(QFont("微软雅黑", 16))
        
        self.name_input = QLineEdit()
        self.name_input.setMinimumHeight(45)
        self.name_input.setFont(QFont("微软雅黑", 16))
        
        if self.cargo:
            self.name_input.setText(self.cargo['name'])
        
        form_layout.addRow(name_label, self.name_input)
        
        layout.addLayout(form_layout)
        
        # 按钮
        button_layout = QHBoxLayout()
        button_layout.addStretch()
        
        ok_btn = QPushButton("确定")
        ok_btn.setMinimumHeight(45)
        ok_btn.setMinimumWidth(100)
        ok_btn.setFont(QFont("微软雅黑", 16))
        ok_btn.clicked.connect(self.save)
        
        cancel_btn = QPushButton("取消")
        cancel_btn.setMinimumHeight(45)
        cancel_btn.setMinimumWidth(100)
        cancel_btn.setFont(QFont("微软雅黑", 16))
        cancel_btn.clicked.connect(self.reject)
        
        button_layout.addWidget(ok_btn)
        button_layout.addWidget(cancel_btn)
        
        layout.addLayout(button_layout)
        self.setLayout(layout)
    
    def save(self):
        """保存"""
        name = self.name_input.text().strip()
        
        if not name:
            QMessageBox.warning(self, "验证失败", "货物名称不能为空")
            return
        
        # 检查重复
        if self.cargo:
            # 修改模式
            if self.db.cargo_exists(name, exclude_id=self.cargo['id']):
                QMessageBox.warning(self, "验证失败", "该名称已存在")
                return
            
            if self.db.update_cargo(self.cargo['id'], name):
                QMessageBox.information(self, "成功", "修改成功！")
                self.accept()
            else:
                QMessageBox.warning(self, "错误", "修改失败")
        else:
            # 新增模式
            if self.db.cargo_exists(name):
                QMessageBox.warning(self, "验证失败", "该名称已存在")
                return
            
            try:
                self.db.insert_cargo(name)
                QMessageBox.information(self, "成功", "新增成功！")
                self.accept()
            except Exception as e:
                QMessageBox.critical(self, "错误", f"新增失败：{str(e)}")