feat: 支持本地数据保存时加密

app/utils/ImageUtils.py

@@ -0,0 +1,95 @@
+#   AUTO_MAA:A MAA Multi Account Management and Automation Tool
+#   Copyright © 2025 ClozyA
+
+#   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
+作者：ClozyA
+"""
+
+import base64
+import hashlib
+from pathlib import Path
+
+from PIL import Image
+
+
+class ImageUtils:
+    @staticmethod
+    def get_base64_from_file(image_path):
+        """从本地文件读取并返回base64编码字符串"""
+        with open(image_path, "rb") as f:
+            return base64.b64encode(f.read()).decode("utf-8")
+
+    @staticmethod
+    def calculate_md5_from_file(image_path):
+        """从本地文件读取并返回md5值（hex字符串）"""
+        with open(image_path, "rb") as f:
+            return hashlib.md5(f.read()).hexdigest()
+
+    @staticmethod
+    def calculate_md5_from_base64(base64_content):
+        """从base64字符串计算md5"""
+        image_data = base64.b64decode(base64_content)
+        return hashlib.md5(image_data).hexdigest()
+
+    @staticmethod
+    def compress_image_if_needed(image_path: Path, max_size_mb=2) -> Path:
+        """
+        如果图片大于max_size_mb，则压缩并覆盖原文件，返回原始路径（Path对象）
+        """
+        if hasattr(Image, "Resampling"):  # Pillow 9.1.0及以后
+            RESAMPLE = Image.Resampling.LANCZOS
+        else:
+            RESAMPLE = Image.ANTIALIAS
+
+        max_size = max_size_mb * 1024 * 1024
+        if image_path.stat().st_size <= max_size:
+            return image_path
+
+        img = Image.open(image_path)
+        suffix = image_path.suffix.lower()
+        quality = 90 if suffix in [".jpg", ".jpeg"] else None
+        step = 5
+
+        if suffix in [".jpg", ".jpeg"]:
+            while True:
+                img.save(image_path, quality=quality, optimize=True)
+                if image_path.stat().st_size <= max_size or quality <= 10:
+                    break
+                quality -= step
+        elif suffix == ".png":
+            width, height = img.size
+            while True:
+                img.save(image_path, optimize=True)
+                if (
+                    image_path.stat().st_size <= max_size
+                    or width <= 200
+                    or height <= 200
+                ):
+                    break
+                width = int(width * 0.95)
+                height = int(height * 0.95)
+                img = img.resize((width, height), RESAMPLE)
+        else:
+            raise ValueError("仅支持JPG/JPEG和PNG格式图片的压缩。")
+
+        return image_path

app/utils/ProcessManager.py

@@ -0,0 +1,157 @@
+#   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
+
+
+import psutil
+import subprocess
+from pathlib import Path
+from datetime import datetime
+
+from PySide6.QtCore import QTimer, QObject, Signal
+
+
+class ProcessManager(QObject):
+    """进程监视器类，用于跟踪主进程及其所有子进程的状态"""
+
+    processClosed = Signal()
+
+    def __init__(self):
+        super().__init__()
+
+        self.main_pid = None
+        self.tracked_pids = set()
+
+        self.check_timer = QTimer()
+        self.check_timer.timeout.connect(self.check_processes)
+
+    def open_process(self, path: Path, args: list = [], tracking_time: int = 60) -> int:
+        """
+        启动一个新进程并返回其pid，并开始监视该进程
+
+        :param path: 可执行文件的路径
+        :param args: 启动参数列表
+        :param tracking_time: 子进程追踪持续时间（秒）
+        :return: 新进程的PID
+        """
+
+        process = subprocess.Popen(
+            [path, *args],
+            cwd=path.parent,
+            creationflags=subprocess.CREATE_NO_WINDOW,
+            stdin=subprocess.DEVNULL,
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+        )
+
+        self.start_monitoring(process.pid, tracking_time)
+
+    def start_monitoring(self, pid: int, tracking_time: int = 60) -> None:
+        """
+        启动进程监视器，跟踪指定的主进程及其子进程
+
+        :param pid: 被监视进程的PID
+        :param tracking_time: 子进程追踪持续时间（秒）
+        """
+
+        self.clear()
+
+        self.main_pid = pid
+        self.tracking_time = tracking_time
+
+        # 扫描并记录所有相关进程
+        try:
+            # 获取主进程
+            main_proc = psutil.Process(self.main_pid)
+            self.tracked_pids.add(self.main_pid)
+
+            # 递归获取所有子进程
+            if tracking_time:
+                for child in main_proc.children(recursive=True):
+                    self.tracked_pids.add(child.pid)
+
+        except psutil.NoSuchProcess:
+            pass
+
+        # 启动持续追踪机制
+        self.start_time = datetime.now()
+        self.check_timer.start(100)
+
+    def check_processes(self) -> None:
+        """检查跟踪的进程是否仍在运行，并更新子进程列表"""
+
+        # 仅在时限内持续更新跟踪的进程列表，发现新的子进程
+        if (datetime.now() - self.start_time).total_seconds() < self.tracking_time:
+
+            current_pids = set(self.tracked_pids)
+            for pid in current_pids:
+                try:
+                    proc = psutil.Process(pid)
+                    for child in proc.children():
+                        if child.pid not in self.tracked_pids:
+                            # 新发现的子进程
+                            self.tracked_pids.add(child.pid)
+                except psutil.NoSuchProcess:
+                    continue
+
+        if not self.is_running():
+            self.clear()
+            self.processClosed.emit()
+
+    def is_running(self) -> bool:
+        """检查所有跟踪的进程是否还在运行"""
+
+        for pid in self.tracked_pids:
+            try:
+                proc = psutil.Process(pid)
+                if proc.is_running():
+                    return True
+            except psutil.NoSuchProcess:
+                continue
+
+        return False
+
+    def kill(self, if_force: bool = False) -> None:
+        """停止监视器并中止所有跟踪的进程"""
+
+        self.check_timer.stop()
+
+        for pid in self.tracked_pids:
+            try:
+                proc = psutil.Process(pid)
+                if if_force:
+                    kill_process = subprocess.Popen(
+                        ["taskkill", "/F", "/T", "/PID", str(pid)],
+                        creationflags=subprocess.CREATE_NO_WINDOW,
+                    )
+                    kill_process.wait()
+                proc.terminate()
+            except psutil.NoSuchProcess:
+                continue
+
+        if self.main_pid:
+            self.processClosed.emit()
+        self.clear()
+
+    def clear(self) -> None:
+        """清空跟踪的进程列表"""
+
+        self.main_pid = None
+        self.check_timer.stop()
+        self.tracked_pids.clear()

app/utils/__init__.py

@@ -25,5 +25,6 @@ __license__ = "GPL-3.0 license"
 
 
 from .logger import get_logger
+from .security import dpapi_encrypt, dpapi_decrypt
 
-__all__ = ["get_logger"]
+__all__ = ["get_logger", "dpapi_encrypt", "dpapi_decrypt"]

app/utils/security.py

@@ -0,0 +1,69 @@
+#   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
+
+
+import base64
+import win32crypt
+
+
+def dpapi_encrypt(
+    note: str, description: None | str = None, entropy: None | bytes = None
+) -> str:
+    """
+    使用Windows DPAPI加密数据
+
+    :param note: 数据明文
+    :type note: str
+    :param description: 描述信息
+    :type description: str
+    :param entropy: 随机熵
+    :type entropy: bytes
+    :return: 加密后的数据
+    :rtype: str
+    """
+
+    if note == "":
+        return ""
+
+    encrypted = win32crypt.CryptProtectData(
+        note.encode("utf-8"), description, entropy, None, None, 0
+    )
+    return base64.b64encode(encrypted).decode("utf-8")
+
+
+def dpapi_decrypt(note: str, entropy: None | bytes = None) -> str:
+    """
+    使用Windows DPAPI解密数据
+
+    :param note: 数据密文
+    :type note: str
+    :param entropy: 随机熵
+    :type entropy: bytes
+    :return: 解密后的明文
+    :rtype: str
+    """
+
+    if note == "":
+        return ""
+
+    decrypted = win32crypt.CryptUnprotectData(
+        base64.b64decode(note), entropy, None, None, 0
+    )
+    return decrypted[1].decode("utf-8")