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

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")