215 lines
7.6 KiB
Python
215 lines
7.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
系统设置 ViewModel - 响应式数据管理
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from PyQt5.QtCore import QObject, pyqtSignal, pyqtProperty
|
|||
|
|
from models.system_model import SystemModel
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SystemSettingsViewModel(QObject):
|
|||
|
|
"""系统设置的响应式 ViewModel"""
|
|||
|
|
|
|||
|
|
# 信号：数据变化时自动触发
|
|||
|
|
machinesChanged = pyqtSignal()
|
|||
|
|
phasesChanged = pyqtSignal()
|
|||
|
|
usersChanged = pyqtSignal()
|
|||
|
|
currentMachineChanged = pyqtSignal()
|
|||
|
|
plcDataChanged = pyqtSignal()
|
|||
|
|
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self.model = SystemModel()
|
|||
|
|
|
|||
|
|
# 数据缓存
|
|||
|
|
self._machines = []
|
|||
|
|
self._phases = []
|
|||
|
|
self._users = []
|
|||
|
|
self._current_machine_id = None
|
|||
|
|
self._plc_data = {}
|
|||
|
|
|
|||
|
|
# 删除标记
|
|||
|
|
self.removed_machine_ids = set()
|
|||
|
|
self.removed_phase_ids = set()
|
|||
|
|
self.removed_user_ids = set()
|
|||
|
|
|
|||
|
|
# ============ 机台列表 ============
|
|||
|
|
@pyqtProperty(list, notify=machinesChanged)
|
|||
|
|
def machines(self):
|
|||
|
|
"""机台列表（响应式属性）"""
|
|||
|
|
return self._machines
|
|||
|
|
|
|||
|
|
def loadMachines(self):
|
|||
|
|
"""加载机台数据"""
|
|||
|
|
self._machines = self.model.load_machines()
|
|||
|
|
self.machinesChanged.emit()
|
|||
|
|
|
|||
|
|
def saveMachines(self, machines_data):
|
|||
|
|
"""保存机台数据"""
|
|||
|
|
self.model.save_machines(machines_data, list(self.removed_machine_ids))
|
|||
|
|
self.removed_machine_ids.clear()
|
|||
|
|
self.loadMachines() # 重新加载以刷新
|
|||
|
|
# 如果当前有选中的机台，重新加载 PLC 数据
|
|||
|
|
if self._current_machine_id:
|
|||
|
|
self._loadPlcData()
|
|||
|
|
|
|||
|
|
def deleteMachines(self, machine_ids):
|
|||
|
|
"""删除机台数据"""
|
|||
|
|
if not machine_ids:
|
|||
|
|
return False
|
|||
|
|
delete_records = [{"id": mid} for mid in machine_ids]
|
|||
|
|
success = self.model.delete_machines(delete_records)
|
|||
|
|
if success:
|
|||
|
|
self.loadMachines() # 重新加载以刷新
|
|||
|
|
return success
|
|||
|
|
|
|||
|
|
# ============ 样品阶段列表 ============
|
|||
|
|
@pyqtProperty(list, notify=phasesChanged)
|
|||
|
|
def phases(self):
|
|||
|
|
"""样品阶段列表（响应式属性）"""
|
|||
|
|
return self._phases
|
|||
|
|
|
|||
|
|
def loadPhases(self):
|
|||
|
|
"""加载样品阶段数据"""
|
|||
|
|
self._phases = self.model.load_project_phases()
|
|||
|
|
self.phasesChanged.emit()
|
|||
|
|
|
|||
|
|
def savePhases(self, phases_data):
|
|||
|
|
"""保存样品阶段数据"""
|
|||
|
|
self.model.save_project_phases(phases_data, list(self.removed_phase_ids))
|
|||
|
|
self.removed_phase_ids.clear()
|
|||
|
|
self.loadPhases()
|
|||
|
|
|
|||
|
|
def deletePhases(self, phase_ids):
|
|||
|
|
"""删除样品阶段数据"""
|
|||
|
|
if not phase_ids:
|
|||
|
|
return False
|
|||
|
|
delete_records = [{"id": pid} for pid in phase_ids]
|
|||
|
|
success = self.model.delete_project_phases(delete_records)
|
|||
|
|
if success:
|
|||
|
|
self.loadPhases() # 重新加载以刷新
|
|||
|
|
return success
|
|||
|
|
|
|||
|
|
# ============ 用户列表 ============
|
|||
|
|
@pyqtProperty(list, notify=usersChanged)
|
|||
|
|
def users(self):
|
|||
|
|
"""用户列表（响应式属性）"""
|
|||
|
|
return self._users
|
|||
|
|
|
|||
|
|
def loadUsers(self):
|
|||
|
|
"""加载用户数据"""
|
|||
|
|
self._users = self.model.load_users()
|
|||
|
|
self.usersChanged.emit()
|
|||
|
|
|
|||
|
|
def saveUsers(self, users_data):
|
|||
|
|
"""保存用户数据"""
|
|||
|
|
self.model.save_users(users_data, list(self.removed_user_ids))
|
|||
|
|
self.removed_user_ids.clear()
|
|||
|
|
self.loadUsers()
|
|||
|
|
|
|||
|
|
def deleteUsers(self, user_ids):
|
|||
|
|
"""删除用户数据"""
|
|||
|
|
if not user_ids:
|
|||
|
|
return False
|
|||
|
|
delete_records = [{"id": uid} for uid in user_ids]
|
|||
|
|
success = self.model.delete_users(delete_records)
|
|||
|
|
if success:
|
|||
|
|
self.loadUsers() # 重新加载以刷新
|
|||
|
|
return success
|
|||
|
|
|
|||
|
|
# ============ 当前选中机台 ============
|
|||
|
|
@pyqtProperty(str, notify=currentMachineChanged)
|
|||
|
|
def currentMachineId(self):
|
|||
|
|
"""当前选中的机台 ID（响应式属性）"""
|
|||
|
|
return self._current_machine_id or ""
|
|||
|
|
|
|||
|
|
def setCurrentMachineId(self, machine_id):
|
|||
|
|
"""设置当前机台 ID，自动触发 PLC 数据加载"""
|
|||
|
|
if self._current_machine_id != machine_id:
|
|||
|
|
self._current_machine_id = machine_id
|
|||
|
|
self.currentMachineChanged.emit()
|
|||
|
|
self._loadPlcData() # 自动加载 PLC 数据
|
|||
|
|
|
|||
|
|
# ============ PLC 参数数据 ============
|
|||
|
|
@pyqtProperty('QVariantMap', notify=plcDataChanged)
|
|||
|
|
def plcData(self):
|
|||
|
|
"""当前机台的 PLC 参数（响应式属性）"""
|
|||
|
|
return self._plc_data
|
|||
|
|
|
|||
|
|
def _loadPlcData(self):
|
|||
|
|
"""私有方法：根据当前机台 ID 加载 PLC 数据"""
|
|||
|
|
if not self._current_machine_id:
|
|||
|
|
self._plc_data = {}
|
|||
|
|
self.plcDataChanged.emit()
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 从缓存的机台列表中查找
|
|||
|
|
machine = next((m for m in self._machines if m.get("id") == self._current_machine_id), None)
|
|||
|
|
if not machine:
|
|||
|
|
self._plc_data = {}
|
|||
|
|
else:
|
|||
|
|
# 提取 PLC 相关字段
|
|||
|
|
self._plc_data = {
|
|||
|
|
"plcAddress": str(machine.get("plcAddress", "")),
|
|||
|
|
"com": str(machine.get("com", "")),
|
|||
|
|
"testType": str(machine.get("testType", "")),
|
|||
|
|
"baudrate": machine.get("baudrate", {}),
|
|||
|
|
"status1": str(machine.get("status1", "")) in ("1", "true", "enabled"),
|
|||
|
|
"status2": str(machine.get("status2", "")) in ("1", "true", "enabled"),
|
|||
|
|
"status3": str(machine.get("status3", "")) in ("1", "true", "enabled"),
|
|||
|
|
"status4": str(machine.get("status4", "")) in ("1", "true", "enabled"),
|
|||
|
|
"register01": machine.get("register01", {}),
|
|||
|
|
"register02": machine.get("register02", {}),
|
|||
|
|
"register03": machine.get("register03", {}),
|
|||
|
|
"register04": machine.get("register04", {}),
|
|||
|
|
}
|
|||
|
|
# print(f"Loaded PLC data: {self._plc_data}")
|
|||
|
|
|
|||
|
|
self.plcDataChanged.emit() # 触发界面刷新
|
|||
|
|
|
|||
|
|
def savePlcData(self, plc_updates):
|
|||
|
|
"""保存 PLC 参数到当前机台
|
|||
|
|
重要：只传入 PLC 相关字段，不传入机台基本信息；必须包含 SN 字段用于后台更新
|
|||
|
|
"""
|
|||
|
|
if not self._current_machine_id:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 找到当前机台
|
|||
|
|
machine = next((m for m in self._machines if m.get("id") == self._current_machine_id), None)
|
|||
|
|
if not machine:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 必须传入 id 和 SN（后台通过 SN 更新），以及 PLC 相关字段
|
|||
|
|
update_data = {
|
|||
|
|
"id": self._current_machine_id, # 用于 SQLite 直接访问时的 WHERE 条件
|
|||
|
|
"SN": machine.get("SN"), # 用于 HTTP API 后台更新（update_key）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 只添加非 None 的 PLC 字段
|
|||
|
|
for key, value in plc_updates.items():
|
|||
|
|
if value is not None:
|
|||
|
|
update_data[key] = value
|
|||
|
|
|
|||
|
|
# 保存到数据库（只更新传入的字段）
|
|||
|
|
self.model.save_machines([update_data])
|
|||
|
|
|
|||
|
|
# 重新加载机台列表和 PLC 数据
|
|||
|
|
self.loadMachines()
|
|||
|
|
self._loadPlcData()
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# ============ 初始化加载 ============
|
|||
|
|
def loadAll(self):
|
|||
|
|
"""加载所有数据"""
|
|||
|
|
self.loadMachines()
|
|||
|
|
self.loadPhases()
|
|||
|
|
self.loadUsers()
|
|||
|
|
|
|||
|
|
# 默认选中第一个机台
|
|||
|
|
if self._machines:
|
|||
|
|
self.setCurrentMachineId(self._machines[0].get("id"))
|