# # Copyright © <2024> # This file is part of AUTO_MAA. # AUTO_MAA is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published # by the Free Software Foundation, either version 3 of the License, # or (at your option) any later version. # AUTO_MAA is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty # of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See # the GNU General Public License for more details. # You should have received a copy of the GNU General Public License # along with AUTO_MAA. If not, see . # DLmaster_361@163.com """ AUTO_MAA AUTO_MAA调度队列界面 v4.2 作者：DLmaster_361 """ from loguru import logger from PySide6.QtWidgets import ( QWidget, QVBoxLayout, QStackedWidget, QHBoxLayout, ) from qfluentwidgets import ( Action, qconfig, Pivot, ScrollArea, FluentIcon, MessageBox, HeaderCardWidget, TextBrowser, CommandBar, SwitchSettingCard, ComboBoxSettingCard, ) from PySide6.QtCore import Qt from typing import List import json import shutil from app.core import Config, MainInfoBar from .Widget import ( LineEditSettingCard, TimeEditSettingCard, NoOptionComboBoxSettingCard, ) class QueueManager(QWidget): def __init__(self, parent=None): super().__init__(parent) self.setObjectName("调度队列") layout = QVBoxLayout(self) self.tools = CommandBar() self.queue_manager = QueueSettingBox(self) # 逐个添加动作 self.tools.addActions( [ Action( FluentIcon.ADD_TO, "新建调度队列", triggered=self.add_setting_box ), Action( FluentIcon.REMOVE_FROM, "删除调度队列", triggered=self.del_setting_box, ), ] ) self.tools.addSeparator() self.tools.addActions( [ Action( FluentIcon.LEFT_ARROW, "向左移动", triggered=self.left_setting_box ), Action( FluentIcon.RIGHT_ARROW, "向右移动", triggered=self.right_setting_box ), ] ) layout.addWidget(self.tools) layout.addWidget(self.queue_manager) def add_setting_box(self): """添加一个调度队列""" index = len(self.queue_manager.search_queue()) + 1 qconfig.load( Config.app_path / f"config/QueueConfig/调度队列_{index}.json", Config.queue_config, ) Config.clear_queue_config() Config.queue_config.save() self.queue_manager.add_QueueSettingBox(index) self.queue_manager.switch_SettingBox(index) def del_setting_box(self): """删除一个调度队列实例""" name = self.queue_manager.pivot.currentRouteKey() if name == None: logger.warning("未选择调度队列") MainInfoBar.push_info_bar( "warning", "未选择调度队列", "请先选择一个调度队列", 5000 ) return None if name in Config.running_list: logger.warning("调度队列正在运行") MainInfoBar.push_info_bar( "warning", "调度队列正在运行", "请先停止调度队列", 5000 ) return None choice = MessageBox( "确认", f"确定要删除 {name} 吗？", self.window(), ) if choice.exec(): queue_list = self.queue_manager.search_queue() move_list = [_ for _ in queue_list if int(_[0][5:]) > int(name[5:])] index = max(int(name[5:]) - 1, 1) self.queue_manager.clear_SettingBox() (Config.app_path / f"config/QueueConfig/{name}.json").unlink() for queue in move_list: if (Config.app_path / f"config/QueueConfig/{queue[0]}.json").exists(): (Config.app_path / f"config/QueueConfig/{queue[0]}.json").rename( Config.app_path / f"config/QueueConfig/调度队列_{int(queue[0][5:])-1}.json", ) self.queue_manager.show_SettingBox(index) def left_setting_box(self): """向左移动调度队列实例""" name = self.queue_manager.pivot.currentRouteKey() if name == None: logger.warning("未选择调度队列") MainInfoBar.push_info_bar( "warning", "未选择调度队列", "请先选择一个调度队列", 5000 ) return None index = int(name[5:]) if index == 1: logger.warning("向左移动调度队列时已到达最左端") MainInfoBar.push_info_bar( "warning", "已经是第一个调度队列", "无法向左移动", 5000 ) return None if name in Config.running_list or f"调度队列_{index-1}" in Config.running_list: logger.warning("相关调度队列正在运行") MainInfoBar.push_info_bar( "warning", "相关调度队列正在运行", "请先停止调度队列", 5000 ) return None self.queue_manager.clear_SettingBox() (Config.app_path / f"config/QueueConfig/调度队列_{index}.json").rename( Config.app_path / f"config/QueueConfig/调度队列_0.json", ) shutil.move( str(Config.app_path / f"config/QueueConfig/调度队列_{index-1}.json"), str(Config.app_path / f"config/QueueConfig/调度队列_{index}.json"), ) (Config.app_path / f"config/QueueConfig/调度队列_0.json").rename( Config.app_path / f"config/QueueConfig/调度队列_{index-1}.json", ) self.queue_manager.show_SettingBox(index - 1) def right_setting_box(self): """向右移动调度队列实例""" name = self.queue_manager.pivot.currentRouteKey() if name == None: logger.warning("未选择调度队列") MainInfoBar.push_info_bar( "warning", "未选择调度队列", "请先选择一个调度队列", 5000 ) return None queue_list = self.queue_manager.search_queue() index = int(name[5:]) if index == len(queue_list): logger.warning("向右移动调度队列时已到达最右端") MainInfoBar.push_info_bar( "warning", "已经是最后一个调度队列", "无法向右移动", 5000 ) return None if name in Config.running_list or f"调度队列_{index+1}" in Config.running_list: logger.warning("相关调度队列正在运行") MainInfoBar.push_info_bar( "warning", "相关调度队列正在运行", "请先停止调度队列", 5000 ) return None self.queue_manager.clear_SettingBox() (Config.app_path / f"config/QueueConfig/调度队列_{index}.json").rename( Config.app_path / f"config/QueueConfig/调度队列_0.json", ) (Config.app_path / f"config/QueueConfig/调度队列_{index+1}.json").rename( Config.app_path / f"config/QueueConfig/调度队列_{index}.json", ) (Config.app_path / f"config/QueueConfig/调度队列_0.json").rename( Config.app_path / f"config/QueueConfig/调度队列_{index+1}.json", ) self.queue_manager.show_SettingBox(index + 1) def refresh(self): """刷新调度队列界面""" if len(self.queue_manager.search_queue()) == 0: index = 0 else: index = int(self.queue_manager.pivot.currentRouteKey()[5:]) self.queue_manager.clear_SettingBox() self.queue_manager.show_SettingBox(index) class QueueSettingBox(QWidget): def __init__(self, parent=None): super().__init__(parent) self.setObjectName("调度队列管理") self.pivot = Pivot(self) self.stackedWidget = QStackedWidget(self) self.Layout = QVBoxLayout(self) self.script_list: List[QueueMemberSettingBox] = [] self.Layout.addWidget(self.pivot, 0, Qt.AlignHCenter) self.Layout.addWidget(self.stackedWidget) self.Layout.setContentsMargins(0, 0, 0, 0) self.pivot.currentItemChanged.connect( lambda index: self.switch_SettingBox(int(index[5:]), if_change_pivot=False) ) self.show_SettingBox(1) def show_SettingBox(self, index) -> None: """加载所有子界面""" queue_list = self.search_queue() qconfig.load( Config.app_path / "config/临时.json", Config.queue_config, ) Config.clear_queue_config() for queue in queue_list: self.add_QueueSettingBox(int(queue[0][5:])) if (Config.app_path / "config/临时.json").exists(): (Config.app_path / "config/临时.json").unlink() self.switch_SettingBox(index) def switch_SettingBox(self, index: int, if_change_pivot: bool = True) -> None: """切换到指定的子界面""" queue_list = self.search_queue() if len(queue_list) == 0: return None if index > len(queue_list): return None qconfig.load( Config.app_path / f"config/QueueConfig/{self.script_list[index-1].objectName()}.json", Config.queue_config, ) if if_change_pivot: self.pivot.setCurrentItem(self.script_list[index - 1].objectName()) self.stackedWidget.setCurrentWidget(self.script_list[index - 1]) def clear_SettingBox(self) -> None: """清空所有子界面""" for sub_interface in self.script_list: self.stackedWidget.removeWidget(sub_interface) sub_interface.deleteLater() self.script_list.clear() self.pivot.clear() qconfig.load( Config.app_path / "config/临时.json", Config.queue_config, ) Config.clear_queue_config() if (Config.app_path / "config/临时.json").exists(): (Config.app_path / "config/临时.json").unlink() def add_QueueSettingBox(self, uid: int) -> None: """添加一个调度队列设置界面""" maa_setting_box = QueueMemberSettingBox(uid, self) self.script_list.append(maa_setting_box) self.stackedWidget.addWidget(self.script_list[-1]) self.pivot.addItem(routeKey=f"调度队列_{uid}", text=f"调度队列 {uid}") def search_queue(self) -> list: """搜索所有调度队列实例""" queue_list = [] if (Config.app_path / "config/QueueConfig").exists(): for json_file in (Config.app_path / "config/QueueConfig").glob("*.json"): with json_file.open("r", encoding="utf-8") as f: info = json.load(f) queue_list.append([json_file.stem, info["QueueSet"]["Name"]]) return queue_list class QueueMemberSettingBox(QWidget): def __init__(self, uid: int, parent=None): super().__init__(parent) self.setObjectName(f"调度队列_{uid}") layout = QVBoxLayout() scrollArea = ScrollArea() scrollArea.setWidgetResizable(True) content_widget = QWidget() content_layout = QVBoxLayout(content_widget) self.queue_set = self.QueueSetSettingCard(self) self.time = self.TimeSettingCard(self) self.task = self.TaskSettingCard(self) self.history = self.HistoryCard(self, f"调度队列_{uid}") content_layout.addWidget(self.queue_set) content_layout.addWidget(self.time) content_layout.addWidget(self.task) content_layout.addWidget(self.history) content_layout.addStretch(1) scrollArea.setWidget(content_widget) layout.addWidget(scrollArea) self.setLayout(layout) class QueueSetSettingCard(HeaderCardWidget): def __init__(self, parent=None): super().__init__(parent) self.setTitle("队列设置") Layout = QVBoxLayout() self.card_Name = LineEditSettingCard( "请输入调度队列名称", FluentIcon.EDIT, "调度队列名称", "用于标识调度队列的名称", Config.queue_config.queueSet_Name, ) self.card_Enable = SwitchSettingCard( FluentIcon.HOME, "状态", "调度队列状态", Config.queue_config.queueSet_Enabled, ) self.card_AfterAccomplish = ComboBoxSettingCard( configItem=Config.queue_config.queueSet_AfterAccomplish, icon=FluentIcon.POWER_BUTTON, title="调度队列结束后", content="选择调度队列结束后的操作", texts=[ "无动作", "退出AUTO_MAA", "睡眠（win系统需禁用休眠）", "休眠", "关机", ], ) Layout.addWidget(self.card_Name) Layout.addWidget(self.card_Enable) Layout.addWidget(self.card_AfterAccomplish) self.viewLayout.addLayout(Layout) class TimeSettingCard(HeaderCardWidget): def __init__(self, parent=None): super().__init__(parent) self.setTitle("定时设置") widget_1 = QWidget() Layout_1 = QVBoxLayout(widget_1) widget_2 = QWidget() Layout_2 = QVBoxLayout(widget_2) Layout = QHBoxLayout() self.card_Time_0 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 1", "", Config.queue_config.time_TimeEnabled_0, Config.queue_config.time_TimeSet_0, ) self.card_Time_1 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 2", "", Config.queue_config.time_TimeEnabled_1, Config.queue_config.time_TimeSet_1, ) self.card_Time_2 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 3", "", Config.queue_config.time_TimeEnabled_2, Config.queue_config.time_TimeSet_2, ) self.card_Time_3 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 4", "", Config.queue_config.time_TimeEnabled_3, Config.queue_config.time_TimeSet_3, ) self.card_Time_4 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 5", "", Config.queue_config.time_TimeEnabled_4, Config.queue_config.time_TimeSet_4, ) self.card_Time_5 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 6", "", Config.queue_config.time_TimeEnabled_5, Config.queue_config.time_TimeSet_5, ) self.card_Time_6 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 7", "", Config.queue_config.time_TimeEnabled_6, Config.queue_config.time_TimeSet_6, ) self.card_Time_7 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 8", "", Config.queue_config.time_TimeEnabled_7, Config.queue_config.time_TimeSet_7, ) self.card_Time_8 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 9", "", Config.queue_config.time_TimeEnabled_8, Config.queue_config.time_TimeSet_8, ) self.card_Time_9 = TimeEditSettingCard( FluentIcon.STOP_WATCH, "定时 10", "", Config.queue_config.time_TimeEnabled_9, Config.queue_config.time_TimeSet_9, ) Layout_1.addWidget(self.card_Time_0) Layout_1.addWidget(self.card_Time_1) Layout_1.addWidget(self.card_Time_2) Layout_1.addWidget(self.card_Time_3) Layout_1.addWidget(self.card_Time_4) Layout_2.addWidget(self.card_Time_5) Layout_2.addWidget(self.card_Time_6) Layout_2.addWidget(self.card_Time_7) Layout_2.addWidget(self.card_Time_8) Layout_2.addWidget(self.card_Time_9) Layout.addWidget(widget_1) Layout.addWidget(widget_2) self.viewLayout.addLayout(Layout) class TaskSettingCard(HeaderCardWidget): def __init__(self, parent=None): super().__init__(parent) self.setTitle("任务队列") Layout = QVBoxLayout() member_list = self.search_member() self.card_Member_1 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_1, FluentIcon.APPLICATION, "任务实例 1", "第一个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_2 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_2, FluentIcon.APPLICATION, "任务实例 2", "第二个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_3 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_3, FluentIcon.APPLICATION, "任务实例 3", "第三个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_4 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_4, FluentIcon.APPLICATION, "任务实例 4", "第四个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_5 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_5, FluentIcon.APPLICATION, "任务实例 5", "第五个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_6 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_6, FluentIcon.APPLICATION, "任务实例 6", "第六个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_7 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_7, FluentIcon.APPLICATION, "任务实例 7", "第七个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_8 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_8, FluentIcon.APPLICATION, "任务实例 8", "第八个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_9 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_9, FluentIcon.APPLICATION, "任务实例 9", "第九个调起的脚本任务实例", member_list[0], member_list[1], ) self.card_Member_10 = NoOptionComboBoxSettingCard( Config.queue_config.queue_Member_10, FluentIcon.APPLICATION, "任务实例 10", "第十个调起的脚本任务实例", member_list[0], member_list[1], ) Layout.addWidget(self.card_Member_1) Layout.addWidget(self.card_Member_2) Layout.addWidget(self.card_Member_3) Layout.addWidget(self.card_Member_4) Layout.addWidget(self.card_Member_5) Layout.addWidget(self.card_Member_6) Layout.addWidget(self.card_Member_7) Layout.addWidget(self.card_Member_8) Layout.addWidget(self.card_Member_9) Layout.addWidget(self.card_Member_10) self.viewLayout.addLayout(Layout) def search_member(self) -> list: """搜索所有脚本实例""" member_list_name = ["禁用"] member_list_text = ["未启用"] if (Config.app_path / "config/MaaConfig").exists(): for subdir in (Config.app_path / "config/MaaConfig").iterdir(): if subdir.is_dir(): member_list_name.append(subdir.name) with (subdir / "config.json").open("r", encoding="utf-8") as f: info = json.load(f) if info["MaaSet"]["Name"] != "": member_list_text.append( f"{subdir.name} - {info["MaaSet"]["Name"]}" ) else: member_list_text.append(subdir.name) return [member_list_name, member_list_text] class HistoryCard(HeaderCardWidget): def __init__(self, parent=None, name: str = None): super().__init__(parent) self.setTitle("历史运行记录") self.text = TextBrowser() self.text.setMinimumHeight(300) history = Config.get_history(name) self.text.setPlainText(history["History"]) self.viewLayout.addWidget(self.text)