38a04fc4b2
commit 8724c545a8af8f34565aa71620e66cbd71547f37 Author: DLmaster <DLmaster_361@163.com> Date: Fri Apr 11 18:08:28 2025 +0800 feat(core): 预接入mirrorc commit d57ebaa281ff7c418aa8f11fe8e8ba260d8dbeca Author: DLmaster <DLmaster_361@163.com> Date: Thu Apr 10 12:37:26 2025 +0800 chore(core): 基础配置相关内容重构 - 添加理智药设置选项 #34 - 输入对话框添加回车键确认能力 #35 - 用户列表UI改版升级 - 配置类取消单例限制 - 配置读取方式与界面渲染方法优化 commit 710542287d04719c8443b91acb227de1dccc20d0 Author: DLmaster <DLmaster_361@163.com> Date: Fri Mar 28 23:32:17 2025 +0800 chore(core): search相关结构重整 commit 8009c69236655e29119ce62ff53a0360abaed2af Merge:648f42b9f88f92Author: DLmaster <DLmaster_361@163.com> Date: Mon Mar 24 15:31:40 2025 +0800 Merge branch 'dev' into user_list_dev
288 lines
11 KiB
Python
288 lines
11 KiB
Python
# <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
|
||
"""
|
||
|
||
from PySide6.QtWidgets import QWidget
|
||
from PySide6.QtCore import Signal
|
||
import requests
|
||
from loguru import logger
|
||
from plyer import notification
|
||
import re
|
||
import smtplib
|
||
from email.mime.text import MIMEText
|
||
from email.mime.multipart import MIMEMultipart
|
||
from email.header import Header
|
||
from email.utils import formataddr
|
||
|
||
from serverchan_sdk import sc_send
|
||
|
||
from app.core import Config
|
||
from app.services.security import Crypto
|
||
|
||
|
||
class Notification(QWidget):
|
||
|
||
push_info_bar = Signal(str, str, str, int)
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
|
||
def push_plyer(self, title, message, ticker, t):
|
||
"""推送系统通知"""
|
||
|
||
if Config.get(Config.notify_IfPushPlyer):
|
||
|
||
notification.notify(
|
||
title=title,
|
||
message=message,
|
||
app_name="AUTO_MAA",
|
||
app_icon=str(Config.app_path / "resources/icons/AUTO_MAA.ico"),
|
||
timeout=t,
|
||
ticker=ticker,
|
||
toast=True,
|
||
)
|
||
|
||
return True
|
||
|
||
def send_mail(self, mode, title, content) -> None:
|
||
"""推送邮件通知"""
|
||
|
||
if Config.get(Config.notify_IfSendMail):
|
||
|
||
if (
|
||
Config.get(Config.notify_SMTPServerAddress) == ""
|
||
or Config.get(Config.notify_AuthorizationCode) == ""
|
||
or not bool(
|
||
re.match(
|
||
r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$",
|
||
Config.get(Config.notify_FromAddress),
|
||
)
|
||
)
|
||
or not bool(
|
||
re.match(
|
||
r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$",
|
||
Config.get(Config.notify_ToAddress),
|
||
)
|
||
)
|
||
):
|
||
logger.error(
|
||
"请正确设置邮件通知的SMTP服务器地址、授权码、发件人地址和收件人地址"
|
||
)
|
||
self.push_info_bar.emit(
|
||
"error",
|
||
"邮件通知推送异常",
|
||
"请正确设置邮件通知的SMTP服务器地址、授权码、发件人地址和收件人地址",
|
||
-1,
|
||
)
|
||
return None
|
||
|
||
try:
|
||
# 定义邮件正文
|
||
if mode == "文本":
|
||
message = MIMEText(content, "plain", "utf-8")
|
||
elif mode == "网页":
|
||
message = MIMEMultipart("alternative")
|
||
message["From"] = formataddr(
|
||
(
|
||
Header("AUTO_MAA通知服务", "utf-8").encode(),
|
||
Config.get(Config.notify_FromAddress),
|
||
)
|
||
) # 发件人显示的名字
|
||
message["To"] = formataddr(
|
||
(
|
||
Header("AUTO_MAA用户", "utf-8").encode(),
|
||
Config.get(Config.notify_ToAddress),
|
||
)
|
||
) # 收件人显示的名字
|
||
message["Subject"] = Header(title, "utf-8")
|
||
|
||
if mode == "网页":
|
||
message.attach(MIMEText(content, "html", "utf-8"))
|
||
|
||
smtpObj = smtplib.SMTP_SSL(
|
||
Config.get(Config.notify_SMTPServerAddress),
|
||
465,
|
||
)
|
||
smtpObj.login(
|
||
Config.get(Config.notify_FromAddress),
|
||
Crypto.win_decryptor(Config.get(Config.notify_AuthorizationCode)),
|
||
)
|
||
smtpObj.sendmail(
|
||
Config.get(Config.notify_FromAddress),
|
||
Config.get(Config.notify_ToAddress),
|
||
message.as_string(),
|
||
)
|
||
smtpObj.quit()
|
||
logger.success("邮件发送成功")
|
||
except Exception as e:
|
||
logger.error(f"发送邮件时出错：\n{e}")
|
||
self.push_info_bar.emit("error", "发送邮件时出错", f"{e}", -1)
|
||
|
||
def ServerChanPush(self, title, content):
|
||
"""使用Server酱推送通知"""
|
||
|
||
if Config.get(Config.notify_IfServerChan):
|
||
|
||
if Config.get(Config.notify_ServerChanKey) == "":
|
||
logger.error("请正确设置Server酱的SendKey")
|
||
self.push_info_bar.emit(
|
||
"error",
|
||
"Server酱通知推送异常",
|
||
"请正确设置Server酱的SendKey",
|
||
-1,
|
||
)
|
||
return None
|
||
else:
|
||
send_key = Config.get(Config.notify_ServerChanKey)
|
||
|
||
option = {}
|
||
is_valid = lambda s: s == "" or (
|
||
s == "|".join(s.split("|")) and (s.count("|") == 0 or all(s.split("|")))
|
||
)
|
||
"""
|
||
is_valid => True, 如果启用的话需要正确设置Tag和Channel。
|
||
允许空的Tag和Channel即不启用，但不允许例如a||b，|a|b，a|b|，||||
|
||
"""
|
||
send_tag = "|".join(
|
||
_.strip() for _ in Config.get(Config.notify_ServerChanTag).split("|")
|
||
)
|
||
send_channel = "|".join(
|
||
_.strip()
|
||
for _ in Config.get(Config.notify_ServerChanChannel).split("|")
|
||
)
|
||
|
||
if is_valid(send_tag):
|
||
option["tags"] = send_tag
|
||
else:
|
||
option["tags"] = ""
|
||
logger.warning("请正确设置Auto_MAA中ServerChan的Tag。")
|
||
self.push_info_bar.emit(
|
||
"warning",
|
||
"Server酱通知推送异常",
|
||
"请正确设置Auto_MAA中ServerChan的Tag。",
|
||
-1,
|
||
)
|
||
|
||
if is_valid(send_channel):
|
||
option["channel"] = send_channel
|
||
else:
|
||
option["channel"] = ""
|
||
logger.warning("请正确设置Auto_MAA中ServerChan的Channel。")
|
||
self.push_info_bar.emit(
|
||
"warning",
|
||
"Server酱通知推送异常",
|
||
"请正确设置Auto_MAA中ServerChan的Channel。",
|
||
-1,
|
||
)
|
||
|
||
response = sc_send(send_key, title, content, option)
|
||
if response["code"] == 0:
|
||
logger.info("Server酱推送通知成功")
|
||
return True
|
||
else:
|
||
logger.info("Server酱推送通知失败")
|
||
logger.error(response)
|
||
self.push_info_bar.emit(
|
||
"error",
|
||
"Server酱通知推送失败",
|
||
f'使用Server酱推送通知时出错：\n{response["data"]['error']}',
|
||
-1,
|
||
)
|
||
return f'使用Server酱推送通知时出错：\n{response["data"]['error']}'
|
||
|
||
def CompanyWebHookBotPush(self, title, content):
|
||
"""使用企业微信群机器人推送通知"""
|
||
if Config.get(Config.notify_IfCompanyWebHookBot):
|
||
|
||
if Config.get(Config.notify_CompanyWebHookBotUrl) == "":
|
||
logger.error("请正确设置企业微信群机器人的WebHook地址")
|
||
self.push_info_bar.emit(
|
||
"error",
|
||
"企业微信群机器人通知推送异常",
|
||
"请正确设置企业微信群机器人的WebHook地址",
|
||
-1,
|
||
)
|
||
return None
|
||
|
||
content = f"{title}\n{content}"
|
||
data = {"msgtype": "text", "text": {"content": content}}
|
||
response = requests.post(
|
||
url=Config.get(Config.notify_CompanyWebHookBotUrl),
|
||
json=data,
|
||
)
|
||
if response.json()["errcode"] == 0:
|
||
logger.info("企业微信群机器人推送通知成功")
|
||
return True
|
||
else:
|
||
logger.info("企业微信群机器人推送通知失败")
|
||
logger.error(response.json())
|
||
self.push_info_bar.emit(
|
||
"error",
|
||
"企业微信群机器人通知推送失败",
|
||
f'使用企业微信群机器人推送通知时出错：\n{response.json()["errmsg"]}',
|
||
-1,
|
||
)
|
||
return (
|
||
f'使用企业微信群机器人推送通知时出错：\n{response.json()["errmsg"]}'
|
||
)
|
||
|
||
def send_test_notification(self):
|
||
"""发送测试通知到所有已启用的通知渠道"""
|
||
# 发送系统通知
|
||
self.push_plyer(
|
||
"测试通知",
|
||
"这是 AUTO_MAA 外部通知测试信息。如果你看到了这段内容，说明 AUTO_MAA 的通知功能已经正确配置且可以正常工作！",
|
||
"测试通知",
|
||
3,
|
||
)
|
||
|
||
# 发送邮件通知
|
||
if Config.get(Config.notify_IfSendMail):
|
||
self.send_mail(
|
||
"文本",
|
||
"AUTO_MAA测试通知",
|
||
"这是 AUTO_MAA 外部通知测试信息。如果你看到了这段内容，说明 AUTO_MAA 的通知功能已经正确配置且可以正常工作！",
|
||
)
|
||
|
||
# 发送Server酱通知
|
||
if Config.get(Config.notify_IfServerChan):
|
||
self.ServerChanPush(
|
||
"AUTO_MAA测试通知",
|
||
"这是 AUTO_MAA 外部通知测试信息。如果你看到了这段内容，说明 AUTO_MAA 的通知功能已经正确配置且可以正常工作！",
|
||
)
|
||
|
||
# 发送企业微信机器人通知
|
||
if Config.get(Config.notify_IfCompanyWebHookBot):
|
||
self.CompanyWebHookBotPush(
|
||
"AUTO_MAA测试通知",
|
||
"这是 AUTO_MAA 外部通知测试信息。如果你看到了这段内容，说明 AUTO_MAA 的通知功能已经正确配置且可以正常工作！",
|
||
)
|
||
|
||
return True
|
||
|
||
|
||
Notify = Notification()
|