class Main(QWidget): ES_CONTINUOUS = 0x80000000 ES_SYSTEM_REQUIRED = 0x00000001 def __init__(self, config: AppConfig, notify: Notification, crypto: CryptoHandler): super().__init__() Config = config Notify = notify Crypto = crypto # uiLoader.registerCustomWidget(PushButton) # uiLoader.registerCustomWidget(LineEdit) # uiLoader.registerCustomWidget(TextBrowser) # uiLoader.registerCustomWidget(TableWidget) # uiLoader.registerCustomWidget(TimePicker) # uiLoader.registerCustomWidget(SpinBox) # uiLoader.registerCustomWidget(CheckBox) # uiLoader.registerCustomWidget(HeaderCardWidget) # uiLoader.registerCustomWidget(BodyLabel) # # 导入ui配置 # self.ui = uiLoader.load(Config.app_path / "resources/gui/main.ui") # self.ui.setWindowIcon( # QIcon(str(Config.app_path / "resources/icons/AUTO_MAA.ico")) # ) # # 初始化控件 # self.main_tab: QTabWidget = self.ui.findChild(QTabWidget, "tabWidget_main") # self.main_tab.currentChanged.connect(self.change_config) # self.user_set: QToolBox = self.ui.findChild(QToolBox, "toolBox_userset") # self.user_set.currentChanged.connect(lambda: self.update_user_info("normal")) # self.user_list_simple: TableWidget = self.ui.findChild( # TableWidget, "tableWidget_userlist_simple" # ) # self.user_list_simple.itemChanged.connect( # lambda item: self.change_user_Item(item, "simple") # ) # self.user_list_beta: TableWidget = self.ui.findChild( # TableWidget, "tableWidget_userlist_beta" # ) # self.user_list_beta.itemChanged.connect( # lambda item: self.change_user_Item(item, "beta") # ) # self.user_add: PushButton = self.ui.findChild(PushButton, "pushButton_new") # self.user_add.setIcon(FluentIcon.ADD_TO) # self.user_add.clicked.connect(self.add_user) # self.user_del: PushButton = self.ui.findChild(PushButton, "pushButton_del") # self.user_del.setIcon(FluentIcon.REMOVE_FROM) # self.user_del.clicked.connect(self.del_user) # self.user_switch: PushButton = self.ui.findChild( # PushButton, "pushButton_switch" # ) # self.user_switch.setIcon(FluentIcon.MOVE) # self.user_switch.clicked.connect(self.switch_user) # self.read_PASSWORD: PushButton = self.ui.findChild( # PushButton, "pushButton_password" # ) # self.read_PASSWORD.setIcon(FluentIcon.HIDE) # self.read_PASSWORD.clicked.connect(lambda: self.read("key")) # self.refresh: PushButton = self.ui.findChild(PushButton, "pushButton_refresh") # self.refresh.setIcon(FluentIcon.SYNC) # self.refresh.clicked.connect(lambda: self.update_user_info("clear")) # self.run_now: PushButton = self.ui.findChild(PushButton, "pushButton_runnow") # self.run_now.setIcon(FluentIcon.PLAY) # self.run_now.clicked.connect(lambda: self.maa_starter("自动代理")) # self.check_start: PushButton = self.ui.findChild( # PushButton, "pushButton_checkstart" # ) # self.check_start.setIcon(FluentIcon.PLAY) # self.check_start.clicked.connect(lambda: self.maa_starter("人工排查")) # self.maa_path: LineEdit = self.ui.findChild(LineEdit, "lineEdit_MAApath") # self.maa_path.textChanged.connect(self.change_config) # self.maa_path.setReadOnly(True) # self.get_maa_path: PushButton = self.ui.findChild( # PushButton, "pushButton_getMAApath" # ) # self.get_maa_path.setIcon(FluentIcon.FOLDER) # self.get_maa_path.clicked.connect(lambda: self.read("file_path_maa")) # self.set_maa: PushButton = self.ui.findChild(PushButton, "pushButton_setMAA") # self.set_maa.setIcon(FluentIcon.SETTING) # self.set_maa.clicked.connect(lambda: self.maa_starter("设置MAA_全局")) # self.routine: SpinBox = self.ui.findChild(SpinBox, "spinBox_routine") # self.routine.valueChanged.connect(self.change_config) # self.annihilation: SpinBox = self.ui.findChild(SpinBox, "spinBox_annihilation") # self.annihilation.valueChanged.connect(self.change_config) # self.num: SpinBox = self.ui.findChild(SpinBox, "spinBox_numt") # self.num.valueChanged.connect(self.change_config) # self.if_self_start: CheckBox = self.ui.findChild( # CheckBox, "checkBox_ifselfstart" # ) # self.if_self_start.stateChanged.connect(self.change_config) # self.if_sleep: CheckBox = self.ui.findChild(CheckBox, "checkBox_IfAllowSleep") # self.if_sleep.stateChanged.connect(self.change_config) # self.if_proxy_directly: CheckBox = self.ui.findChild( # CheckBox, "checkBox_ifproxydirectly" # ) # self.if_proxy_directly.stateChanged.connect(self.change_config) # self.if_send_mail: CheckBox = self.ui.findChild(CheckBox, "checkBox_ifsendmail") # self.if_send_mail.stateChanged.connect(self.change_config) # self.mail_address: LineEdit = self.ui.findChild( # LineEdit, "lineEdit_mailaddress" # ) # self.mail_address.textChanged.connect(self.change_config) # self.if_send_error_only: CheckBox = self.ui.findChild( # CheckBox, "checkBox_ifonlyerror" # ) # self.if_send_error_only.stateChanged.connect(self.change_config) # self.if_silence: CheckBox = self.ui.findChild(CheckBox, "checkBox_silence") # self.if_silence.stateChanged.connect(self.change_config) # self.boss_key: LineEdit = self.ui.findChild(LineEdit, "lineEdit_boss") # self.boss_key.textChanged.connect(self.change_config) # self.if_to_tray: CheckBox = self.ui.findChild(CheckBox, "checkBox_iftotray") # self.if_to_tray.stateChanged.connect(self.change_config) # self.check_update: PushButton = self.ui.findChild( # PushButton, "pushButton_check_update" # ) # self.check_update.setIcon(FluentIcon.UPDATE) # self.check_update.clicked.connect(self.check_version) # self.tips: TextBrowser = self.ui.findChild(TextBrowser, "textBrowser_tips") # self.tips.setOpenExternalLinks(True) # self.run_text: TextBrowser = self.ui.findChild(TextBrowser, "textBrowser_run") # self.wait_text: TextBrowser = self.ui.findChild(TextBrowser, "textBrowser_wait") # self.over_text: TextBrowser = self.ui.findChild(TextBrowser, "textBrowser_over") # self.error_text: TextBrowser = self.ui.findChild( # TextBrowser, "textBrowser_error" # ) # self.log_text: TextBrowser = self.ui.findChild(TextBrowser, "textBrowser_log") # self.start_time: List[Tuple[CheckBox, TimePicker]] = [] # for i in range(10): # self.start_time.append( # [ # self.ui.findChild(CheckBox, f"checkBox_t{i + 1}"), # self.ui.findChild(TimePicker, f"timeEdit_{i + 1}"), # ] # ) # self.start_time[i][0].stateChanged.connect(self.change_config) # self.start_time[i][1].timeChanged.connect(self.change_config) # self.change_password: PushButton = self.ui.findChild( # PushButton, "pushButton_changePASSWORD" # ) # self.change_password.setIcon(FluentIcon.VPN) # self.change_password.clicked.connect(self.change_PASSWORD) # 初始化线程 self.MaaManager = MaaManager(Config) self.MaaManager.question.connect(lambda: self.read("question_runner")) self.MaaManager.update_gui.connect(self.update_board) self.MaaManager.update_user_info.connect(self.change_user_info) self.MaaManager.push_notification.connect(Notify.push_notification) self.MaaManager.send_mail.connect(Notify.send_mail) self.MaaManager.accomplish.connect(lambda: self.maa_ender("自动代理_结束")) self.MaaManager.get_json.connect(self.get_maa_config) self.MaaManager.set_silence.connect(self.switch_silence) # self.last_time = "0000-00-00 00:00" # self.Timer = QtCore.QTimer() # self.Timer.timeout.connect(self.set_theme) # self.Timer.timeout.connect(self.set_system) # self.Timer.timeout.connect(self.timed_start) # self.Timer.start(1000) # 载入GUI数据 # self.update_user_info("normal") # self.update_config() # 启动后直接开始代理 if Config.content["Default"]["SelfSet.IfProxyDirectly"] == "True": self.maa_starter("自动代理") # def update_config(self): # """将Config中的程序配置同步至GUI界面""" # # 阻止GUI程序配置被立即读入程序形成死循环 # self.if_update_config = False # self.main_tab.setCurrentIndex( # Config.content["Default"]["SelfSet.MainIndex"] # ) # self.maa_path.setText(str(Path(Config.content["Default"]["MaaSet.path"]))) # self.routine.setValue(Config.content["Default"]["TimeLimit.routine"]) # self.annihilation.setValue( # Config.content["Default"]["TimeLimit.annihilation"] # ) # self.num.setValue(Config.content["Default"]["TimesLimit.run"]) # self.mail_address.setText(Config.content["Default"]["SelfSet.MailAddress"]) # self.boss_key.setText(Config.content["Default"]["SelfSet.BossKey"]) # self.if_self_start.setChecked( # bool(Config.content["Default"]["SelfSet.IfSelfStart"] == "True") # ) # self.if_sleep.setChecked( # bool(Config.content["Default"]["SelfSet.IfAllowSleep"] == "True") # ) # self.if_proxy_directly.setChecked( # bool(Config.content["Default"]["SelfSet.IfProxyDirectly"] == "True") # ) # self.if_send_mail.setChecked( # bool(Config.content["Default"]["SelfSet.IfSendMail"] == "True") # ) # self.mail_address.setVisible( # bool(Config.content["Default"]["SelfSet.IfSendMail"] == "True") # ) # self.if_send_error_only.setChecked( # bool( # Config.content["Default"]["SelfSet.IfSendMail.OnlyError"] == "True" # ) # ) # self.if_send_error_only.setVisible( # bool(Config.content["Default"]["SelfSet.IfSendMail"] == "True") # ) # self.if_silence.setChecked( # bool(Config.content["Default"]["SelfSet.IfSilence"] == "True") # ) # self.boss_key.setVisible( # bool(Config.content["Default"]["SelfSet.IfSilence"] == "True") # ) # self.if_to_tray.setChecked( # bool(Config.content["Default"]["SelfSet.IfToTray"] == "True") # ) # for i in range(10): # self.start_time[i][0].setChecked( # bool(Config.content["Default"][f"TimeSet.set{i + 1}"] == "True") # ) # time = QtCore.QTime( # int(Config.content["Default"][f"TimeSet.run{i + 1}"][:2]), # int(Config.content["Default"][f"TimeSet.run{i + 1}"][3:]), # ) # self.start_time[i][1].setTime(time) # self.if_update_config = True # 同步用户信息更改至GUI self.update_user_info("normal") def change_config(self): """将GUI中发生修改的程序配置同步至Config变量""" # 验证能否写入Config变量 if not self.if_update_config: return None # 验证MAA路径 if Path(Config.content["Default"]["MaaSet.path"]) != Path( self.maa_path.text() ): if (Path(self.maa_path.text()) / "MAA.exe").exists() and ( Path(self.maa_path.text()) / "config/gui.json" ).exists(): Config.content["Default"]["MaaSet.path"] = str( Path(self.maa_path.text()) ) self.get_maa_config(["Default"]) else: choice = MessageBox( "错误", "该路径下未找到MAA.exe或MAA配置文件，请重新设置MAA路径！", self.ui, ) if choice.exec(): pass Config.content["Default"][ "SelfSet.MainIndex" ] = self.main_tab.currentIndex() Config.content["Default"]["TimeLimit.routine"] = self.routine.value() Config.content["Default"][ "TimeLimit.annihilation" ] = self.annihilation.value() Config.content["Default"]["TimesLimit.run"] = self.num.value() Config.content["Default"]["SelfSet.MailAddress"] = self.mail_address.text() Config.content["Default"]["SelfSet.BossKey"] = self.boss_key.text() if self.if_sleep.isChecked(): Config.content["Default"]["SelfSet.IfAllowSleep"] = "True" else: Config.content["Default"]["SelfSet.IfAllowSleep"] = "False" if self.if_self_start.isChecked(): Config.content["Default"]["SelfSet.IfSelfStart"] = "True" else: Config.content["Default"]["SelfSet.IfSelfStart"] = "False" if self.if_proxy_directly.isChecked(): Config.content["Default"]["SelfSet.IfProxyDirectly"] = "True" else: Config.content["Default"]["SelfSet.IfProxyDirectly"] = "False" if self.if_send_mail.isChecked(): Config.content["Default"]["SelfSet.IfSendMail"] = "True" else: Config.content["Default"]["SelfSet.IfSendMail"] = "False" if self.if_send_error_only.isChecked(): Config.content["Default"]["SelfSet.IfSendMail.OnlyError"] = "True" else: Config.content["Default"]["SelfSet.IfSendMail.OnlyError"] = "False" if self.if_silence.isChecked(): Config.content["Default"]["SelfSet.IfSilence"] = "True" else: Config.content["Default"]["SelfSet.IfSilence"] = "False" if self.if_to_tray.isChecked(): Config.content["Default"]["SelfSet.IfToTray"] = "True" else: Config.content["Default"]["SelfSet.IfToTray"] = "False" for i in range(10): if self.start_time[i][0].isChecked(): Config.content["Default"][f"TimeSet.set{i + 1}"] = "True" else: Config.content["Default"][f"TimeSet.set{i + 1}"] = "False" time = self.start_time[i][1].getTime().toString("HH:mm") Config.content["Default"][f"TimeSet.run{i + 1}"] = time # 将配置信息同步至本地JSON文件 Config.save_config() # 同步程序配置至GUI self.update_config() def get_maa_config(self, info): """获取MAA配置文件""" # 获取全局MAA配置文件 if info == ["Default"]: shutil.copy( Path(Config.content["Default"]["MaaSet.path"]) / "config/gui.json", Config.app_path / "data/MAAconfig/Default", ) # 获取基建配置文件 # 获取高级用户MAA配置文件 elif info[2] in ["routine", "annihilation"]: ( Config.app_path / f"data/MAAconfig/{self.user_mode_list[info[0]]}/{info[1]}/{info[2]}" ).mkdir(parents=True, exist_ok=True) shutil.copy( Path(Config.content["Default"]["MaaSet.path"]) / "config/gui.json", Config.app_path / f"data/MAAconfig/{self.user_mode_list[info[0]]}/{info[1]}/{info[2]}", ) def set_theme(self): """手动更新主题色到组件""" self.user_list_simple.setStyleSheet("QTableWidget::item {}") self.user_list_beta.setStyleSheet("QTableWidget::item {}") def switch_silence(self, mode, emulator_path, boss_key): """切换静默模式""" if mode == "启用": self.Timer.timeout.disconnect() self.Timer.timeout.connect(self.set_theme) self.Timer.timeout.connect(self.set_system) self.Timer.timeout.connect(self.timed_start) self.Timer.timeout.connect( lambda: self.set_silence(emulator_path, boss_key) ) elif mode == "禁用": self.Timer.timeout.disconnect() self.Timer.timeout.connect(self.set_theme) self.Timer.timeout.connect(self.set_system) self.Timer.timeout.connect(self.timed_start) def maa_starter(self, mode): """启动MaaManager线程运行任务""" # 检查MAA路径是否可用 if ( not ( Path(Config.content["Default"]["MaaSet.path"]) / "MAA.exe" ).exists() and ( Path(Config.content["Default"]["MaaSet.path"]) / "config/gui.json" ).exists() ): choice = MessageBox("错误", "您还未正确配置MAA路径！", self.ui) choice.cancelButton.hide() choice.buttonLayout.insertStretch(1) if choice.exec(): return None self.maa_running_set(f"{mode}_开始") # 配置参数 self.MaaManager.mode = mode Config.cur.execute("SELECT * FROM adminx WHERE True") data = Config.cur.fetchall() self.MaaManager.data = [list(row) for row in data] # 启动执行线程 self.MaaManager.start() def maa_ender(self, mode): """中止MAA线程""" self.switch_silence("禁用", "", []) self.MaaManager.requestInterruption() self.MaaManager.wait() self.maa_running_set(mode) def maa_running_set(self, mode): """处理MAA运行过程中的GUI组件变化""" if "开始" in mode: self.MaaManager.accomplish.disconnect() self.user_add.setEnabled(False) self.user_del.setEnabled(False) self.user_switch.setEnabled(False) self.set_maa.setEnabled(False) self.update_user_info("read_only") if mode == "自动代理_开始": self.MaaManager.accomplish.connect( lambda: self.maa_ender("自动代理_结束") ) self.check_start.setEnabled(False) self.run_now.clicked.disconnect() self.run_now.setText("结束运行") self.run_now.clicked.connect(lambda: self.maa_ender("自动代理_结束")) elif mode == "人工排查_开始": self.MaaManager.accomplish.connect( lambda: self.maa_ender("人工排查_结束") ) self.run_now.setEnabled(False) self.check_start.clicked.disconnect() self.check_start.setText("中止排查") self.check_start.clicked.connect( lambda: self.maa_ender("人工排查_结束") ) elif mode == "设置MAA_全局_开始" or mode == "设置MAA_用户_开始": self.MaaManager.accomplish.connect( lambda: self.maa_ender("设置MAA_结束") ) self.run_now.setEnabled(False) self.check_start.setEnabled(False) elif "结束" in mode: shutil.copy( Config.app_path / "data/MAAconfig/Default/gui.json", Path(Config.content["Default"]["MaaSet.path"]) / "config", ) self.user_add.setEnabled(True) self.user_del.setEnabled(True) self.user_switch.setEnabled(True) self.set_maa.setEnabled(True) self.update_user_info("editable") if mode == "自动代理_结束": self.check_start.setEnabled(True) self.run_now.clicked.disconnect() self.run_now.setText("立即执行") self.run_now.clicked.connect(lambda: self.maa_starter("自动代理")) elif mode == "人工排查_结束": self.run_now.setEnabled(True) self.check_start.clicked.disconnect() self.check_start.setText("开始排查") self.check_start.clicked.connect(lambda: self.maa_starter("人工排查")) elif mode == "设置MAA_结束": self.run_now.setEnabled(True) self.check_start.setEnabled(True) def server_date(self): """获取当前的服务器日期""" dt = datetime.datetime.now() if dt.time() < datetime.datetime.min.time().replace(hour=4): dt = dt - datetime.timedelta(days=1) return dt.strftime("%Y-%m-%d")