336 lines
13 KiB
Python
336 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
DUT表单对话框（全局复用）
|
||
- 新建/编辑公用一个对话框，根据模式禁用部分字段
|
||
- 数据以 dict(dut_form) 形式在模块间传递
|
||
- 设备分配为二级选择：机台 -> 通道
|
||
"""
|
||
|
||
from PyQt5.QtWidgets import (
|
||
QDialog, QVBoxLayout, QHBoxLayout, QFormLayout, QLineEdit, QComboBox,
|
||
QTextEdit, QPushButton, QLabel, QWidget
|
||
)
|
||
from PyQt5.QtCore import Qt
|
||
import requests
|
||
|
||
# 尝试导入配置
|
||
try:
|
||
from config import USE_HTTP_API, HTTP_API_BASE_URL
|
||
except ImportError:
|
||
USE_HTTP_API = False
|
||
HTTP_API_BASE_URL = "http://127.0.0.1:5050"
|
||
|
||
|
||
class DUTFormDialog(QDialog):
|
||
def __init__(self, mode="create", dut_form=None, lock_fields=None, system_model=None, parent=None):
|
||
super().__init__(parent)
|
||
self.mode = mode # "create" | "edit"
|
||
self.dut_form = dut_form or {}
|
||
self.lock_fields = set(lock_fields or [])
|
||
self.system_model = system_model # 用于加载机台数据
|
||
self.setWindowTitle("编辑试验样品详情" if self.mode == "edit" else "新建试验样品")
|
||
self.setModal(True)
|
||
self.resize(800, 480)
|
||
|
||
# HTTP API 配置
|
||
self.use_http_api = USE_HTTP_API
|
||
self.api_base_url = HTTP_API_BASE_URL
|
||
|
||
# 机台数据缓存
|
||
self.machines = []
|
||
self.current_machine = None
|
||
|
||
# 项目阶段数据缓存
|
||
self.project_phases = [] # 存储从后端获取的项目阶段列表
|
||
|
||
self._build_ui()
|
||
self._load_machines() # 加载机台数据
|
||
self._load_project_phases() # 加载项目阶段数据
|
||
self._fill_form(self.dut_form)
|
||
self._apply_lock()
|
||
|
||
def _build_ui(self):
|
||
layout = QVBoxLayout(self)
|
||
layout.setContentsMargins(20, 20, 20, 20)
|
||
layout.setSpacing(12)
|
||
|
||
#title = QLabel("编辑试验样品详情" if self.mode == "edit" else "新建试验样品详情")
|
||
#title.setObjectName("title")
|
||
#title.setAlignment(Qt.AlignLeft | Qt.AlignVCenter)
|
||
#layout.addWidget(title)
|
||
|
||
form = QFormLayout()
|
||
form.setLabelAlignment(Qt.AlignRight)
|
||
form.setFormAlignment(Qt.AlignTop)
|
||
form.setHorizontalSpacing(16)
|
||
form.setVerticalSpacing(12)
|
||
|
||
# 左右两列容器
|
||
row = QHBoxLayout()
|
||
left = QVBoxLayout()
|
||
right = QVBoxLayout()
|
||
|
||
# 左列控件
|
||
self.input_SN = QLineEdit()
|
||
self.input_project = QLineEdit()
|
||
self.input_testReq = QLineEdit()
|
||
self.input_phase = QComboBox()
|
||
self.input_phase.setEditable(True)
|
||
# 项目阶段选项将在 _load_project_phases() 中从后端加载
|
||
|
||
# 右列控件 - 设备分配（二级选择）
|
||
self.input_dtMachine = QComboBox() # 机台选择
|
||
self.input_dtMachine.setEditable(False)
|
||
self.input_dtMachine.addItem("", None) # 空选项
|
||
self.input_dtMachine.currentIndexChanged.connect(self._on_machine_changed)
|
||
|
||
self.input_station = QComboBox() # 通道选择
|
||
self.input_station.setEditable(False)
|
||
self.input_station.addItem("", None) # 空选项
|
||
|
||
self.input_weeks = QLineEdit()
|
||
self.input_workOrder = QLineEdit()
|
||
self.input_inspector = QLineEdit()
|
||
|
||
# 说明
|
||
self.input_description = QTextEdit()
|
||
self.input_description.setPlaceholderText("请输入内容")
|
||
|
||
# 左列布局
|
||
left_form = QFormLayout()
|
||
left_form.setLabelAlignment(Qt.AlignRight)
|
||
left_form.addRow("* 产品序列号", self.input_SN)
|
||
left_form.addRow("项目代码", self.input_project)
|
||
left_form.addRow("项目方案", self.input_testReq)
|
||
left_form.addRow("项目阶段", self.input_phase)
|
||
left.addLayout(left_form)
|
||
|
||
# 右列布局
|
||
right_form = QFormLayout()
|
||
right_form.setLabelAlignment(Qt.AlignRight)
|
||
right_form.addRow("机台选择", self.input_dtMachine)
|
||
right_form.addRow("通道选择", self.input_station)
|
||
right_form.addRow("测试周次", self.input_weeks)
|
||
right_form.addRow("工单名称", self.input_workOrder)
|
||
right_form.addRow("测试人员", self.input_inspector)
|
||
right.addLayout(right_form)
|
||
|
||
row.addLayout(left, 1)
|
||
row.addLayout(right, 1)
|
||
layout.addLayout(row)
|
||
|
||
# 说明一行单独处理
|
||
desc_form = QFormLayout()
|
||
desc_form.setLabelAlignment(Qt.AlignRight)
|
||
desc_form.addRow("说明", self.input_description)
|
||
layout.addLayout(desc_form)
|
||
|
||
# 底部按钮
|
||
btn_row = QHBoxLayout()
|
||
btn_row.addStretch()
|
||
self.btn_submit = QPushButton("修改" if self.mode == "edit" else "保存")
|
||
self.btn_close = QPushButton("关闭")
|
||
btn_row.addWidget(self.btn_submit)
|
||
btn_row.addWidget(self.btn_close)
|
||
layout.addLayout(btn_row)
|
||
|
||
self.btn_submit.clicked.connect(self.accept)
|
||
self.btn_close.clicked.connect(self.reject)
|
||
|
||
def _load_machines(self):
|
||
"""加载机台数据"""
|
||
if not self.system_model:
|
||
return
|
||
|
||
try:
|
||
self.machines = self.system_model.load_machines()
|
||
|
||
# 填充机台下拉框
|
||
self.input_dtMachine.clear()
|
||
self.input_dtMachine.addItem("", None) # 空选项
|
||
|
||
for machine in self.machines:
|
||
sn = machine.get("SN", "")
|
||
label = machine.get("label", "")
|
||
display = f"{sn} - {label}" if label else sn
|
||
self.input_dtMachine.addItem(display, machine)
|
||
except Exception as e:
|
||
print(f"加载机台数据失败: {e}")
|
||
|
||
def _load_project_phases(self):
|
||
"""从后端加载项目阶段数据"""
|
||
try:
|
||
# 尝试使用 HTTP API
|
||
if self.use_http_api:
|
||
try:
|
||
url = f"{self.api_base_url}/dbTableAccess"
|
||
response = requests.get(
|
||
url,
|
||
params={"table": "projectPhase"},
|
||
timeout=5
|
||
)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
data = result.get("data", [])
|
||
|
||
if data:
|
||
self.project_phases = data
|
||
print(f"从 HTTP API 加载了 {len(data)} 个项目阶段")
|
||
else:
|
||
print("后端返回的项目阶段数据为空")
|
||
self.project_phases = []
|
||
|
||
except Exception as e:
|
||
print(f"HTTP API 加载项目阶段失败: {e}，回退到 SystemModel")
|
||
# 回退到使用 SystemModel
|
||
if self.system_model:
|
||
self.project_phases = self.system_model.load_project_phases()
|
||
else:
|
||
self.project_phases = []
|
||
else:
|
||
# 直接使用 SystemModel
|
||
if self.system_model:
|
||
self.project_phases = self.system_model.load_project_phases()
|
||
else:
|
||
self.project_phases = []
|
||
|
||
# 填充下拉框：显示 name，存储 PN 作为 itemData
|
||
self.input_phase.clear()
|
||
for phase in self.project_phases:
|
||
name = phase.get("name", "")
|
||
pn = phase.get("PN", "")
|
||
if name: # 只添加有名称的项
|
||
self.input_phase.addItem(name, pn) # 显示 name，数据存储 PN
|
||
|
||
print(f"项目阶段下拉框已填充 {self.input_phase.count()} 个选项")
|
||
|
||
except Exception as e:
|
||
print(f"加载项目阶段数据失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
|
||
def _on_machine_changed(self, index):
|
||
"""机台选择变化时更新通道列表"""
|
||
self.input_station.clear()
|
||
self.input_station.addItem("", None) # 空选项
|
||
|
||
machine = self.input_dtMachine.currentData()
|
||
if not machine:
|
||
return
|
||
|
||
self.current_machine = machine
|
||
|
||
# 添加有效的通道（station1-4）
|
||
for i in range(1, 5):
|
||
station_key = f"station{i}"
|
||
status_key = f"status{i}"
|
||
try:
|
||
# 安全获取通道名称，处理None和空值
|
||
station_name = machine.get(station_key)
|
||
if station_name is None:
|
||
continue
|
||
|
||
station_name = str(station_name).strip()
|
||
station_status = machine.get(status_key, "")
|
||
|
||
# 过滤掉空值和 NA/N/A 的通道
|
||
if station_name and station_name.upper() not in ['NA', 'N/A']:
|
||
display = f"{i:02d} - {station_name}"
|
||
if station_status:
|
||
display += f" ({station_status})"
|
||
self.input_station.addItem(display, f"{i:02d}")
|
||
except Exception as e:
|
||
print(f"加载通道数据失败: {e}")
|
||
|
||
def _fill_form(self, f):
|
||
# 将已有数据填充到控件
|
||
self.input_SN.setText(str(f.get("SN", "")))
|
||
self.input_project.setText(str(f.get("project", "")))
|
||
self.input_testReq.setText(str(f.get("testReq", "")))
|
||
|
||
# 项目阶段：f.get("phase") 存储的是 PN 值，需要匹配 itemData
|
||
phase_pn = str(f.get("phase", ""))
|
||
if phase_pn:
|
||
# 查找匹配的 PN
|
||
found = False
|
||
for i in range(self.input_phase.count()):
|
||
item_pn = self.input_phase.itemData(i)
|
||
if item_pn == phase_pn:
|
||
self.input_phase.setCurrentIndex(i)
|
||
found = True
|
||
break
|
||
|
||
# 如果没有找到匹配项，使用可编辑功能填充
|
||
if not found:
|
||
self.input_phase.setEditText(phase_pn)
|
||
|
||
# 设备分配：使用 Model 解析后的 dtMachine 和 station 字段
|
||
dtm = f.get("dtMachine", "").strip()
|
||
station = f.get("station", "").strip()
|
||
|
||
# 选中机台
|
||
if dtm:
|
||
for i in range(self.input_dtMachine.count()):
|
||
machine = self.input_dtMachine.itemData(i)
|
||
if machine and machine.get("SN") == dtm:
|
||
self.input_dtMachine.setCurrentIndex(i)
|
||
break
|
||
|
||
# 选中通道（需要在机台选中后）
|
||
if station:
|
||
for i in range(self.input_station.count()):
|
||
station_num = self.input_station.itemData(i)
|
||
if station_num == station:
|
||
self.input_station.setCurrentIndex(i)
|
||
break
|
||
|
||
self.input_weeks.setText(str(f.get("weeks", "")))
|
||
self.input_workOrder.setText(str(f.get("workOrder", "")))
|
||
self.input_inspector.setText(str(f.get("inspector", "")))
|
||
self.input_description.setPlainText(str(f.get("description", "")))
|
||
|
||
def _apply_lock(self):
|
||
# 根据锁定字段禁用编辑（默认在编辑模式锁定 SN）
|
||
defaults = {"SN"} if self.mode == "edit" else set()
|
||
locked = defaults | self.lock_fields
|
||
if "SN" in locked:
|
||
self.input_SN.setDisabled(True)
|
||
# 注意：设备分配在编辑模式下是可以更改的
|
||
|
||
def get_form_dict(self):
|
||
# 以 dut_form 的结构返回数据
|
||
# 项目阶段：优先获取 itemData（PN值），如果没有则使用用户输入的文本
|
||
phase_pn = self.input_phase.currentData() # 获取当前选中项的 PN
|
||
if phase_pn is None: # 如果是用户手动输入的文本
|
||
phase_pn = self.input_phase.currentText().strip()
|
||
|
||
dut_form = {
|
||
"SN": self.input_SN.text().strip(),
|
||
"project": self.input_project.text().strip(),
|
||
"testReq": self.input_testReq.text().strip(),
|
||
"phase": phase_pn, # 存储 PN 值
|
||
"weeks": int(self.input_weeks.text() or 0),
|
||
"workOrder": self.input_workOrder.text().strip(),
|
||
"inspector": self.input_inspector.text().strip(),
|
||
"description": self.input_description.toPlainText().strip(),
|
||
}
|
||
|
||
# 设备分配：从二级选择中获取 dtMachine 和 station
|
||
# Model 会自动将这两个字段合成为 JSON 字符串存入 stationAssigned
|
||
machine = self.input_dtMachine.currentData()
|
||
station_num = self.input_station.currentData()
|
||
|
||
if machine and station_num:
|
||
dut_form["dtMachine"] = machine.get("SN", "")
|
||
dut_form["station"] = station_num
|
||
else:
|
||
dut_form["dtMachine"] = ""
|
||
dut_form["station"] = ""
|
||
|
||
# 保留可能存在的 id 等字段
|
||
if "id" in self.dut_form:
|
||
dut_form["id"] = self.dut_form["id"]
|
||
return dut_form
|