690 lines
36 KiB
Python
690 lines
36 KiB
Python
import json
|
||
import base64
|
||
import os
|
||
from PyQt6 import *
|
||
from PyQt6.QtCore import *
|
||
from PyQt6.QtGui import *
|
||
from PyQt6.QtWidgets import *
|
||
from ui.Ui_projectForm import Ui_projectForm
|
||
from projectModel.projectManager import projectManager
|
||
from insertProTaskDlg import InsertProTaskDlg
|
||
from userModel.userManager import userManager
|
||
from taskModel.taskManager import taskManager
|
||
from createProjectDlg import CreateProjectDlg
|
||
from instructionModel.instructionManager import instructionManager
|
||
from taskModel.taskActuatorManager import taskActuatorManager
|
||
from deviceModel.deviceManager import deviceManager
|
||
from devmodelModel.devModelManager import devModelManager
|
||
from taskInstructionModel.taskInstructionManager import taskInstructionManager
|
||
from taskModel.taskActuatorManager import taskActuatorManager
|
||
from config import config
|
||
from common import common
|
||
from common import NoWheelComboBox
|
||
from common import TaskProgressBar
|
||
from common import AlignDelegate
|
||
from logs import log
|
||
from influxDB import influxdb
|
||
class QxStandardItem(QStandardItem):
|
||
def __init__(self, text=''):
|
||
super().__init__(text)
|
||
def updateProgress(self, taskId, value, maxValue):
|
||
obj = self.data(Qt.ItemDataRole.UserRole)
|
||
if obj is not None and isinstance(obj, dict):
|
||
if str(taskId) == str(obj.get('task_instruction_id')):
|
||
self.setData({'value': value, 'maximum': maxValue, 'task_instruction_id': obj.get('task_instruction_id'), 'name':obj.get('name')}, Qt.ItemDataRole.UserRole)
|
||
class ProgressBarDelegate(QStyledItemDelegate):
|
||
def paint(self, painter, option, index):
|
||
if index.column() == 5: # 仅对第二列添加进度条
|
||
data = index.data(Qt.ItemDataRole.UserRole) # 获取进度条数据
|
||
if data is not None and isinstance(data, dict):
|
||
value = data.get('value', -9999)
|
||
maximum = data.get('maximum', -9999)
|
||
progress_rect = option.rect # 进度条的矩形边界
|
||
progress_width = int((value / maximum) * progress_rect.width()) # 计算进度条的宽度
|
||
progress_rect.adjust(2, 2, -2, -2)
|
||
# 绘制进度条背景
|
||
painter.fillRect(progress_rect, QColor("#F8FAFF"))
|
||
# 绘制实际数值文本
|
||
text = f"{value}/{maximum}"
|
||
text_rect = QRect(option.rect)
|
||
# text_rect.adjust(0, 0, -1, 0) # 调整文本矩形边界
|
||
# 修改字体和颜色
|
||
# font = QFont("Arial", 10, QFont.Weight.Bold) # 创建字体对象
|
||
# painter.setFont(font) # 设置字体
|
||
pen = painter.pen()
|
||
pen.setColor(Qt.GlobalColor.black) # 设置画笔颜色为红色
|
||
painter.setPen(pen)
|
||
# 绘制实际进度
|
||
progress_rect.setWidth(progress_width)
|
||
if value == -1 and maximum == -1:
|
||
return
|
||
if value >=0 and maximum >= 1 :
|
||
painter.fillRect(progress_rect, QColor("#007EFF"))
|
||
painter.drawText(text_rect, Qt.AlignmentFlag.AlignCenter, text)
|
||
|
||
else:
|
||
super().paint(painter, option, index)
|
||
class ProjectForm(QWidget):
|
||
TASK = 0
|
||
VIRTUAL = 1
|
||
INSTRUCTION = 2
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.ui = Ui_projectForm()
|
||
self.ui.setupUi(self)
|
||
self.settings = QSettings("OCE", "PlatFormPlus")
|
||
|
||
self.currentProjectIndex = self.settings.value("currentProjectIndex", defaultValue=0)
|
||
self.targetIds = []
|
||
self.taskModelBlockSignals = False
|
||
self.targetParams = {}
|
||
self.topWidget = None
|
||
self.supportedTypes = instructionManager.getSupportedType()
|
||
self.createProjectDlg = CreateProjectDlg()
|
||
self.insertTaskDlg = InsertProTaskDlg()
|
||
self.insertTaskDlg.accepted.connect(self.refreshTask)
|
||
self.taskModel = QStandardItemModel()
|
||
self.ui.tableViewTask.setModel(self.taskModel)
|
||
self.taskDetailModel = QStandardItemModel()
|
||
self.ui.tableViewTaskDetail.setModel(self.taskDetailModel)
|
||
self.taskModel.itemChanged.connect(self.taskChanged)
|
||
userManager.loginSignal.connect(self.loadProject)
|
||
self.ui.pbAdd.clicked.connect(self.addProject)
|
||
self.ui.pbDelete.clicked.connect(self.deleteProject)
|
||
self.ui.pbEdit.clicked.connect(self.editProject)
|
||
self.basePath = config.data["project"]["base_path"]
|
||
self.createProjectDlg.createFinished.connect(self.createProFinished)
|
||
self.createProjectDlg.editFinished.connect(self.editProFinished)
|
||
self.projectModel = QStandardItemModel()
|
||
self.propertyModel = QStandardItemModel()
|
||
self.proHistoryModel = QStandardItemModel()
|
||
self.fileSystemModel = QFileSystemModel()
|
||
self.ui.tableViewPro.setModel(self.projectModel)
|
||
self.ui.tableViewProperty.setModel(self.propertyModel)
|
||
self.ui.tableViewHistory.setModel(self.proHistoryModel)
|
||
self.fileSystemModel.setRootPath(self.basePath)
|
||
self.ui.pbInserTask.clicked.connect(self.insertTask)
|
||
self.ui.pbDeleteTask.clicked.connect(self.deleteTask)
|
||
self.ui.pbStartTask.clicked.connect(self.startTask)
|
||
self.ui.pbStopTask.clicked.connect(self.stopTask)
|
||
self.ui.pbAddProperty.clicked.connect(self.addProperty)
|
||
self.ui.pbDeleteProperty.clicked.connect(self.deleteProperty)
|
||
self.propertyModel.itemChanged.connect(self.updateProperty)
|
||
self.ui.tableViewPro.selectionModel().selectionChanged.connect(self.handleProSelection)
|
||
|
||
self.ui.tableViewTask.selectionModel().selectionChanged.connect(self.handleTaskSelection)
|
||
self.ui.treeViewFile.setModel(self.fileSystemModel)
|
||
self.ui.treeViewFile.doubleClicked.connect(self.handleFileOpen)
|
||
|
||
self.batchSetItemDelegate()
|
||
self.setSplitterStretchFactor()
|
||
taskActuatorManager.updateProgress.connect(self._onUpdateProgress)
|
||
# self.batchInstallEventFilter()
|
||
|
||
def batchSetItemDelegate(self):
|
||
self.ui.tableViewTask.horizontalHeader().setDefaultAlignment(Qt.AlignmentFlag.AlignLeft|Qt.AlignmentFlag.AlignVCenter)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setDefaultAlignment(Qt.AlignmentFlag.AlignLeft|Qt.AlignmentFlag.AlignVCenter)
|
||
self.ui.tableViewHistory.horizontalHeader().setDefaultAlignment(Qt.AlignmentFlag.AlignLeft|Qt.AlignmentFlag.AlignVCenter)
|
||
self.ui.tableViewPro.horizontalHeader().setDefaultAlignment(Qt.AlignmentFlag.AlignLeft|Qt.AlignmentFlag.AlignVCenter)
|
||
self.ui.tableViewProperty.horizontalHeader().setDefaultAlignment(Qt.AlignmentFlag.AlignLeft|Qt.AlignmentFlag.AlignVCenter)
|
||
self.ui.tableViewTask.setItemDelegate(AlignDelegate(self.ui.tableViewTask))
|
||
self.ui.tableViewTaskDetail.setItemDelegate(AlignDelegate(self.ui.tableViewTaskDetail))
|
||
self.ui.tableViewHistory.setItemDelegate(AlignDelegate(self.ui.tableViewHistory))
|
||
self.ui.tableViewProperty.setItemDelegate(AlignDelegate(self.ui.tableViewProperty))
|
||
self.ui.tableViewPro.setItemDelegate(AlignDelegate(self.ui.tableViewPro))
|
||
|
||
# def batchInstallEventFilter(self):
|
||
# self.ui.w1.resizeEvent = self.onResize
|
||
# self.ui.w2.resizeEvent = self.onResize
|
||
# self.ui.w3.resizeEvent = self.onResize
|
||
|
||
# def onResize(self, event):
|
||
# self.setAllColumnWidth()
|
||
def taskChanged(self):
|
||
if self.taskModelBlockSignals == True:
|
||
return
|
||
# 遍历self.taskModel中的每一行，获取每一行的任务ID和参数，并存储到self.targetParams字典中
|
||
for row in range(self.taskModel.rowCount()):
|
||
task_id_item = self.taskModel.item(row, 0)
|
||
param_item = self.taskModel.item(row, 4)
|
||
task_id = ""
|
||
if task_id_item:
|
||
task_id = task_id_item.text()
|
||
param = ""
|
||
if param_item:
|
||
param = param_item.text()
|
||
|
||
if task_id != "":
|
||
self.targetParams[task_id] = param
|
||
projectManager.saveProTask(self.targetIds, self.targetParams)
|
||
|
||
def refreshTask(self):
|
||
for target_id in self.insertTaskDlg.targetIds:
|
||
if target_id not in self.targetIds:
|
||
self.targetIds.append(target_id)
|
||
self.targetParams[target_id] = ""
|
||
projectManager.saveProTask(self.targetIds, self.targetParams)
|
||
self.loadTask()
|
||
|
||
def loadTask(self):
|
||
self.taskModel.clear()
|
||
self.taskDetailModel.clear()
|
||
tasks = taskManager.getInfos(self.targetIds)
|
||
self.taskModelBlockSignals = True
|
||
if len(tasks) > 0:
|
||
|
||
for task in tasks:
|
||
index = self.taskModel.rowCount()
|
||
task_id = str(task["id"])
|
||
param = ""
|
||
if self.targetParams:
|
||
param = task_id in self.targetParams and self.targetParams[task_id] or ""
|
||
progressItem = QxStandardItem()
|
||
progressItem.setData({'task_instruction_id': task["id"], 'name': task["name"]}, Qt.ItemDataRole.UserRole)
|
||
self.taskModel.setItem(index, 0, QStandardItem(str(task["id"])))
|
||
self.taskModel.setItem(index, 1, QStandardItem(task["name"]))
|
||
self.taskModel.setItem(index, 2, QStandardItem(str(task["loop"])))
|
||
self.taskModel.setItem(index, 3, QStandardItem(str(task["delay"])))
|
||
self.taskModel.setItem(index, 4, QStandardItem(param))
|
||
self.taskModel.setItem(index, 5, progressItem)
|
||
self.taskModel.setItem(index, 6, QStandardItem(str(task["group_id"])))
|
||
self.setUnEditableStyle(self.taskModel.item(index, 1))
|
||
self.setUnEditableStyle(self.taskModel.item(index, 2))
|
||
self.setUnEditableStyle(self.taskModel.item(index, 3))
|
||
self.setUnEditableStyle(self.taskModel.item(index, 5))
|
||
delegate = ProgressBarDelegate()
|
||
self.ui.tableViewTask.setItemDelegateForColumn(5, delegate)
|
||
self.taskModelBlockSignals = False
|
||
self.batchOperation()
|
||
|
||
def _onUpdateProgress(self, taskId, value, maxValue):
|
||
for row in range(self.taskModel.rowCount()):
|
||
item = self.taskModel.item(row, 5)
|
||
if item is not None:
|
||
obj = item.data(Qt.ItemDataRole.UserRole)
|
||
if obj is not None and isinstance(obj, dict):
|
||
if str(taskId) == str(obj.get('task_instruction_id')):
|
||
self.taskModelBlockSignals = True
|
||
item.setData({'value': value, 'maximum': maxValue, 'task_instruction_id': obj.get('task_instruction_id'), 'name': obj.get('name')}, Qt.ItemDataRole.UserRole)
|
||
self.taskModelBlockSignals = False
|
||
return
|
||
|
||
def insertTask(self):
|
||
self.insertTaskDlg.showDlg()
|
||
|
||
def deleteTask(self):
|
||
if len(self.ui.tableViewTask.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewTask.selectionModel().selectedRows()[0].row()
|
||
row_taskId_index = self.taskModel.index(row, 0, QModelIndex())
|
||
row_taskName_index = self.taskModel.index(row, 1, QModelIndex())
|
||
taskid = self.taskModel.data(row_taskId_index, Qt.ItemDataRole.DisplayRole)
|
||
taskName = self.taskModel.data(row_taskName_index, Qt.ItemDataRole.DisplayRole)
|
||
if self.topWidget == None:
|
||
self.topWidget = self.findTopLevelParent(self)
|
||
if self.topWidget != None:
|
||
if self.topWidget.removeDialog(taskName,"删除用例"):
|
||
self.taskDetailModel.clear()
|
||
self.taskModel.removeRow(row)
|
||
self.targetIds.remove(taskid)
|
||
self.targetParams.pop(taskid, None)
|
||
projectManager.saveProTask(self.targetIds, self.targetParams)
|
||
self.batchOperation()
|
||
|
||
def startTask(self):
|
||
for row in self.ui.tableViewTask.selectionModel().selectedRows():
|
||
row_index = self.taskModel.index(row.row(), 0, QModelIndex())
|
||
taskid = self.taskModel.data(row_index, Qt.ItemDataRole.DisplayRole)
|
||
params = taskid in self.targetParams and self.targetParams[taskid] or None
|
||
taskActuatorManager.start(taskid, params)
|
||
|
||
def stopTask(self):
|
||
for row in range(self.taskModel.rowCount()):
|
||
task_id_item = self.taskModel.item(row, 0)
|
||
taskid = task_id_item.text()
|
||
taskActuatorManager.stop(taskid)
|
||
|
||
#任务列表选中
|
||
def handleTaskSelection(self, index):
|
||
if len(self.ui.tableViewTask.selectionModel().selectedRows()) == 1:
|
||
self.refreshTaskDetail()
|
||
|
||
def refreshTaskDetail(self):
|
||
if len(self.ui.tableViewTask.selectionModel().selectedRows()) == 1:
|
||
row = self.ui.tableViewTask.selectionModel().selectedRows()[0].row()
|
||
row_index = self.taskModel.index(row, 0, QModelIndex())
|
||
taskid = self.taskModel.data(row_index, Qt.ItemDataRole.DisplayRole)
|
||
self.loadTaskDetail(taskid)
|
||
|
||
def setItemsEnable(self, enable):
|
||
if not enable:
|
||
self.ui.tableViewTask.setSelectionMode(QAbstractItemView.SelectionMode.NoSelection)
|
||
else:
|
||
self.ui.tableViewTask.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
|
||
|
||
|
||
#加载任务详情
|
||
def loadTaskDetail(self, taskId):
|
||
self.setItemsEnable(False)
|
||
self.taskDetailModelBlockSignals = True
|
||
self.taskDetailModel.clear()
|
||
self.loadPageTaskDetail(taskId, 0, 50)
|
||
self.taskDetailModelBlockSignals = False
|
||
self.setItemsEnable(True)
|
||
self.batchOperation()
|
||
|
||
def loadPageTaskDetail(self, taskId, row, limit):
|
||
QCoreApplication.processEvents()
|
||
if taskId:
|
||
taskInstructions = taskInstructionManager.getPageInfo(taskId,row, limit)
|
||
if len(taskInstructions) > 0:
|
||
for taskInstruction in taskInstructions:
|
||
index = self.taskDetailModel.rowCount()
|
||
self.taskDetailModel.setItem(index, 0, QStandardItem(str(taskInstruction["id"])))
|
||
self.taskDetailModel.setItem(index, 1, QStandardItem(str(taskInstruction["task_id"])))
|
||
name = ""
|
||
type = ""
|
||
devicename = ""
|
||
dev_model_id = ""
|
||
devinterface = []
|
||
if taskInstruction["target_type"] == "instruction":
|
||
instruction = instructionManager.getInfo(str(taskInstruction["target_id"]))
|
||
if instruction:
|
||
name = instruction["name"]
|
||
type = instruction["type"]
|
||
if instruction["type"] is not None and instruction["type"] in self.supportedTypes:
|
||
type = self.supportedTypes[instruction["type"]]["name"]
|
||
device_id = taskInstruction["device_id"]
|
||
device = deviceManager.getInfo(device_id)
|
||
if device != None:
|
||
devicename = device["name"]
|
||
dev_model_id = device["dev_model_id"]
|
||
devinterface = device["interface"]
|
||
elif taskInstruction["target_type"] == "task":
|
||
task = taskManager.getInfo(taskInstruction["target_id"])
|
||
if task != None:
|
||
devicename = "任务"
|
||
name = task["name"]
|
||
|
||
self.taskDetailModel.setItem(index, 2, QStandardItem(devicename))
|
||
self.taskDetailModel.setItem(index, 3, QStandardItem(name))
|
||
self.taskDetailModel.setItem(index, 4, QStandardItem(type))
|
||
self.taskDetailModel.setItem(index, 5, QStandardItem(str(taskInstruction["delay"])))
|
||
self.taskDetailModel.setItem(index, 6, QStandardItem(str(taskInstruction["loop"])))
|
||
self.taskDetailModel.setItem(index, 7, QStandardItem(taskInstruction["target_param"]))
|
||
self.taskDetailModel.setItem(index, 8, QStandardItem(""))
|
||
self.taskDetailModel.setItem(index, 9, QStandardItem(str(taskInstruction["level"])))
|
||
self.setUnEditableStyle(self.taskDetailModel.item(index, 2))
|
||
self.setUnEditableStyle(self.taskDetailModel.item(index, 3))
|
||
self.setUnEditableStyle(self.taskDetailModel.item(index, 4))
|
||
|
||
devModel = devModelManager.getInfo(dev_model_id)
|
||
if devModel != None:
|
||
attr_json_dict = devModel["attr"]
|
||
devmodelInterfacetypes = {}
|
||
if "interfaceType" in attr_json_dict:
|
||
devmodelInterfacetypes = attr_json_dict["interfaceType"]
|
||
devInterfaceIndex = 0
|
||
if len(devmodelInterfacetypes) > 0:
|
||
cmb = NoWheelComboBox()
|
||
cmb.currentIndexChanged.connect(self.changeInterface)
|
||
for devmodelInterfacetype in devmodelInterfacetypes:
|
||
cmb.addItem(devmodelInterfacetype)
|
||
interface_index = "interface_index" in taskInstruction and taskInstruction["interface_index"] or 0
|
||
interface_index = interface_index != "" and interface_index or 0
|
||
if interface_index < len(devmodelInterfacetypes):
|
||
cmb.setCurrentIndex(interface_index)
|
||
self.taskDetailModel.setItem(index, 8, QStandardItem(str(interface_index)))
|
||
|
||
else:
|
||
cmb.setCurrentIndex(0)
|
||
self.taskDetailModel.setItem(index, 8, QStandardItem("0"))
|
||
self.ui.tableViewTaskDetail.setIndexWidget(self.taskDetailModel.index(index, 8), cmb)
|
||
self.batchOperation()
|
||
self.loadPageTaskDetail(taskId, row + limit,limit)
|
||
else:
|
||
return
|
||
# 切换接口
|
||
def changeInterface(self,index):
|
||
if len(self.ui.tableViewTaskDetail.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewTaskDetail.selectionModel().selectedRows()[0].row()
|
||
self.taskDetailModel.setItem(row, 8 , QStandardItem(str(index)))
|
||
def addProject(self):
|
||
self.createProjectDlg.showCreateDlg()
|
||
|
||
def setCurrentProject(self):
|
||
indexs = self.ui.tableViewPro.selectionModel().selectedRows()
|
||
if len(indexs) == 1:
|
||
row = indexs[0].row()
|
||
row_proId_index = self.projectModel.index(row, 0, QModelIndex())
|
||
proId = self.projectModel.data(row_proId_index)
|
||
projectManager.setCurrentProId(proId)
|
||
|
||
def handleFileOpen(self):
|
||
indexs = self.ui.treeViewFile.selectionModel().selectedRows()
|
||
if len(indexs) == 1:
|
||
os.startfile(self.fileSystemModel.filePath(indexs[0]))
|
||
|
||
def findTopLevelParent(self, widget):
|
||
parent = widget.parent()
|
||
while parent and not isinstance(parent, QMainWindow): # 顶层窗口是 QMainWindow
|
||
widget = parent
|
||
parent = widget.parent()
|
||
return parent
|
||
|
||
# 删除工程
|
||
def deleteProject(self):
|
||
indexs = self.ui.tableViewPro.selectionModel().selectedRows()
|
||
if len(indexs) == 1:
|
||
row = indexs[0].row()
|
||
row_proName_index = self.projectModel.index(row, 1, QModelIndex())
|
||
projectName = self.projectModel.data(row_proName_index, Qt.ItemDataRole.DisplayRole)
|
||
if self.topWidget == None:
|
||
self.topWidget = self.findTopLevelParent(self)
|
||
if self.topWidget != None:
|
||
if self.topWidget.removeDialog(projectName,"删除工程"):
|
||
row_index = self.projectModel.index(row, 0, QModelIndex())
|
||
proId = self.projectModel.data(row_index, Qt.ItemDataRole.DisplayRole)
|
||
if projectManager.delete(proId):
|
||
self.clearChildrenModel()
|
||
self.loadProject()
|
||
else:
|
||
QMessageBox.warning(self, '警告', '请选择要删除的工程', QMessageBox.StandardButton.Ok)
|
||
return
|
||
|
||
def clearChildrenModel(self):
|
||
self.propertyModel.clear()
|
||
self.proHistoryModel.clear()
|
||
|
||
def editProject(self):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewPro.selectionModel().selectedRows()[0].row()
|
||
row_proId_index = self.projectModel.index(row, 0, QModelIndex())
|
||
pro_id = self.projectModel.data(row_proId_index, Qt.ItemDataRole.DisplayRole)
|
||
if pro_id != "":
|
||
self.createProjectDlg.showEditDlg(pro_id)
|
||
|
||
def refreshSelect(self):
|
||
selectedRows = self.ui.tableViewPro.selectionModel().selectedRows()
|
||
self.loadProject()
|
||
if len(selectedRows) < 1:
|
||
self.ui.tableViewPro.selectRow(0)
|
||
if len(selectedRows)== 1:
|
||
row = selectedRows[0].row()
|
||
self.ui.tableViewPro.selectRow(row)
|
||
|
||
def handleProSelection(self):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewPro.selectionModel().selectedRows()[0].row()
|
||
row_index = self.projectModel.index(row, 0, QModelIndex())
|
||
proId = self.projectModel.data(row_index, Qt.ItemDataRole.DisplayRole)
|
||
self.loadProProperty(proId)
|
||
self.batchOperation()
|
||
|
||
# 添加属性
|
||
def addProperty(self):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
# 列表末位名称为空不允许创建
|
||
if self.propertyModel.rowCount() > 0:
|
||
lastIndex = self.propertyModel.rowCount() - 1
|
||
if self.propertyModel.item(lastIndex, 0).text().strip() == "":
|
||
self.ui.tableViewProperty.edit(self.propertyModel.index(lastIndex, 0))
|
||
return
|
||
index = self.propertyModel.rowCount()
|
||
self.propertyModel.appendRow([QStandardItem(), QStandardItem()])
|
||
self.ui.tableViewProperty.setCurrentIndex(self.propertyModel.index(index, 0) )
|
||
self.ui.tableViewProperty.edit(self.propertyModel.index(index, 0))
|
||
|
||
# 删除属性
|
||
def deleteProperty(self):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
if len(self.ui.tableViewProperty.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewProperty.selectionModel().selectedRows()[0].row()
|
||
#不删固定属性
|
||
if row > 3:
|
||
self.propertyModel.removeRow(row)
|
||
self.updateProperty()
|
||
|
||
# 更新属性
|
||
def updateProperty(self):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewPro.selectionModel().selectedRows()[0].row()
|
||
row_id_index = self.projectModel.index(row, 0, QModelIndex())
|
||
proId = self.projectModel.data(row_id_index, Qt.ItemDataRole.DisplayRole)
|
||
#固定属性
|
||
remark = ""
|
||
operator = ""
|
||
createDate = ""
|
||
lastDate = ""
|
||
attr = {}
|
||
for row in range(self.propertyModel.rowCount()):
|
||
if self.propertyModel.item(row, 0).text() == "备注":
|
||
remark = self.propertyModel.item(row, 1).text()
|
||
elif self.propertyModel.item(row, 0).text() == "人员":
|
||
operator = self.propertyModel.item(row, 1).text()
|
||
elif self.propertyModel.item(row, 0).text() == "创建时间":
|
||
createDate = self.propertyModel.item(row, 1).text()
|
||
elif self.propertyModel.item(row, 0).text() == "上次打开时间":
|
||
lastDate = self.propertyModel.item(row, 1).text()
|
||
else:
|
||
attr[self.propertyModel.item(row, 0).text()] = self.propertyModel.item(row, 1).text()
|
||
param = {
|
||
"remark":remark,
|
||
"operator":operator,
|
||
"createDate":createDate,
|
||
"lastDate":lastDate,
|
||
"attr": attr
|
||
}
|
||
projectManager.update(proId, param)
|
||
|
||
#设置不可编辑样式
|
||
def setUnEditableStyle(self, item):
|
||
item.setEditable(False)
|
||
# item.setForeground(QColor(79, 79, 79))
|
||
# item.setBackground(QColor(240, 240, 240))
|
||
|
||
def loadProProperty(self, proId):
|
||
info = projectManager.getInfo(proId)
|
||
taskInfo = projectManager.getProTask(proId)
|
||
self.targetIds = "task" in taskInfo and taskInfo["task"] or []
|
||
self.targetParams = "params" in taskInfo and taskInfo["params"] or {}
|
||
self.loadTask()
|
||
info["history"] = projectManager.getCmdHistory(proId)
|
||
if info == None:
|
||
self.loadProject()
|
||
self.clearChildrenModel()
|
||
else:
|
||
#阻塞propertyModel信号
|
||
self.propertyModel.blockSignals(True)
|
||
self.propertyModel.clear()
|
||
self.propertyModel.setItem(0, 0, QStandardItem("备注"))
|
||
self.propertyModel.setItem(0, 1, QStandardItem(info["remark"]))
|
||
self.propertyModel.setItem(1, 0, QStandardItem("人员"))
|
||
self.propertyModel.setItem(1, 1, QStandardItem(info["operator"]))
|
||
self.propertyModel.setItem(2, 0, QStandardItem("创建时间"))
|
||
self.propertyModel.setItem(2, 1, QStandardItem(info["createDate"]))
|
||
self.propertyModel.setItem(3, 0, QStandardItem("上次打开时间"))
|
||
self.propertyModel.setItem(3, 1, QStandardItem(info["lastDate"]))
|
||
self.setUnEditableStyle(self.propertyModel.item(0, 0))
|
||
self.setUnEditableStyle(self.propertyModel.item(1, 0))
|
||
self.setUnEditableStyle(self.propertyModel.item(2, 0))
|
||
self.setUnEditableStyle(self.propertyModel.item(3, 0))
|
||
self.setUnEditableStyle(self.propertyModel.item(1, 1))
|
||
self.setUnEditableStyle(self.propertyModel.item(2, 1))
|
||
self.setUnEditableStyle(self.propertyModel.item(3, 1))
|
||
for key in info["attr"]:
|
||
self.propertyModel.appendRow([QStandardItem(key), QStandardItem(info["attr"][key])])
|
||
#解除propertyModel阻塞
|
||
self.propertyModel.blockSignals(False)
|
||
|
||
self.proHistoryModel.blockSignals(True)
|
||
self.proHistoryModel.clear()
|
||
for history in info["history"]:
|
||
strtype = ProjectForm.TASK == history["type"] and "任务" or (ProjectForm.VIRTUAL == history["type"] and "虚拟任务" or "指令")
|
||
self.proHistoryModel.appendRow([
|
||
QStandardItem(history["time"]),
|
||
QStandardItem(strtype),
|
||
QStandardItem(history["name"]),
|
||
QStandardItem(history["event"]),
|
||
QStandardItem(history["operator"])])
|
||
self.proHistoryModel.blockSignals(False)
|
||
self.ui.treeViewFile.setRootIndex(self.fileSystemModel.index(os.path.join(self.basePath, proId)))
|
||
|
||
def setSplitterStretchFactor(self):
|
||
self.ui.splitter.setStretchFactor(0, 30)
|
||
self.ui.splitter.setStretchFactor(1, 30)
|
||
self.ui.splitter.setStretchFactor(2, 40)
|
||
|
||
# def resizeEvent(self, event):
|
||
# # 在窗口或控件的尺寸发生变化时调用
|
||
# super().resizeEvent(event)
|
||
|
||
# 批量操作
|
||
def batchOperation(self):
|
||
self.setAllHorizontalHeaderLabels()
|
||
self.hideAllColumn()
|
||
self.setAllColumnWidth()
|
||
|
||
# 批量设置列宽
|
||
def setAllColumnWidth(self):
|
||
try:
|
||
if self.projectModel.rowCount() > 0:
|
||
self.ui.tableViewPro.horizontalHeader().setSectionResizeMode(1,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewPro.horizontalHeader().setSectionResizeMode(2,QHeaderView.ResizeMode.Interactive)
|
||
|
||
if self.propertyModel.rowCount() > 0:
|
||
self.ui.tableViewProperty.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewProperty.horizontalHeader().setSectionResizeMode(1, QHeaderView.ResizeMode.Stretch)
|
||
|
||
if self.proHistoryModel.rowCount() > 0:
|
||
self.ui.tableViewHistory.horizontalHeader().setSectionResizeMode(0,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewHistory.horizontalHeader().setSectionResizeMode(1,QHeaderView.ResizeMode.ResizeToContents)
|
||
self.ui.tableViewHistory.horizontalHeader().setSectionResizeMode(2,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewHistory.horizontalHeader().setSectionResizeMode(3,QHeaderView.ResizeMode.ResizeToContents)
|
||
self.ui.tableViewHistory.horizontalHeader().setSectionResizeMode(4,QHeaderView.ResizeMode.ResizeToContents)
|
||
|
||
if self.taskModel.rowCount() > 0:
|
||
self.ui.tableViewTask.horizontalHeader().setSectionResizeMode(1,QHeaderView.ResizeMode.ResizeToContents)
|
||
self.ui.tableViewTask.horizontalHeader().setSectionResizeMode(2,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTask.horizontalHeader().setSectionResizeMode(3,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTask.horizontalHeader().setSectionResizeMode(4,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTask.horizontalHeader().setSectionResizeMode(5,QHeaderView.ResizeMode.Interactive)
|
||
|
||
if self.taskDetailModel.rowCount() > 0:
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(2,QHeaderView.ResizeMode.ResizeToContents)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(3,QHeaderView.ResizeMode.ResizeToContents)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(4,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(5,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(6,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(7,QHeaderView.ResizeMode.Stretch)
|
||
self.ui.tableViewTaskDetail.horizontalHeader().setSectionResizeMode(8,QHeaderView.ResizeMode.Interactive)
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
def save_column_widths(self):
|
||
settings = QSettings("TG", "projectFrom")
|
||
tables = [
|
||
self.ui.tableViewPro,
|
||
self.ui.tableViewProperty,
|
||
self.ui.tableViewHistory,
|
||
self.ui.tableViewTask,
|
||
self.ui.tableViewTaskDetail
|
||
]
|
||
|
||
for i, table in enumerate(tables):
|
||
for column in range(table.model().columnCount()):
|
||
width = table.columnWidth(column)
|
||
print(f"SAVE table_{i}_column_{column}_width: {width}")
|
||
settings.setValue(f"table_{i}_column_{column}_width", width)
|
||
|
||
def load_column_widths(self):
|
||
settings = QSettings("TG", "projectFrom")
|
||
tables = [
|
||
self.ui.tableViewPro,
|
||
self.ui.tableViewProperty,
|
||
self.ui.tableViewHistory,
|
||
self.ui.tableViewTask,
|
||
self.ui.tableViewTaskDetail
|
||
]
|
||
|
||
for i, table in enumerate(tables):
|
||
for column in range(table.model().columnCount()):
|
||
|
||
width = settings.value(f"table_{i}_column_{column}_width", type=int)
|
||
print(f"LOAD table_{i}_column_{column}_width:", width)
|
||
if width is not None:
|
||
print(f"LOAD table_{i}_column_{column}_width:", width)
|
||
table.setColumnWidth(column, width)
|
||
|
||
# 批量设置标题
|
||
def setAllHorizontalHeaderLabels(self):
|
||
self.projectModel.setHorizontalHeaderLabels(["ID"," 名称"," 活动"])
|
||
self.propertyModel.setHorizontalHeaderLabels([" 属性", " 值"])
|
||
self.proHistoryModel.setHorizontalHeaderLabels([" 时间", " 类型", " 名称", " 事件", " 人员"])
|
||
self.taskModel.setHorizontalHeaderLabels(["ID"," 名称", " 循环次数", " 间隔时间(ms)", " 参数", "进度","组ID"])
|
||
self.taskDetailModel.setHorizontalHeaderLabels(["ID","任务ID"," 设备"," 名称", " 类型", " 间隔时间(ms)", " 循环次数", " 参数", "端口", "序号"])
|
||
# 隐藏指定列
|
||
def hideAllColumn(self):
|
||
self.ui.tableViewPro.setColumnHidden(0, True)
|
||
self.ui.tableViewPro.setColumnHidden(3, True)
|
||
self.ui.tableViewTask.setColumnHidden(0, True)
|
||
self.ui.tableViewTask.setColumnHidden(6, True)
|
||
self.ui.tableViewTaskDetail.setColumnHidden(0, True)
|
||
self.ui.tableViewTaskDetail.setColumnHidden(1, True)
|
||
self.ui.tableViewTaskDetail.setColumnHidden(9, True)
|
||
|
||
def proCheckedChanged(self):
|
||
checked = self.sender().isChecked()
|
||
if checked:
|
||
index = self.ui.tableViewPro.currentIndex()
|
||
if index.isValid():
|
||
proId = ""
|
||
self.currentProjectIndex = index.row()
|
||
self.settings.setValue("currentProjectIndex", self.currentProjectIndex)
|
||
proId = self.projectModel.item(index.row(), 0).text()
|
||
proName = self.projectModel.item(index.row(), 1).text()
|
||
projectManager.setCurrentProId(proId)
|
||
influxdb.create(proName)
|
||
|
||
def editProFinished(self, proname):
|
||
if len(self.ui.tableViewPro.selectionModel().selectedRows())== 1:
|
||
row = self.ui.tableViewPro.selectionModel().selectedRows()[0].row()
|
||
proId = self.projectModel.item(row, 0).text()
|
||
self.projectModel.setItem(row, 1, QStandardItem(proname))
|
||
projectManager.updateProId(proId)
|
||
if self.projectModel.rowCount() > 0:
|
||
self.ui.tableViewPro.selectRow(row)
|
||
|
||
def createProFinished(self):
|
||
self.loadProject()
|
||
if self.projectModel.rowCount() > 0:
|
||
self.ui.tableViewPro.selectRow(self.projectModel.rowCount()-1)
|
||
|
||
def loadProject(self):
|
||
try:
|
||
self.projectModel.clear()
|
||
infos = projectManager.getInfo("all")
|
||
if len(infos) == 0:
|
||
projectManager.setCurrentProId(common.emputyProId)
|
||
return
|
||
for info in infos:
|
||
index = self.projectModel.rowCount()
|
||
self.projectModel.setItem(index, 0, QStandardItem(infos[info]["id"]))
|
||
self.projectModel.setItem(index, 1, QStandardItem(infos[info]["name"]))
|
||
radio = QRadioButton()
|
||
radio.setStyleSheet("QRadioButton::indicator {"
|
||
" subcontrol-origin: padding;"
|
||
" subcontrol-position: center;"
|
||
"}")
|
||
radio.toggled.connect(self.proCheckedChanged)
|
||
self.projectModel.setItem(index, 2 , QStandardItem())
|
||
if index == self.currentProjectIndex:
|
||
self.ui.tableViewPro.selectRow(index)
|
||
radio.setChecked(True)
|
||
projectManager.setCurrentProId(infos[info]["id"])
|
||
else:
|
||
radio.setChecked(False)
|
||
self.ui.tableViewPro.setIndexWidget(self.projectModel.index(index, 2), radio)
|
||
|
||
self.batchOperation()
|
||
except Exception as e:
|
||
log.error(e) |