87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
|
import os
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
import base64
|
|||
|
|
from PyQt6 import *
|
|||
|
|
from PyQt6.QtCore import *
|
|||
|
|
from PyQt6.QtGui import *
|
|||
|
|
from PyQt6.QtWidgets import *
|
|||
|
|
from ui.Ui_createTaskDlg import Ui_createTaskDlg
|
|||
|
|
from taskModel.taskManager import taskManager
|
|||
|
|
from show_editor import CodeDialog
|
|||
|
|
import ui.res_rc
|
|||
|
|
|
|||
|
|
class CreateTaskDlg(QDialog):
|
|||
|
|
createFinished = pyqtSignal()
|
|||
|
|
editFinished = pyqtSignal()
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self.ui = Ui_createTaskDlg()
|
|||
|
|
self.ui.setupUi(self)
|
|||
|
|
self.taskId = ""
|
|||
|
|
self.taskGroupId = ""
|
|||
|
|
with open(os.path.join(os.path.dirname(__file__),"default_task.py"), "r", encoding="utf-8") as f:
|
|||
|
|
self.defaultScriptBase64 = f.read()
|
|||
|
|
# 删除UTF-8 BOM（如果有的话）
|
|||
|
|
# UTF-8 BOM的字节序列是\xef\xbb\xbf，解码为字符串后是'\ufeff'
|
|||
|
|
if self.defaultScriptBase64.startswith('\ufeff'):
|
|||
|
|
self.defaultScriptBase64 = self.defaultScriptBase64[1:] # 切片以删除第一个字符（即BOM）
|
|||
|
|
self.scriptStr = self.defaultScriptBase64
|
|||
|
|
self.codeDialog = CodeDialog("任务脚本编辑")
|
|||
|
|
self.ui.pbScript.clicked.connect(self.handleScript)
|
|||
|
|
self.ui.pbSave.clicked.connect(self.handleAccept)
|
|||
|
|
self.ui.pbCancel.clicked.connect(self.reject)
|
|||
|
|
|
|||
|
|
def showCreateDlg(self, group_id):
|
|||
|
|
self.setWindowTitle("创建任务")
|
|||
|
|
self.taskId = ""
|
|||
|
|
self.taskGroupId = group_id
|
|||
|
|
self.scriptStr = self.defaultScriptBase64
|
|||
|
|
self.ui.leName.setText("")
|
|||
|
|
self.ui.sLoop.setValue(1)
|
|||
|
|
self.ui.sDelay.setValue(0)
|
|||
|
|
self.ui.leRemark.setText("")
|
|||
|
|
self.exec()
|
|||
|
|
|
|||
|
|
def showEditDlg(self, task):
|
|||
|
|
self.setWindowTitle("编辑任务")
|
|||
|
|
self.taskId = task["id"]
|
|||
|
|
self.ui.leName.setText(task["name"])
|
|||
|
|
self.ui.sLoop.setValue(int(task["loop"]))
|
|||
|
|
self.ui.sDelay.setValue(int(task["delay"]))
|
|||
|
|
self.ui.leRemark.setText(task["remark"])
|
|||
|
|
self.scriptStr = task["script"] and str(base64.b64decode(task["script"] ), 'utf-8') or self.defaultScriptBase64
|
|||
|
|
self.exec()
|
|||
|
|
|
|||
|
|
def handleScript(self):
|
|||
|
|
self.scriptStr = self.codeDialog.showEditor(self.scriptStr)
|
|||
|
|
|
|||
|
|
def handleAccept(self):
|
|||
|
|
name = self.ui.leName.text()
|
|||
|
|
loop = str(self.ui.sLoop.value())
|
|||
|
|
delay = str(self.ui.sDelay.value())
|
|||
|
|
remark = self.ui.leRemark.text()
|
|||
|
|
script = base64.b64encode(self.scriptStr.encode("utf-8"))
|
|||
|
|
if name == "":
|
|||
|
|
QMessageBox.warning(self, "错误", "任务名称不能为空", QMessageBox.StandardButton.Ok)
|
|||
|
|
return
|
|||
|
|
if self.windowTitle() == "创建任务":
|
|||
|
|
id = taskManager.create(self.taskGroupId, name, loop, delay, remark, script)
|
|||
|
|
if id != "":
|
|||
|
|
self.createFinished.emit()
|
|||
|
|
elif self.windowTitle() == "编辑任务":
|
|||
|
|
param = {
|
|||
|
|
"name": name,
|
|||
|
|
"loop": loop,
|
|||
|
|
"delay": delay,
|
|||
|
|
"remark": remark,
|
|||
|
|
"script": script
|
|||
|
|
}
|
|||
|
|
result = taskManager.update(self.taskId, param)
|
|||
|
|
if result:
|
|||
|
|
self.editFinished.emit()
|
|||
|
|
self.accept()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|