simple-dibang/ui/company_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 CompanyTab(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_company)
        # 添加快捷键
        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_company)
        
        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_company)
        # 添加快捷键
        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):
        """刷新列表"""
        companies = self.db.get_all_companies()
        self.table.setRowCount(len(companies))
        
        for i, company in enumerate(companies):
            # 序号
            self.table.setItem(i, 0, QTableWidgetItem(str(i + 1)))
            
            # 公司名称
            self.table.setItem(i, 1, QTableWidgetItem(company['name']))
        
        self.table.resizeRowsToContents()
    
    def get_selected_company_id(self) -> int:
        """获取选中的公司ID"""
        current_row = self.table.currentRow()
        if current_row < 0:
            return -1
        
        companies = self.db.get_all_companies()
        if current_row >= len(companies):
            return -1
        
        return companies[current_row]['id']
    
    def add_company(self):
        """新增公司"""
        dialog = CompanyDialog(self.db, parent=self)
        if dialog.exec_() == QDialog.Accepted:
            self.refresh_list()
    
    def edit_company(self):
        """修改公司"""
        company_id = self.get_selected_company_id()
        if company_id < 0:
            QMessageBox.warning(self, "提示", "请先选择要修改的公司")
            return
        
        companies = self.db.get_all_companies()
        company = next((c for c in companies if c['id'] == company_id), None)
        if not company:
            QMessageBox.warning(self, "错误", "未找到该公司")
            return
        
        dialog = CompanyDialog(self.db, company=company, parent=self)
        if dialog.exec_() == QDialog.Accepted:
            self.refresh_list()
    
    def delete_company(self):
        """删除公司"""
        company_id = self.get_selected_company_id()
        if company_id < 0:
            QMessageBox.warning(self, "提示", "请先选择要删除的公司")
            return
        
        # 检查是否被引用
        if self.db.is_company_used(company_id):
            QMessageBox.warning(self, "错误", "该公司已被地磅单引用，无法删除")
            return
        
        # 确认删除
        reply = QMessageBox.question(
            self, "确认", "确定要删除该公司吗？",
            QMessageBox.Yes | QMessageBox.No,
            QMessageBox.No
        )
        
        if reply == QMessageBox.Yes:
            if self.db.delete_company(company_id):
                QMessageBox.information(self, "成功", "删除成功！")
                self.refresh_list()
            else:
                QMessageBox.warning(self, "错误", "删除失败")


class CompanyDialog(QDialog):
    """公司编辑对话框"""
    
    def __init__(self, db: Database, company=None, parent=None):
        super().__init__(parent)
        self.db = db
        self.company = company
        self.init_ui()
    
    def init_ui(self):
        """初始化对话框界面"""
        if self.company:
            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.company:
            self.name_input.setText(self.company['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.company:
            # 修改模式
            if self.db.company_exists(name, exclude_id=self.company['id']):
                QMessageBox.warning(self, "验证失败", "该名称已存在")
                return
            
            if self.db.update_company(self.company['id'], name):
                QMessageBox.information(self, "成功", "修改成功！")
                self.accept()
            else:
                QMessageBox.warning(self, "错误", "修改失败")
        else:
            # 新增模式
            if self.db.company_exists(name):
                QMessageBox.warning(self, "验证失败", "该名称已存在")
                return
            
            try:
                self.db.insert_company(name)
                QMessageBox.information(self, "成功", "新增成功！")
                self.accept()
            except Exception as e:
                QMessageBox.critical(self, "错误", f"新增失败：{str(e)}")