# # 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 PySide6.QtWidgets import ( QWidget, # QMainWindow, # QApplication, # QSystemTrayIcon, # QFileDialog, # QTabWidget, # QToolBox, # QComboBox, # QTableWidgetItem, # QHeaderView, # QVBoxLayout, QStackedWidget, QHBoxLayout, ) from qfluentwidgets import ( Action, PushButton, LineEdit, PasswordLineEdit, qconfig, TableWidget, Pivot, TimePicker, ComboBox, CheckBox, ScrollArea, SpinBox, FluentIcon, SwitchButton, RoundMenu, MessageBox, MessageBoxBase, HeaderCardWidget, BodyLabel, CommandBar, SubtitleLabel, GroupHeaderCardWidget, SwitchSettingCard, ExpandGroupSettingCard, SingleDirectionScrollArea, PushSettingCard, ) from PySide6.QtUiTools import QUiLoader from PySide6.QtGui import QIcon, QCloseEvent from PySide6 import QtCore from functools import partial from typing import List, Tuple from pathlib import Path import os import datetime import ctypes import subprocess import shutil import win32gui import win32process import psutil import pyautogui import time import winreg import requests uiLoader = QUiLoader() from app import AppConfig, MaaConfig from app.services import Notification, CryptoHandler from app.utils import Updater, version_text from .Widget import InputMessageBox, LineEditSettingCard, SpinBoxSettingCard class MemberManager(QWidget): def __init__(self, config: AppConfig, notify: Notification, crypto: CryptoHandler): super(MemberManager, self).__init__() self.setObjectName("脚本管理") self.config = config self.notify = notify self.crypto = crypto layout = QVBoxLayout(self) self.tools = CommandBar() self.member_manager = MemberSettingBox(self.config) # 逐个添加动作 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, ), ] ) # 批量添加动作 self.tools.addAction( Action( FluentIcon.HIDE, "显示/隐藏密码", checkable=True, triggered=self.show_password, ), ) layout.addWidget(self.tools) layout.addWidget(self.member_manager) def add_setting_box(self): """添加一个脚本实例""" choice = InputMessageBox( self, "选择一个脚本类型并添加相应脚本实例", "选择脚本类型", "选择", ["MAA"] ) if choice.exec() and choice.input.currentIndex() != -1: if choice.input.currentText() == "MAA": index = len(self.member_manager.search_member()) + 1 qconfig.load( self.config.app_path / f"config/MaaConfig/脚本_{index}/config.json", self.config.maa_config, ) self.config.maa_config.set(self.config.maa_config.MaaSet_Name, "") self.config.maa_config.set(self.config.maa_config.MaaSet_Path, ".") self.config.maa_config.set( self.config.maa_config.RunSet_AnnihilationTimeLimit, 40 ) self.config.maa_config.set( self.config.maa_config.RunSet_RoutineTimeLimit, 10 ) self.config.maa_config.set( self.config.maa_config.RunSet_RunTimesLimit, 3 ) self.config.maa_config.set(self.config.maa_config.MaaSet_Name, "") self.config.maa_config.set(self.config.maa_config.MaaSet_Name, "") self.config.maa_config.set(self.config.maa_config.MaaSet_Name, "") self.config.maa_config.save() self.member_manager.add_MaaSettingBox(index) self.member_manager.switch_SettingBox(index) def del_setting_box(self): """删除一个脚本实例""" name = self.member_manager.pivot.currentRouteKey() if name == None: return None choice = MessageBox( "确认", f"确定要删除 {name} 实例吗？", self, ) if choice.exec(): member_list = self.member_manager.search_member() move_list = [_ for _ in member_list if int(_[0][3:]) > int(name[3:])] type = [_[1] for _ in member_list if _[0] == name] index = max(int(name[3:]) - 1, 1) shutil.rmtree(self.config.app_path / f"config/{type[0]}Config/{name}") for member in move_list: if ( self.config.app_path / f"config/{member[1]}Config/{member[0]}" ).exists(): ( self.config.app_path / f"config/{member[1]}Config/{member[0]}" ).rename( self.config.app_path / f"config/{member[1]}Config/{member[0][:3]}{int(member[0][3:])-1}", ) self.member_manager.clear_SettingBox() self.member_manager.show_SettingBox() self.member_manager.switch_SettingBox(index, if_after_clear=True) def left_setting_box(self): """向左移动脚本实例""" name = self.member_manager.pivot.currentRouteKey() if name == None: return None member_list = self.member_manager.search_member() index = int(name[3:]) if index == 1: return None type_right = [_[1] for _ in member_list if _[0] == name] type_left = [_[1] for _ in member_list if _[0] == f"脚本_{index-1}"] (self.config.app_path / f"config/{type_right[0]}Config/脚本_{index}").rename( self.config.app_path / f"config/{type_right[0]}Config/脚本_0", ) (self.config.app_path / f"config/{type_left[0]}Config/脚本_{index-1}").rename( self.config.app_path / f"config/{type_left[0]}Config/脚本_{index}", ) (self.config.app_path / f"config/{type_right[0]}Config/脚本_0").rename( self.config.app_path / f"config/{type_right[0]}Config/脚本_{index-1}", ) self.member_manager.clear_SettingBox() self.member_manager.show_SettingBox() self.member_manager.switch_SettingBox(index - 1, if_after_clear=True) def right_setting_box(self): """向左移动脚本实例""" name = self.member_manager.pivot.currentRouteKey() if name == None: return None member_list = self.member_manager.search_member() index = int(name[3:]) if index == len(member_list): return None type_left = [_[1] for _ in member_list if _[0] == name] type_right = [_[1] for _ in member_list if _[0] == f"脚本_{index+1}"] (self.config.app_path / f"config/{type_left[0]}Config/脚本_{index}").rename( self.config.app_path / f"config/{type_left[0]}Config/脚本_0", ) (self.config.app_path / f"config/{type_right[0]}Config/脚本_{index+1}").rename( self.config.app_path / f"config/{type_right[0]}Config/脚本_{index}", ) (self.config.app_path / f"config/{type_left[0]}Config/脚本_0").rename( self.config.app_path / f"config/{type_left[0]}Config/脚本_{index+1}", ) self.member_manager.clear_SettingBox() self.member_manager.show_SettingBox() self.member_manager.switch_SettingBox(index + 1, if_after_clear=True) def show_password(self): pass class MemberSettingBox(QWidget): def __init__(self, config: AppConfig): super().__init__() self.setObjectName("脚本管理") self.config = config self.pivot = Pivot(self) self.stackedWidget = QStackedWidget(self) self.Layout = QVBoxLayout(self) self.SubInterface: List[MaaSettingBox] = [] self.Layout.addWidget(self.pivot, 0, QtCore.Qt.AlignHCenter) self.Layout.addWidget(self.stackedWidget) self.Layout.setContentsMargins(0, 0, 0, 0) self.pivot.currentItemChanged.connect( lambda index: self.stackedWidget.setCurrentWidget( self.findChild(QWidget, index) ) ) self.pivot.currentItemChanged.connect( lambda index: qconfig.load( self.config.app_path / f"config/MaaConfig/{index}/config.json", self.config.maa_config, ) ) self.show_SettingBox() self.switch_SettingBox(1) def show_SettingBox(self) -> None: """加载所有子界面""" member_list = self.search_member() for member in member_list: if member[1] == "Maa": self.add_MaaSettingBox(int(member[0][3:])) def switch_SettingBox(self, index: int, if_after_clear: bool = False) -> None: """切换到指定的子界面""" member_list = self.search_member() if index > len(member_list): return None type = [_[1] for _ in member_list if _[0] == f"脚本_{index}"] if if_after_clear: self.pivot.currentItemChanged.disconnect() self.stackedWidget.setCurrentWidget(self.SubInterface[index - 1]) self.pivot.setCurrentItem(self.SubInterface[index - 1].objectName()) qconfig.load( self.config.app_path / f"config/{type[0]}Config/{self.SubInterface[index-1].objectName()}/config.json", self.config.maa_config, ) if if_after_clear: self.pivot.currentItemChanged.connect( lambda index: self.stackedWidget.setCurrentWidget( self.findChild(QWidget, index) ) ) self.pivot.currentItemChanged.connect( lambda index: qconfig.load( self.config.app_path / f"config/MaaConfig/{index}/config.json", self.config.maa_config, ) ) def clear_SettingBox(self) -> None: """清空所有子界面""" for sub_interface in self.SubInterface: self.stackedWidget.removeWidget(sub_interface) sub_interface.deleteLater() self.SubInterface.clear() self.pivot.clear() def add_MaaSettingBox(self, uid: int) -> None: """添加一个MAA设置界面""" maa_setting_box = MaaSettingBox(self.config, uid) self.SubInterface.append(maa_setting_box) self.stackedWidget.addWidget(self.SubInterface[-1]) self.pivot.addItem(routeKey=f"脚本_{uid}", text=f"脚本 {uid}") def search_member(self) -> list: """搜索所有脚本实例""" member_list = [] if (self.config.app_path / "config/MaaConfig").exists(): for subdir in (self.config.app_path / "config/MaaConfig").iterdir(): if subdir.is_dir(): member_list.append([subdir.name, "Maa"]) return member_list class MaaSettingBox(QWidget): def __init__(self, config: AppConfig, uid: int): super().__init__() self.setObjectName(f"脚本_{uid}") self.config = config layout = QVBoxLayout() scrollArea = ScrollArea() scrollArea.setWidgetResizable(True) content_widget = QWidget() content_layout = QVBoxLayout(content_widget) self.app_setting = self.AppSettingCard(self, self.config.maa_config) content_layout.addWidget(self.app_setting) content_layout.addStretch(1) scrollArea.setWidget(content_widget) layout.addWidget(scrollArea) self.setLayout(layout) class AppSettingCard(HeaderCardWidget): def __init__(self, parent=None, maa_config: MaaConfig = None): super().__init__(parent) self.setTitle("MAA实例") self.maa_config = maa_config Layout = QVBoxLayout() self.card_Name = LineEditSettingCard( "实例名称", FluentIcon.EDIT, "实例名称", "用于标识MAA实例的名称", self.maa_config.MaaSet_Name, ) self.card_Path = PushSettingCard( "选择文件夹", FluentIcon.FOLDER, "MAA目录", self.maa_config.get(self.maa_config.MaaSet_Path), ) self.card_Set = PushSettingCard( "设置", FluentIcon.HOME, "MAA全局配置", "简洁模式下MAA将继承全局配置", ) self.RunSet = self.RunSetSettingCard(self, self.maa_config) self.card_Path.clicked.connect(self.PathClicked) Layout.addWidget(self.card_Name) Layout.addWidget(self.card_Path) Layout.addWidget(self.card_Set) Layout.addWidget(self.RunSet) self.viewLayout.addLayout(Layout) def PathClicked(self): folder = QFileDialog.getExistingDirectory(self, "选择MAA目录", "./") if not folder or self.maa_config.get(self.maa_config.MaaSet_Path) == folder: return self.maa_config.set(self.maa_config.MaaSet_Path, folder) self.card_Path.setContent(folder) class RunSetSettingCard(ExpandGroupSettingCard): def __init__(self, parent=None, maa_config: MaaConfig = None): super().__init__( FluentIcon.SETTING, "运行", "MAA运行调控选项", parent, ) self.maa_config = maa_config widget = QWidget() Layout = QVBoxLayout(widget) self.AnnihilationTimeLimit = SpinBoxSettingCard( (1, 1024), FluentIcon.PAGE_RIGHT, "剿灭代理超时限制", "MAA日志无变化时间超过该阈值视为超时", self.maa_config.RunSet_AnnihilationTimeLimit, ) self.RoutineTimeLimit = SpinBoxSettingCard( (1, 1024), FluentIcon.PAGE_RIGHT, "日常代理超时限制", "MAA日志无变化时间超过该阈值视为超时", self.maa_config.RunSet_RoutineTimeLimit, ) self.RunTimesLimit = SpinBoxSettingCard( (1, 1024), FluentIcon.PAGE_RIGHT, "代理重试次数限制", "若超过该次数限制仍未完成代理，视为代理失败", self.maa_config.RunSet_RunTimesLimit, ) Layout.addWidget(self.AnnihilationTimeLimit) Layout.addWidget(self.RoutineTimeLimit) Layout.addWidget(self.RunTimesLimit) self.viewLayout.setContentsMargins(0, 0, 0, 0) self.viewLayout.setSpacing(0) self.addGroupWidget(widget)