初步完成调度队列开发

app/services/__init__.py

@@ -31,5 +31,6 @@ __license__ = "GPL-3.0 license"
 
 from .notification import Notification
 from .security import CryptoHandler
+from .system import SystemHandler
 
-__all__ = ["Notification", "CryptoHandler"]
+__all__ = ["Notification", "CryptoHandler", "SystemHandler"]

app/services/notification.py

@@ -44,15 +44,17 @@ class Notification:
     def push_notification(self, title, message, ticker, t):
         """推送系统通知"""
 
-        notification.notify(
-            title=title,
-            message=message,
-            app_name="AUTO_MAA",
-            app_icon=str(self.config.app_path / "resources/icons/AUTO_MAA.ico"),
-            timeout=t,
-            ticker=ticker,
-            toast=True,
-        )
+        if self.config.global_config.get(self.config.global_config.notify_IfPushPlyer):
+
+            notification.notify(
+                title=title,
+                message=message,
+                app_name="AUTO_MAA",
+                app_icon=str(self.config.app_path / "resources/icons/AUTO_MAA.ico"),
+                timeout=t,
+                ticker=ticker,
+                toast=True,
+            )
 
         return True
 
@@ -62,34 +64,43 @@ class Notification:
         # 声明：此邮箱为AUTO_MAA项目组资产，未经授权不得私自使用
         # 注意：此声明注释只有使用者更换发信邮箱时才能删除，本条规则优先级高于GPLv3
 
-        # 第三方 SMTP 服务配置
-        mail_host = "smtp.163.com"  # 设置服务器
-        mail_sender = "AUTO_MAA_server@163.com"  # 用户名
-        mail_key = "SYrq87nDLD4RNB5T"  # 授权码 24/11/15
+        if self.config.global_config.get(self.config.global_config.notify_IfSendMail):
 
-        # 定义邮件正文
-        message = MIMEText(content, "plain", "utf-8")
-        message["From"] = formataddr(
-            (Header("AUTO_MAA通知服务", "utf-8").encode(), "AUTO_MAA_server@163.com")
-        )  # 发件人显示的名字
-        message["To"] = formataddr(
-            (
-                Header("AUTO_MAA用户", "utf-8").encode(),
-                self.config.content["Default"]["SelfSet.MailAddress"],
-            )
-        )  # 收件人显示的名字
-        message["Subject"] = Header(title, "utf-8")
+            # 第三方 SMTP 服务配置
+            mail_host = "smtp.163.com"  # 设置服务器
+            mail_sender = "AUTO_MAA_server@163.com"  # 用户名
+            mail_key = "SYrq87nDLD4RNB5T"  # 授权码 24/11/15
 
-        try:
-            smtpObj = smtplib.SMTP_SSL(mail_host, 465)  # 465为SMTP_SSL默认端口
-            smtpObj.login(mail_sender, mail_key)
-            smtpObj.sendmail(
-                mail_sender,
-                self.config.content["Default"]["SelfSet.MailAddress"],
-                message.as_string(),
-            )
-            return True
-        except smtplib.SMTPException as e:
-            return f"发送邮件时出错：\n{e}"
-        finally:
-            smtpObj.quit()
+            # 定义邮件正文
+            message = MIMEText(content, "plain", "utf-8")
+            message["From"] = formataddr(
+                (
+                    Header("AUTO_MAA通知服务", "utf-8").encode(),
+                    "AUTO_MAA_server@163.com",
+                )
+            )  # 发件人显示的名字
+            message["To"] = formataddr(
+                (
+                    Header("AUTO_MAA用户", "utf-8").encode(),
+                    self.config.global_config.get(
+                        self.config.global_config.notify_MailAddress
+                    ),
+                )
+            )  # 收件人显示的名字
+            message["Subject"] = Header(title, "utf-8")
+
+            try:
+                smtpObj = smtplib.SMTP_SSL(mail_host, 465)  # 465为SMTP_SSL默认端口
+                smtpObj.login(mail_sender, mail_key)
+                smtpObj.sendmail(
+                    mail_sender,
+                    self.config.global_config.get(
+                        self.config.global_config.notify_MailAddress
+                    ),
+                    message.as_string(),
+                )
+                return True
+            except smtplib.SMTPException as e:
+                return f"发送邮件时出错：\n{e}"
+            finally:
+                smtpObj.quit()

app/services/security.py

@@ -20,7 +20,7 @@
 
 """
 AUTO_MAA
-AUTO_MAA主程序
+AUTO_MAA安全服务
 v4.2
 作者：DLmaster_361
 """

app/services/system.py

@@ -0,0 +1,125 @@
+#   <AUTO_MAA:A MAA Multi Account Management and Automation Tool>
+#   Copyright © <2024> <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/>.
+
+#   DLmaster_361@163.com
+
+"""
+AUTO_MAA
+AUTO_MAA系统服务
+v4.2
+作者：DLmaster_361
+"""
+
+import ctypes
+import win32gui
+import win32process
+import winreg
+import psutil
+
+from app import AppConfig
+
+
+class SystemHandler:
+
+    ES_CONTINUOUS = 0x80000000
+    ES_SYSTEM_REQUIRED = 0x00000001
+
+    def __init__(self, config: AppConfig):
+
+        self.config = config
+
+        self.set_Sleep()
+        self.set_SelfStart()
+
+    def set_Sleep(self):
+        """同步系统休眠状态"""
+
+        if self.config.global_config.get(
+            self.config.global_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):
+        """同步开机自启"""
+
+        if (
+            self.config.global_config.get(self.config.global_config.start_IfSelfStart)
+            and not self.is_startup()
+        ):
+            key = winreg.OpenKey(
+                winreg.HKEY_CURRENT_USER,
+                r"Software\Microsoft\Windows\CurrentVersion\Run",
+                winreg.KEY_SET_VALUE,
+                winreg.KEY_ALL_ACCESS | winreg.KEY_WRITE | winreg.KEY_CREATE_SUB_KEY,
+            )
+            winreg.SetValueEx(
+                key, self.config.app_name, 0, winreg.REG_SZ, self.config.app_path_sys
+            )
+            winreg.CloseKey(key)
+        elif (
+            not self.config.global_config.get(
+                self.config.global_config.start_IfSelfStart
+            )
+            and self.is_startup()
+        ):
+            key = winreg.OpenKey(
+                winreg.HKEY_CURRENT_USER,
+                r"Software\Microsoft\Windows\CurrentVersion\Run",
+                winreg.KEY_SET_VALUE,
+                winreg.KEY_ALL_ACCESS | winreg.KEY_WRITE | winreg.KEY_CREATE_SUB_KEY,
+            )
+            winreg.DeleteValue(key, self.config.app_name)
+            winreg.CloseKey(key)
+
+    def is_startup(self):
+        """判断程序是否已经开机自启"""
+
+        key = winreg.OpenKey(
+            winreg.HKEY_CURRENT_USER,
+            r"Software\Microsoft\Windows\CurrentVersion\Run",
+            0,
+            winreg.KEY_READ,
+        )
+
+        try:
+            value, _ = winreg.QueryValueEx(key, self.config.app_name)
+            winreg.CloseKey(key)
+            return True
+        except FileNotFoundError:
+            winreg.CloseKey(key)
+            return False
+
+    def get_window_info(self):
+        """获取当前窗口信息"""
+
+        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