#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 试验计划视图，用于试验计划的安排与结果查看 """ from PyQt5.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QTableWidget, QTableWidgetItem, QHeaderView, QCheckBox, QButtonGroup, QGroupBox, QMessageBox, QComboBox, QLineEdit) from PyQt5.QtCore import Qt from models.dut_model import DUTModel from models.system_model import SystemModel from views.dut_form_dialog import DUTFormDialog from ui_utils.user_session import UserSession class TestPlanView(QWidget): def __init__(self): super().__init__() self.dut_model = DUTModel() self.system_model = SystemModel() # 用于加载机台数据 self.current_page = 1 self.page_size = 20 self.search_project = '' # 项目代码筛选 self.search_sn = '' # 样品SN筛选 self.init_ui() self.load_data() def showEvent(self, event): """每次显示视图时重新加载数据""" super().showEvent(event) # 重新从后端API加载数据 print("[测试计划] 视图显示，重新加载数据...") self.dut_model._load_duts() # 强制从API重新加载 self.load_data() # 刷新界面 def init_ui(self): """初始化界面""" layout = QVBoxLayout(self) layout.setContentsMargins(10, 10, 10, 10) layout.setSpacing(10) # 标题 title_label = QLabel("试验计划") title_label.setObjectName("title") layout.addWidget(title_label) # 搜索筛选栏（项目代码和样品SN） search_layout = QHBoxLayout() search_layout.setSpacing(10) # 项目代码搜索 project_label = QLabel("项目代码：") search_layout.addWidget(project_label) self.project_input = QLineEdit() self.project_input.setPlaceholderText("输入项目代码") self.project_input.setFixedWidth(150) self.project_input.textChanged.connect(self.on_search_changed) search_layout.addWidget(self.project_input) # 样品SN搜索 sn_label = QLabel("样品SN：") search_layout.addWidget(sn_label) self.sn_input = QLineEdit() self.sn_input.setPlaceholderText("输入样品SN") self.sn_input.setFixedWidth(200) self.sn_input.textChanged.connect(self.on_search_changed) search_layout.addWidget(self.sn_input) # 清空按钮 clear_btn = QPushButton("清空") clear_btn.clicked.connect(self.on_clear_search) search_layout.addWidget(clear_btn) search_layout.addStretch() layout.addLayout(search_layout) # 顶部筛选栏 filter_layout = QHBoxLayout() filter_layout.setSpacing(20) self.filter_btns = {} self.filter_group = QButtonGroup() self.filter_group.setExclusive(True) filters = ['待安排', '已取消', '待启动', '全部'] for idx, f in enumerate(filters): btn = QCheckBox(f) if f == '全部': btn.setChecked(True) btn.stateChanged.connect(lambda state, name=f: self.on_filter_change(name, state)) self.filter_btns[f] = btn self.filter_group.addButton(btn, idx) filter_layout.addWidget(btn) filter_layout.addStretch() # 新增、删除、复制按钮 self.btn_new = QPushButton("新增") self.btn_new.clicked.connect(self.on_new_plan) filter_layout.addWidget(self.btn_new) self.btn_delete = QPushButton("删除") self.btn_delete.clicked.connect(self.on_delete_plan) filter_layout.addWidget(self.btn_delete) self.btn_copy = QPushButton("复制") self.btn_copy.clicked.connect(self.on_copy_plan) filter_layout.addWidget(self.btn_copy) self.btn_edit = QPushButton("编辑") self.btn_edit.clicked.connect(self.on_edit_plan) filter_layout.addWidget(self.btn_edit) layout.addLayout(filter_layout) # 数据表格 self.table = QTableWidget() self.table.setColumnCount(11) self.table.setHorizontalHeaderLabels([ "序号", "产品序列号", "项目代码", "项目阶段", "项目方案", "测试周次", "测试设备", "试验状态", "当前试验方向", "已完成方向数", "完成日期" ]) # 表格样式 self.table.setSelectionBehavior(QTableWidget.SelectRows) self.table.setSelectionMode(QTableWidget.SingleSelection) self.table.setAlternatingRowColors(True) self.table.horizontalHeader().setStretchLastSection(True) self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Interactive) self.table.verticalHeader().setVisible(False) self.table.setEditTriggers(QTableWidget.NoEditTriggers) layout.addWidget(self.table) # 分页控制栏 pager_layout = QHBoxLayout() pager_layout.setSpacing(10) self.page_info_label = QLabel("") pager_layout.addWidget(self.page_info_label) pager_layout.addStretch() QLabel_page_size = QLabel("每页行数:") pager_layout.addWidget(QLabel_page_size) self.page_size_combo = QComboBox() self.page_size_combo.addItems(["20", "50", "100", "200"]) self.page_size_combo.setCurrentText(str(self.page_size)) self.page_size_combo.currentTextChanged.connect(self.on_page_size_changed) pager_layout.addWidget(self.page_size_combo) self.prev_btn = QPushButton("上一页") self.prev_btn.clicked.connect(self.go_prev_page) pager_layout.addWidget(self.prev_btn) self.next_btn = QPushButton("下一页") self.next_btn.clicked.connect(self.go_next_page) pager_layout.addWidget(self.next_btn) layout.addLayout(pager_layout) def on_search_changed(self): """搜索条件变化""" self.search_project = self.project_input.text().strip() self.search_sn = self.sn_input.text().strip() self.current_page = 1 # 重置到第一页 self.load_data() def on_clear_search(self): """清空搜索条件""" self.project_input.clear() self.sn_input.clear() self.search_project = '' self.search_sn = '' self.current_page = 1 self.load_data() def load_data(self): """加载数据并刷新分页""" # 根据当前筛选条件 current_filter = '全部' for name, btn in self.filter_btns.items(): if btn.isChecked(): current_filter = name break plans = self.dut_model.filterByStatus(current_filter if current_filter != '全部' else '') # 应用项目代码和样品SN筛选 filtered = [] for plan in plans: # 项目代码筛选 if self.search_project: project = str(plan.get('project', '')).lower() if self.search_project.lower() not in project: continue # 样品SN筛选 if self.search_sn: sn = str(plan.get('SN', '')).lower() if self.search_sn.lower() not in sn: continue filtered.append(plan) self.filtered_plans = filtered # 计算总页数并规范当前页 total = len(self.filtered_plans) total_pages = max(1, (total + self.page_size - 1) // self.page_size) if self.current_page > total_pages: self.current_page = total_pages if self.current_page < 1: self.current_page = 1 # 刷新表格与分页信息 self.refresh_table() self.update_pagination_info() self.table.resizeColumnsToContents() def on_filter_change(self, name, state): """筛选条件改变""" if state == Qt.Checked: for n, btn in self.filter_btns.items(): if n != name: btn.setChecked(False) self.load_data() def on_new_plan(self): """新增试验计划""" dlg = DUTFormDialog(mode="create", dut_form={}, system_model=self.system_model, parent=self) if dlg.exec_() == dlg.Accepted: form = dlg.get_form_dict() # 补充非表单字段的默认值 form.update({ 'name': '', 'projectType': '', 'itemOnGoing': '', 'itemsFinished': 0, 'status': '', 'deadLine': '', 'createdate': '', 'direction_codes': '' }) self.dut_model.addDUT(form) self.load_data() def on_edit_plan(self): """编辑选中试验计划""" row = self.table.currentRow() if row < 0 or row >= len(getattr(self, 'current_plans', [])): QMessageBox.warning(self, "编辑", "请先选中一行") return current = self.current_plans[row] dlg = DUTFormDialog(mode="edit", dut_form=current, system_model=self.system_model, parent=self) if dlg.exec_() == dlg.Accepted: form = dlg.get_form_dict() # 保留原有未在表单中的字段 for k in ['name','projectType','itemOnGoing','itemsFinished','status','deadLine','createdate','direction_codes']: if k not in form: form[k] = current.get(k,'') self.dut_model.updateDUT(form) self.load_data() def on_delete_plan(self): """删除试验计划""" row = self.table.currentRow() if row < 0 or row >= len(getattr(self, 'current_plans', [])): QMessageBox.warning(self, "删除", "请先选中一行") return plan_id = self.current_plans[row].get('id') if not plan_id: QMessageBox.warning(self, "删除", "当前行缺少ID，无法删除") return reply = QMessageBox.question(self, "确认删除", "确定要删除选中的试验计划吗？", QMessageBox.Yes | QMessageBox.No, QMessageBox.No) if reply != QMessageBox.Yes: return self.dut_model.removeDUT(plan_id) self.load_data() def on_copy_plan(self): """复制试验计划""" row = self.table.currentRow() if row < 0 or row >= len(getattr(self, 'current_plans', [])): QMessageBox.warning(self, "复制", "请先选中一行") return copy_plan = dict(self.current_plans[row]) copy_plan.pop('id', None) self.dut_model.addDUT(copy_plan) self.load_data() def refresh_table(self): """按当前页与每页条数刷新表格""" total = len(getattr(self, 'filtered_plans', [])) start = (self.current_page - 1) * self.page_size end = min(start + self.page_size, total) page_items = [] if total == 0 else self.filtered_plans[start:end] self.current_plans = page_items self.table.setRowCount(len(self.current_plans)) # 获取当前用户角色 user_role = UserSession.get_role() can_edit_status = user_role in ['admin', 'super'] for idx, plan in enumerate(self.current_plans): self.table.setItem(idx, 0, QTableWidgetItem(str(plan.get('index', idx+1)))) self.table.setItem(idx, 1, QTableWidgetItem(plan.get('SN', ''))) self.table.setItem(idx, 2, QTableWidgetItem(plan.get('project', ''))) self.table.setItem(idx, 3, QTableWidgetItem(plan.get('projectPhase', ''))) self.table.setItem(idx, 4, QTableWidgetItem(plan.get('testReq', ''))) self.table.setItem(idx, 5, QTableWidgetItem(str(plan.get('weeks', '')))) # 测试设备（带颜色） station_item = QTableWidgetItem(plan.get('stationAssigned', '')) station_text = plan.get('stationAssigned', '').lower() if 'com10' in station_text or 'com9' in station_text or 'com6' in station_text: station_item.setBackground(Qt.magenta) elif 'com7' in station_text or 'com1' in station_text: station_item.setBackground(Qt.green) elif 'com8' in station_text or 'com3' in station_text: station_item.setBackground(Qt.cyan) else: station_item.setBackground(Qt.blue) self.table.setItem(idx, 6, station_item) # 试验状态 - 使用下拉框（仅admin和super可编辑） status = plan.get('status', '') status_combo = QComboBox() status_combo.addItems(['待安排', '试验中', '暂停', '已完成', '已取消']) # 设置当前状态 if status in ['待安排', '试验中', '暂停', '已完成', '已取消']: status_combo.setCurrentText(status) else: # 处理旧状态名称 status_map = { '待启动': '待安排', '暂停中': '暂停' } new_status = status_map.get(status, '待安排') status_combo.setCurrentText(new_status) # 根据状态设置背景色 current_status = status_combo.currentText() if current_status == '已取消': status_combo.setStyleSheet("background-color: #d3d3d3;") elif current_status == '试验中': status_combo.setStyleSheet("background-color: #87CEEB;") elif current_status == '暂停': status_combo.setStyleSheet("background-color: #FFFF99;") elif current_status == '已完成': status_combo.setStyleSheet("background-color: #90EE90;") else: status_combo.setStyleSheet("background-color: white;") # 设置是否可编辑 status_combo.setEnabled(can_edit_status) # 绑定状态变化事件 status_combo.setProperty('row', idx) status_combo.setProperty('plan_sn', plan.get('SN', '')) status_combo.currentTextChanged.connect(self.on_status_changed) self.table.setCellWidget(idx, 7, status_combo) self.table.setItem(idx, 8, QTableWidgetItem(plan.get('currentDirection', ''))) self.table.setItem(idx, 9, QTableWidgetItem(str(plan.get('itemsFinished', '')))) self.table.setItem(idx, 10, QTableWidgetItem(plan.get('deadLine', ''))) self.table.resizeColumnsToContents() def on_status_changed(self, new_status): """状态下拉框变化事件""" sender = self.sender() row = sender.property('row') plan_sn = sender.property('plan_sn') if row is None or not plan_sn: return # 获取当前计划数据 if row >= len(self.current_plans): return plan = self.current_plans[row] old_status = plan.get('status', '') # 如果状态未变化，不处理 if old_status == new_status: return # 更新数据 plan['status'] = new_status try: # 调用model更新 self.dut_model.updateDUT(plan) print(f"[状态更新] 样品SN: {plan_sn}, 旧状态: {old_status}, 新状态: {new_status}") # 更新下拉框背景色 if new_status == '已取消': sender.setStyleSheet("background-color: #d3d3d3;") elif new_status == '试验中': sender.setStyleSheet("background-color: #87CEEB;") elif new_status == '暂停': sender.setStyleSheet("background-color: #FFFF99;") elif new_status == '已完成': sender.setStyleSheet("background-color: #90EE90;") else: sender.setStyleSheet("background-color: white;") except Exception as e: QMessageBox.warning(self, "错误", f"更新状态失败：{str(e)}") # 恢复为旧状态 sender.setCurrentText(old_status) def update_pagination_info(self): total = len(getattr(self, 'filtered_plans', [])) total_pages = max(1, (total + self.page_size - 1) // self.page_size) self.page_info_label.setText(f"共 {total} 条，{self.current_page}/{total_pages} 页，每页 {self.page_size} 条") self.prev_btn.setEnabled(self.current_page > 1) self.next_btn.setEnabled(self.current_page < total_pages) def on_page_size_changed(self, text): try: self.page_size = int(text) except Exception: self.page_size = 20 self.current_page = 1 self.refresh_table() self.update_pagination_info() def go_prev_page(self): if self.current_page > 1: self.current_page -= 1 self.refresh_table() self.update_pagination_info() def go_next_page(self): total = len(getattr(self, 'filtered_plans', [])) total_pages = max(1, (total + self.page_size - 1) // self.page_size) if self.current_page < total_pages: self.current_page += 1 self.refresh_table() self.update_pagination_info()