161 lines
6.0 KiB
Python
161 lines
6.0 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
|
||
"""
|
||
import requests
|
||
from loguru import logger
|
||
from plyer import notification
|
||
import smtplib
|
||
from email.mime.text import MIMEText
|
||
from email.header import Header
|
||
from email.utils import formataddr
|
||
|
||
from serverchan_sdk import sc_send
|
||
|
||
from app.core import Config
|
||
|
||
|
||
class Notification:
|
||
|
||
def push_notification(self, title, message, ticker, t):
|
||
"""推送系统通知"""
|
||
|
||
if Config.global_config.get(Config.global_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, title, content):
|
||
"""使用官方专用邮箱推送邮件通知"""
|
||
|
||
# 声明：此邮箱为AUTO_MAA项目组资产，未经授权不得私自使用
|
||
# 注意：此声明注释只有使用者更换发信邮箱时才能删除，本条规则优先级高于GPLv3
|
||
|
||
if Config.global_config.get(Config.global_config.notify_IfSendMail):
|
||
|
||
# 第三方 SMTP 服务配置
|
||
mail_host = "smtp.163.com" # 设置服务器
|
||
mail_sender = "AUTO_MAA_server@163.com" # 用户名
|
||
mail_key = "SYrq87nDLD4RNB5T" # 授权码 24/11/15
|
||
|
||
# 定义邮件正文
|
||
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(),
|
||
Config.global_config.get(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,
|
||
Config.global_config.get(Config.global_config.notify_MailAddress),
|
||
message.as_string(),
|
||
)
|
||
return True
|
||
except smtplib.SMTPException as e:
|
||
return f"发送邮件时出错：\n{e}"
|
||
finally:
|
||
smtpObj.quit()
|
||
|
||
def ServerChanPush(self, title, content):
|
||
"""使用Server酱推送通知"""
|
||
|
||
if Config.global_config.get(Config.global_config.notify_IfServerChan):
|
||
send_key = Config.global_config.get(Config.global_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 = Config.global_config.get(Config.global_config.notify_ServerChanTag)
|
||
send_channel = Config.global_config.get(Config.global_config.notify_ServerChanChannel)
|
||
|
||
if is_valid(send_tag):
|
||
option['tags'] = send_tag
|
||
else:
|
||
option['tags'] = ''
|
||
logger.warning('请正确设置Auto_MAA中ServerChan的Tag。')
|
||
|
||
if is_valid(send_channel):
|
||
option['channel'] = send_channel
|
||
else:
|
||
option['channel'] = ''
|
||
logger.warning('请正确设置Auto_MAA中ServerChan的Channel。')
|
||
|
||
response = sc_send(send_key, title, content, option)
|
||
if response["code"] == 0:
|
||
logger.info("Server酱推送通知成功")
|
||
return True
|
||
else:
|
||
logger.info("Server酱推送通知失败")
|
||
logger.error(response)
|
||
return f'使用Server酱推送通知时出错：\n{response["data"]['error']}'
|
||
|
||
def CompanyWebHookBotPush(self, title, content):
|
||
"""使用企业微信群机器人推送通知"""
|
||
if Config.global_config.get(Config.global_config.notify_IfCompanyWebHookBot):
|
||
content = f'{title}\n{content}'
|
||
data = {
|
||
"msgtype": "text",
|
||
"text": {
|
||
"content": content
|
||
}
|
||
}
|
||
response = requests.post(
|
||
url=Config.global_config.get(Config.global_config.notify_CompanyWebHookBotUrl),
|
||
json=data
|
||
)
|
||
if response.json()["errcode"] == 0:
|
||
logger.info("企业微信群机器人推送通知成功")
|
||
return True
|
||
else:
|
||
logger.info("企业微信群机器人推送通知失败")
|
||
logger.error(response.json())
|
||
return f'使用企业微信群机器人推送通知时出错：\n{response.json()["errmsg"]}'
|
||
|
||
|
||
Notify = Notification()
|