# -*- coding: utf-8 -*- """ Copyright (c) 2025 KenanZhu. All rights reserved. This software is provided "as is", without any warranty of any kind. You may use, modify, and distribute this file under the terms of the MIT License. See the LICENSE file for details. """ import uuid from enum import Enum from datetime import datetime, timedelta from PySide6.QtCore import ( Slot, QDateTime, QUrl ) from PySide6.QtGui import QDesktopServices from PySide6.QtWidgets import ( QLabel, QDialog, QWidget, QSpinBox, QHBoxLayout, QVBoxLayout, QDateTimeEdit, QGroupBox, QPushButton ) from gui.resources.ui.Ui_ALTimerTaskAddDialog import Ui_ALTimerTaskAddDialog from utils.TimerUtils import TimerUtils class ALTimerTaskStatus(Enum): PENDING = "等待中" READY = "已就绪" RUNNING = "执行中" EXECUTED = "已执行" ERROR = "执行失败" OUTDATED = "已过期" UNKNOWN = "未知" class ALTimerTaskAddDialog(QDialog, Ui_ALTimerTaskAddDialog): def __init__( self, parent = None, timer_task: dict = None ): super().__init__(parent) self.__edit_timer_task = timer_task self.setupUi(self) self.modifyUi() self.connectSignals() if self.__edit_timer_task: self.loadTask(self.__edit_timer_task) def modifyUi( self ): self.TimerTypeComboBox.setCurrentIndex(0) self.SpecificTimerWidget = QWidget() self.SpecificTimerLayout = QHBoxLayout(self.SpecificTimerWidget) self.SpecificTimerLayout.setContentsMargins(0, 0, 0, 0) self.SpecificTimerLayout.setSpacing(5) self.SpecificTimerLayout.addWidget(QLabel("定时时间：")) self.SpecificDateTimeEdit = QDateTimeEdit() self.SpecificDateTimeEdit.setCalendarPopup(True) self.SpecificDateTimeEdit.setDisplayFormat("yyyy-MM-dd HH:mm:ss") self.SpecificDateTimeEdit.setMinimumDateTime(QDateTime.currentDateTime()) self.SpecificDateTimeEdit.setDateTime(QDateTime.currentDateTime().addSecs(60)) self.SpecificTimerLayout.addWidget(self.SpecificDateTimeEdit) self.TimerConfigLayout.addWidget(self.SpecificTimerWidget) self.RelativeTimerWidget = QWidget() self.RelativeTimerLayout = QHBoxLayout(self.RelativeTimerWidget) self.RelativeTimerLayout.setContentsMargins(0, 0, 0, 0) self.RelativeTimerLayout.setSpacing(5) self.RelativeTimerLayout.addWidget(QLabel("相对时间：")) self.RelativeDaySpinBox = QSpinBox() self.RelativeDaySpinBox.setMinimum(0) self.RelativeDaySpinBox.setMaximum(364) self.RelativeDaySpinBox.setSuffix("天") self.RelativeTimerLayout.addWidget(self.RelativeDaySpinBox) self.RelativeHourSpinBox = QSpinBox() self.RelativeHourSpinBox.setMinimum(0) self.RelativeHourSpinBox.setMaximum(23) self.RelativeHourSpinBox.setSuffix("时") self.RelativeTimerLayout.addWidget(self.RelativeHourSpinBox) self.RelativeMinuteSpinBox = QSpinBox() self.RelativeMinuteSpinBox.setMinimum(0) self.RelativeMinuteSpinBox.setMaximum(59) self.RelativeMinuteSpinBox.setSuffix("分") self.RelativeTimerLayout.addWidget(self.RelativeMinuteSpinBox) self.RelativeSecondSpinBox = QSpinBox() self.RelativeSecondSpinBox.setMinimum(0) self.RelativeSecondSpinBox.setMaximum(59) self.RelativeSecondSpinBox.setSuffix("秒") self.RelativeTimerLayout.addWidget(self.RelativeSecondSpinBox) self.TimerConfigLayout.addWidget(self.RelativeTimerWidget) self.RelativeTimerWidget.setVisible(False) self.AutoScriptGroupBox = QGroupBox("AutoScript 指令") self.AutoScriptLayout = QVBoxLayout(self.AutoScriptGroupBox) self.AutoScriptLayout.setContentsMargins(3, 3, 3, 3) self.AutoScriptLayout.setSpacing(3) AutoScriptBtnLayout = QHBoxLayout() self.AutoScriptEditButton = QPushButton("编辑") self.AutoScriptEditButton.setMinimumHeight(25) self.AutoScriptEditButton.setFixedWidth(80) AutoScriptBtnLayout.addWidget(self.AutoScriptEditButton) AutoScriptBtnLayout.addStretch() self.AutoScriptHelpButton = QPushButton("?") self.AutoScriptHelpButton.setFixedSize(20, 20) self.AutoScriptHelpButton.setToolTip( "AutoScript 是一种轻量级 DSL 语言，基于 Lua 实现。\n" "用于在重复定时任务执行前，对用户的预约数据进行预处理\n" "\n" "点击查看完整在线文档" ) self.AutoScriptHelpButton.setStyleSheet( "QPushButton { border-radius: 10px; border: 1px solid #999; " "font-weight: bold; color: #555; }" "QPushButton:hover { background-color: #E0E0E0; }" ) AutoScriptBtnLayout.addWidget(self.AutoScriptHelpButton) self.AutoScriptStatusLabel = QLabel("未设置") self.AutoScriptStatusLabel.setStyleSheet("color: #969696;") self.AutoScriptStatusLabel.setFixedHeight(25) AutoScriptBtnLayout.addWidget(self.AutoScriptStatusLabel) self.AutoScriptLayout.addLayout(AutoScriptBtnLayout) self.ALAddTimerTaskLayout.insertWidget( self.ALAddTimerTaskLayout.indexOf(self.TaskConfigGroupBox) + 1, self.AutoScriptGroupBox ) self.AutoScriptGroupBox.setVisible(False) self.__auto_script = "" self.__mock_target_data = None def loadTask( self, task: dict ): self.TaskNameLineEdit.setText(task.get("name", "")) time_type = task.get("time_type", "特定时间") self.TimerTypeComboBox.setCurrentText(time_type) self.SpecificDateTimeEdit.setDateTime( QDateTime(task["execute_time"]) ) self.RelativeDaySpinBox.setValue(0) self.RelativeHourSpinBox.setValue(0) self.RelativeMinuteSpinBox.setValue(0) self.RelativeSecondSpinBox.setValue(0) if task.get("silent", False): self.SilentlyRunRadioButton.setChecked(True) else: self.ShowBeforeRunRadioButton.setChecked(True) repeat = task.get("repeat", False) self.RepeatCheckBox.setChecked(repeat) if repeat: repeat_days = task.get("repeat_days", []) self.MonCheckBox.setChecked(0 in repeat_days) self.TueCheckBox.setChecked(1 in repeat_days) self.WedCheckBox.setChecked(2 in repeat_days) self.ThuCheckBox.setChecked(3 in repeat_days) self.FriCheckBox.setChecked(4 in repeat_days) self.SatCheckBox.setChecked(5 in repeat_days) self.SunCheckBox.setChecked(6 in repeat_days) auto_script = task.get("repeat_auto_script", "") if auto_script: self.__auto_script = auto_script self.AutoScriptStatusLabel.setText("已设置") self.AutoScriptStatusLabel.setStyleSheet("color: #4CAF50;") mock_data = task.get("mock_target_data") if mock_data: self.__mock_target_data = mock_data self.ConfirmButton.setText("保存") def connectSignals( self ): self.CancelButton.clicked.connect(self.reject) self.ConfirmButton.clicked.connect(self.accept) self.TimerTypeComboBox.currentIndexChanged.connect(self.onTimerTypeComboBoxIndexChanged) self.RepeatCheckBox.toggled.connect(self.onRepeatCheckBoxToggled) self.AutoScriptEditButton.clicked.connect(self.onPreviewAutoScript) self.AutoScriptHelpButton.clicked.connect(self.onAutoScriptHelp) def getTimerTask( self ) -> dict: added_time = datetime.now() if not self.TaskNameLineEdit.text(): name = f"未命名任务-{added_time.strftime("%Y%m%d%H%M%S")}" else: name = self.TaskNameLineEdit.text() timer_type_index = self.TimerTypeComboBox.currentIndex() silent = not self.ShowBeforeRunRadioButton.isChecked() if timer_type_index == 0: execute_time = self.SpecificDateTimeEdit.dateTime() tmp_time_str = execute_time.toString("yyyy-MM-dd HH:mm:ss") execute_time = datetime.strptime(tmp_time_str, "%Y-%m-%d %H:%M:%S") else: execute_time = datetime.now() + timedelta( days = self.RelativeDaySpinBox.value(), hours = self.RelativeHourSpinBox.value(), minutes = self.RelativeMinuteSpinBox.value(), seconds = self.RelativeSecondSpinBox.value() ) if self.__edit_timer_task: task_data = dict(self.__edit_timer_task) task_data["name"] = name task_data["execute_time"] = execute_time task_data["silent"] = silent task_data["status"] = ALTimerTaskStatus.PENDING task_data["executed"] = False task_data["repeat_auto_script"] = self.__auto_script task_data["mock_target_data"] = self.__mock_target_data else: task_data = { "name": name, "uuid": uuid.uuid4().hex.upper() + f"-{added_time.strftime("%Y%m%d%H%M%S")}", "time_type": self.TimerTypeComboBox.currentText(), "execute_time": execute_time, "silent": silent, "added_time": added_time, "status": ALTimerTaskStatus.PENDING, "executed": False, "repeat": self.RepeatCheckBox.isChecked(), "repeat_auto_script": self.__auto_script, "mock_target_data": self.__mock_target_data, } repeat = self.RepeatCheckBox.isChecked() task_data["repeat"] = repeat if repeat: if "repeat_history" not in task_data: task_data["repeat_history"] = [] repeat_days = [] if self.MonCheckBox.isChecked(): repeat_days.append(0) if self.TueCheckBox.isChecked(): repeat_days.append(1) if self.WedCheckBox.isChecked(): repeat_days.append(2) if self.ThuCheckBox.isChecked(): repeat_days.append(3) if self.FriCheckBox.isChecked(): repeat_days.append(4) if self.SatCheckBox.isChecked(): repeat_days.append(5) if self.SunCheckBox.isChecked(): repeat_days.append(6) if not repeat_days: repeat_days = [0, 1, 2, 3, 4, 5, 6] task_data["repeat_days"] = repeat_days task_data["repeat_hour"] = execute_time.hour task_data["repeat_minute"] = execute_time.minute task_data["repeat_second"] = execute_time.second task_data["execute_time"] = TimerUtils.getNextTimerRepeatTime( task_data["repeat_days"], task_data["repeat_hour"], task_data["repeat_minute"], task_data["repeat_second"] ) return task_data @Slot(int) def onTimerTypeComboBoxIndexChanged( self, index: int ): self.SpecificTimerWidget.setVisible(index == 0) self.RelativeTimerWidget.setVisible(index == 1) @Slot(bool) def onRepeatCheckBoxToggled( self, checked: bool ): self.MonCheckBox.setEnabled(checked) self.TueCheckBox.setEnabled(checked) self.WedCheckBox.setEnabled(checked) self.ThuCheckBox.setEnabled(checked) self.FriCheckBox.setEnabled(checked) self.SatCheckBox.setEnabled(checked) self.SunCheckBox.setEnabled(checked) self.AutoScriptGroupBox.setVisible(checked) @Slot() def onPreviewAutoScript(self): from gui.ALAutoScriptEditDialog import ALAutoScriptEditDialog Dlg = ALAutoScriptEditDialog(self, self.__auto_script, self.__mock_target_data) if Dlg.exec() == QDialog.DialogCode.Accepted: script = Dlg.getScript() self.__auto_script = script self.__mock_target_data = Dlg.getMockData() if script: self.AutoScriptStatusLabel.setText("已设置") self.AutoScriptStatusLabel.setStyleSheet("color: #4CAF50;") else: self.AutoScriptStatusLabel.setText("未设置") self.AutoScriptStatusLabel.setStyleSheet("color: #969696;") Dlg.deleteLater() @Slot() def onAutoScriptHelp( self ): QDesktopServices.openUrl( QUrl("https://manuals.autolibrary.kenanzhu.com/zh/autoscript") )