353 lines
12 KiB
Python
353 lines
12 KiB
Python
# AUTO_MAA:A MAA Multi Account Management and Automation Tool
|
||
# Copyright © 2024-2025 DLmaster361
|
||
|
||
# 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 <https://www.gnu.org/licenses/>.
|
||
|
||
# Contact: DLmaster_361@163.com
|
||
|
||
"""
|
||
AUTO_MAA
|
||
AUTO_MAA系统服务
|
||
v4.4
|
||
作者：DLmaster_361
|
||
"""
|
||
|
||
from PySide6.QtWidgets import QApplication
|
||
import sys
|
||
import ctypes
|
||
import win32gui
|
||
import win32process
|
||
import psutil
|
||
import subprocess
|
||
import tempfile
|
||
import getpass
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
from app.core import Config, logger
|
||
|
||
|
||
class _SystemHandler:
|
||
|
||
ES_CONTINUOUS = 0x80000000
|
||
ES_SYSTEM_REQUIRED = 0x00000001
|
||
|
||
def __init__(self):
|
||
|
||
self.set_Sleep()
|
||
self.set_SelfStart()
|
||
|
||
def set_Sleep(self) -> None:
|
||
"""同步系统休眠状态"""
|
||
|
||
if Config.get(Config.function_IfAllowSleep):
|
||
# 设置系统电源状态
|
||
ctypes.windll.kernel32.SetThreadExecutionState(
|
||
self.ES_CONTINUOUS | self.ES_SYSTEM_REQUIRED
|
||
)
|
||
else:
|
||
# 恢复系统电源状态
|
||
ctypes.windll.kernel32.SetThreadExecutionState(self.ES_CONTINUOUS)
|
||
|
||
def set_SelfStart(self) -> None:
|
||
"""同步开机自启"""
|
||
|
||
if Config.get(Config.start_IfSelfStart) and not self.is_startup():
|
||
|
||
# 创建任务计划
|
||
try:
|
||
|
||
# 获取当前用户和时间
|
||
current_user = getpass.getuser()
|
||
current_time = datetime.now().strftime("%Y-%m-%dT%H:%M:%S")
|
||
|
||
# XML 模板
|
||
xml_content = f"""<?xml version="1.0" encoding="UTF-16"?>
|
||
<Task version="1.2" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
|
||
<RegistrationInfo>
|
||
<Date>{current_time}</Date>
|
||
<Author>{current_user}</Author>
|
||
<Description>AUTO_MAA自启动服务</Description>
|
||
<URI>\\AUTO_MAA_AutoStart</URI>
|
||
</RegistrationInfo>
|
||
<Triggers>
|
||
<LogonTrigger>
|
||
<StartBoundary>{current_time}</StartBoundary>
|
||
<Enabled>true</Enabled>
|
||
</LogonTrigger>
|
||
</Triggers>
|
||
<Principals>
|
||
<Principal id="Author">
|
||
<LogonType>InteractiveToken</LogonType>
|
||
<RunLevel>HighestAvailable</RunLevel>
|
||
</Principal>
|
||
</Principals>
|
||
<Settings>
|
||
<MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
|
||
<DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
|
||
<StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
|
||
<AllowHardTerminate>false</AllowHardTerminate>
|
||
<StartWhenAvailable>true</StartWhenAvailable>
|
||
<RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>
|
||
<IdleSettings>
|
||
<StopOnIdleEnd>false</StopOnIdleEnd>
|
||
<RestartOnIdle>false</RestartOnIdle>
|
||
</IdleSettings>
|
||
<AllowStartOnDemand>true</AllowStartOnDemand>
|
||
<Enabled>true</Enabled>
|
||
<Hidden>false</Hidden>
|
||
<RunOnlyIfIdle>false</RunOnlyIfIdle>
|
||
<WakeToRun>false</WakeToRun>
|
||
<ExecutionTimeLimit>PT0S</ExecutionTimeLimit>
|
||
<Priority>7</Priority>
|
||
</Settings>
|
||
<Actions Context="Author">
|
||
<Exec>
|
||
<Command>"{Config.app_path_sys}"</Command>
|
||
</Exec>
|
||
</Actions>
|
||
</Task>"""
|
||
|
||
# 创建临时 XML 文件并执行
|
||
with tempfile.NamedTemporaryFile(
|
||
mode="w", suffix=".xml", delete=False, encoding="utf-16"
|
||
) as f:
|
||
f.write(xml_content)
|
||
xml_file = f.name
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
[
|
||
"schtasks",
|
||
"/create",
|
||
"/tn",
|
||
"AUTO_MAA_AutoStart",
|
||
"/xml",
|
||
xml_file,
|
||
"/f",
|
||
],
|
||
creationflags=subprocess.CREATE_NO_WINDOW,
|
||
stdin=subprocess.DEVNULL,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
|
||
if result.returncode == 0:
|
||
logger.success(
|
||
f"程序自启动任务计划已创建: {Config.app_path_sys}",
|
||
module="系统服务",
|
||
)
|
||
else:
|
||
logger.error(
|
||
f"程序自启动任务计划创建失败: {result.stderr}",
|
||
module="系统服务",
|
||
)
|
||
|
||
finally:
|
||
# 删除临时文件
|
||
try:
|
||
Path(xml_file).unlink()
|
||
except:
|
||
pass
|
||
|
||
except Exception as e:
|
||
logger.exception(f"程序自启动任务计划创建失败: {e}", module="系统服务")
|
||
|
||
elif not Config.get(Config.start_IfSelfStart) and self.is_startup():
|
||
|
||
try:
|
||
|
||
result = subprocess.run(
|
||
["schtasks", "/delete", "/tn", "AUTO_MAA_AutoStart", "/f"],
|
||
creationflags=subprocess.CREATE_NO_WINDOW,
|
||
stdin=subprocess.DEVNULL,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
|
||
if result.returncode == 0:
|
||
logger.success("程序自启动任务计划已删除", module="系统服务")
|
||
else:
|
||
logger.error(
|
||
f"程序自启动任务计划删除失败: {result.stderr}",
|
||
module="系统服务",
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.exception(f"程序自启动任务计划删除失败: {e}", module="系统服务")
|
||
|
||
def set_power(self, mode) -> None:
|
||
"""
|
||
执行系统电源操作
|
||
|
||
:param mode: 电源操作模式，支持 "NoAction", "Shutdown", "Hibernate", "Sleep", "KillSelf", "ShutdownForce"
|
||
"""
|
||
|
||
if sys.platform.startswith("win"):
|
||
|
||
if mode == "NoAction":
|
||
|
||
logger.info("不执行系统电源操作", module="系统服务")
|
||
|
||
elif mode == "Shutdown":
|
||
|
||
self.kill_emulator_processes()
|
||
logger.info("执行关机操作", module="系统服务")
|
||
subprocess.run(["shutdown", "/s", "/t", "0"])
|
||
|
||
elif mode == "ShutdownForce":
|
||
logger.info("执行强制关机操作", module="系统服务")
|
||
subprocess.run(["shutdown", "/s", "/t", "0", "/f"])
|
||
|
||
elif mode == "Hibernate":
|
||
|
||
logger.info("执行休眠操作", module="系统服务")
|
||
subprocess.run(["shutdown", "/h"])
|
||
|
||
elif mode == "Sleep":
|
||
|
||
logger.info("执行睡眠操作", module="系统服务")
|
||
subprocess.run(
|
||
["rundll32.exe", "powrprof.dll,SetSuspendState", "0,1,0"]
|
||
)
|
||
|
||
elif mode == "KillSelf":
|
||
|
||
logger.info("执行退出主程序操作", module="系统服务")
|
||
Config.main_window.close()
|
||
QApplication.quit()
|
||
sys.exit(0)
|
||
|
||
elif sys.platform.startswith("linux"):
|
||
|
||
if mode == "NoAction":
|
||
|
||
logger.info("不执行系统电源操作", module="系统服务")
|
||
|
||
elif mode == "Shutdown":
|
||
|
||
logger.info("执行关机操作", module="系统服务")
|
||
subprocess.run(["shutdown", "-h", "now"])
|
||
|
||
elif mode == "Hibernate":
|
||
|
||
logger.info("执行休眠操作", module="系统服务")
|
||
subprocess.run(["systemctl", "hibernate"])
|
||
|
||
elif mode == "Sleep":
|
||
|
||
logger.info("执行睡眠操作", module="系统服务")
|
||
subprocess.run(["systemctl", "suspend"])
|
||
|
||
elif mode == "KillSelf":
|
||
|
||
logger.info("执行退出主程序操作", module="系统服务")
|
||
Config.main_window.close()
|
||
QApplication.quit()
|
||
sys.exit(0)
|
||
|
||
def kill_emulator_processes(self):
|
||
"""这里暂时仅支持 MuMu 模拟器"""
|
||
|
||
logger.info("正在清除模拟器进程", module="系统服务")
|
||
|
||
keywords = ["Nemu", "nemu", "emulator", "MuMu"]
|
||
for proc in psutil.process_iter(["pid", "name"]):
|
||
try:
|
||
pname = proc.info["name"].lower()
|
||
if any(keyword.lower() in pname for keyword in keywords):
|
||
proc.kill()
|
||
logger.info(
|
||
f"已关闭 MuMu 模拟器进程: {proc.info['name']}",
|
||
module="系统服务",
|
||
)
|
||
except (psutil.NoSuchProcess, psutil.AccessDenied):
|
||
continue
|
||
|
||
logger.success("模拟器进程清除完成", module="系统服务")
|
||
|
||
def is_startup(self) -> bool:
|
||
"""判断程序是否已经开机自启"""
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
["schtasks", "/query", "/tn", "AUTO_MAA_AutoStart"],
|
||
creationflags=subprocess.CREATE_NO_WINDOW,
|
||
stdin=subprocess.DEVNULL,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
return result.returncode == 0
|
||
except Exception as e:
|
||
logger.exception(f"检查任务计划程序失败: {e}", module="系统服务")
|
||
return False
|
||
|
||
def get_window_info(self) -> list:
|
||
"""获取当前前台窗口信息"""
|
||
|
||
def callback(hwnd, window_info):
|
||
if win32gui.IsWindowVisible(hwnd) and win32gui.GetWindowText(hwnd):
|
||
_, pid = win32process.GetWindowThreadProcessId(hwnd)
|
||
process = psutil.Process(pid)
|
||
window_info.append((win32gui.GetWindowText(hwnd), process.exe()))
|
||
return True
|
||
|
||
window_info = []
|
||
win32gui.EnumWindows(callback, window_info)
|
||
return window_info
|
||
|
||
def kill_process(self, path: Path) -> None:
|
||
"""
|
||
根据路径中止进程
|
||
|
||
:param path: 进程路径
|
||
"""
|
||
|
||
logger.info(f"开始中止进程: {path}", module="系统服务")
|
||
|
||
for pid in self.search_pids(path):
|
||
killprocess = subprocess.Popen(
|
||
f"taskkill /F /T /PID {pid}",
|
||
shell=True,
|
||
creationflags=subprocess.CREATE_NO_WINDOW,
|
||
)
|
||
killprocess.wait()
|
||
|
||
logger.success(f"进程已中止: {path}", module="系统服务")
|
||
|
||
def search_pids(self, path: Path) -> list:
|
||
"""
|
||
根据路径查找进程PID
|
||
|
||
:param path: 进程路径
|
||
:return: 匹配的进程PID列表
|
||
"""
|
||
|
||
logger.info(f"开始查找进程 PID: {path}", module="系统服务")
|
||
|
||
pids = []
|
||
for proc in psutil.process_iter(["pid", "exe"]):
|
||
try:
|
||
if proc.info["exe"] and proc.info["exe"].lower() == str(path).lower():
|
||
pids.append(proc.info["pid"])
|
||
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
|
||
# 进程可能在此期间已结束或无法访问，忽略这些异常
|
||
pass
|
||
return pids
|
||
|
||
|
||
System = _SystemHandler()
|