feat(notification): 增加用户单独通知功能

app/services/notification.py

@@ -69,219 +69,208 @@ class Notification(QWidget):
 
     def send_mail(self, mode, title, content, to_address) -> 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,}$",
-                        to_address,
-                    )
-                )
-            ):
-                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(),
-                        to_address,
-                    )
-                )  # 收件人显示的名字
-                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(
+        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),
-                    Crypto.win_decryptor(Config.get(Config.notify_AuthorizationCode)),
                 )
-                smtpObj.sendmail(
-                    Config.get(Config.notify_FromAddress),
+            )
+            or not bool(
+                re.match(
+                    r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$",
                     to_address,
-                    message.as_string(),
                 )
-                smtpObj.quit()
-                logger.success("邮件发送成功")
-                return None
-            except Exception as e:
-                logger.error(f"发送邮件时出错：\n{e}")
-                self.push_info_bar.emit("error", "发送邮件时出错", f"{e}", -1)
-                return None
+            )
+        ):
+            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(),
+                    to_address,
+                )
+            )  # 收件人显示的名字
+            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),
+                to_address,
+                message.as_string(),
+            )
+            smtpObj.quit()
+            logger.success("邮件发送成功")
+            return None
+        except Exception as e:
+            logger.error(f"发送邮件时出错：\n{e}")
+            self.push_info_bar.emit("error", "发送邮件时出错", f"{e}", -1)
+            return None
         return None
 
     def ServerChanPush(self, title, content, send_key, tag, channel):
         """使用Server酱推送通知"""
-        if Config.get(Config.notify_IfServerChan):
-            if not send_key:
-                logger.error("请正确设置Server酱的SendKey")
-                self.push_info_bar.emit(
-                    "error", "Server酱通知推送异常", "请正确设置Server酱的SendKey", -1
-                )
-                return None
+        if not send_key:
+            logger.error("请正确设置Server酱的SendKey")
+            self.push_info_bar.emit(
+                "error", "Server酱通知推送异常", "请正确设置Server酱的SendKey", -1
+            )
+            return None
 
-            try:
-                # 构造 URL
-                if send_key.startswith("sctp"):
-                    match = re.match(r"^sctp(\d+)t", send_key)
-                    if match:
-                        url = f"https://{match.group(1)}.push.ft07.com/send/{send_key}.send"
-                    else:
-                        raise ValueError("SendKey 格式错误（sctp）")
+        try:
+            # 构造 URL
+            if send_key.startswith("sctp"):
+                match = re.match(r"^sctp(\d+)t", send_key)
+                if match:
+                    url = f"https://{match.group(1)}.push.ft07.com/send/{send_key}.send"
                 else:
-                    url = f"https://sctapi.ftqq.com/{send_key}.send"
-
-                # 构建 tags 和 channel
-                def is_valid(s):
-                    return s == "" or (
-                        s == "|".join(s.split("|"))
-                        and (s.count("|") == 0 or all(s.split("|")))
-                    )
-
-                tags = "|".join(
-                    _.strip()
-                    for _ in tag.split("|")
-                )
-                channels = "|".join(
-                    _.strip()
-                    for _ in channel.split("|")
-                )
-
-                options = {}
-                if is_valid(tags):
-                    options["tags"] = tags
-                else:
-                    logger.warning("Server酱 Tag 配置不正确，将被忽略")
-                    self.push_info_bar.emit(
-                        "warning",
-                        "Server酱通知推送异常",
-                        "请正确设置 ServerChan 的 Tag",
-                        -1,
-                    )
-
-                if is_valid(channels):
-                    options["channel"] = channels
-                else:
-                    logger.warning("Server酱 Channel 配置不正确，将被忽略")
-                    self.push_info_bar.emit(
-                        "warning",
-                        "Server酱通知推送异常",
-                        "请正确设置 ServerChan 的 Channel",
-                        -1,
-                    )
-
-                # 请求发送
-                params = {"title": title, "desp": content, **options}
-                headers = {"Content-Type": "application/json;charset=utf-8"}
-
-                response = requests.post(url, json=params, headers=headers, timeout=10)
-                result = response.json()
-
-                if result.get("code") == 0:
-                    logger.info("Server酱推送通知成功")
-                    return True
-                else:
-                    error_code = result.get("code", "-1")
-                    logger.error(f"Server酱通知推送失败：响应码：{error_code}")
-                    self.push_info_bar.emit(
-                        "error", "Server酱通知推送失败", f"响应码：{error_code}", -1
-                    )
-                    return f"Server酱通知推送失败：{error_code}"
-
-            except Exception as e:
-                logger.exception("Server酱通知推送异常")
-                self.push_info_bar.emit(
-                    "error", "Server酱通知推送异常", f"请检查相关设置，如还有问题可联系开发者", -1
-                )
-                return f"Server酱通知推送异常：{str(e)}"
-        return None
-
-    def CompanyWebHookBotPush(self, title, content,webhook_url):
-        """使用企业微信群机器人推送通知"""
-        if Config.get(Config.notify_IfCompanyWebHookBot):
-
-            if webhook_url == "":
-                logger.error("请正确设置企业微信群机器人的WebHook地址")
-                self.push_info_bar.emit(
-                    "error",
-                    "企业微信群机器人通知推送异常",
-                    "请正确设置企业微信群机器人的WebHook地址",
-                    -1,
-                )
-                return None
-
-            content = f"{title}\n{content}"
-            data = {"msgtype": "text", "text": {"content": content}}
-
-            for _ in range(3):
-                try:
-                    response = requests.post(
-                        url=webhook_url,
-                        json=data,
-                        timeout=10,
-                    )
-                    info = response.json()
-                    break
-                except Exception as e:
-                    err = e
-                    time.sleep(0.1)
+                    raise ValueError("SendKey 格式错误（sctp）")
             else:
-                logger.error(f"推送企业微信群机器人时出错：{err}")
+                url = f"https://sctapi.ftqq.com/{send_key}.send"
+
+            # 构建 tags 和 channel
+            def is_valid(s):
+                return s == "" or (
+                    s == "|".join(s.split("|"))
+                    and (s.count("|") == 0 or all(s.split("|")))
+                )
+
+            tags = "|".join(_.strip() for _ in tag.split("|"))
+            channels = "|".join(_.strip() for _ in channel.split("|"))
+
+            options = {}
+            if is_valid(tags):
+                options["tags"] = tags
+            else:
+                logger.warning("Server酱 Tag 配置不正确，将被忽略")
                 self.push_info_bar.emit(
-                    "error",
-                    "企业微信群机器人通知推送失败",
-                    f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}',
+                    "warning",
+                    "Server酱通知推送异常",
+                    "请正确设置 ServerChan 的 Tag",
                     -1,
                 )
-                return None
 
-            if info["errcode"] == 0:
-                logger.info("企业微信群机器人推送通知成功")
+            if is_valid(channels):
+                options["channel"] = channels
+            else:
+                logger.warning("Server酱 Channel 配置不正确，将被忽略")
+                self.push_info_bar.emit(
+                    "warning",
+                    "Server酱通知推送异常",
+                    "请正确设置 ServerChan 的 Channel",
+                    -1,
+                )
+
+            # 请求发送
+            params = {"title": title, "desp": content, **options}
+            headers = {"Content-Type": "application/json;charset=utf-8"}
+
+            response = requests.post(url, json=params, headers=headers, timeout=10)
+            result = response.json()
+
+            if result.get("code") == 0:
+                logger.info("Server酱推送通知成功")
                 return True
             else:
-                logger.error(f"企业微信群机器人推送通知失败：{info}")
+                error_code = result.get("code", "-1")
+                logger.error(f"Server酱通知推送失败：响应码：{error_code}")
                 self.push_info_bar.emit(
-                    "error",
-                    "企业微信群机器人通知推送失败",
-                    f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}',
-                    -1,
+                    "error", "Server酱通知推送失败", f"响应码：{error_code}", -1
                 )
-                return f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}'
-        return None
+                return f"Server酱通知推送失败：{error_code}"
+
+        except Exception as e:
+            logger.exception("Server酱通知推送异常")
+            self.push_info_bar.emit(
+                "error",
+                "Server酱通知推送异常",
+                "请检查相关设置和网络连接。如全部配置正确，请稍后再试。",
+                -1,
+            )
+            return f"Server酱通知推送异常：{str(e)}"
+
+    def CompanyWebHookBotPush(self, title, content, webhook_url):
+        """使用企业微信群机器人推送通知"""
+        if webhook_url == "":
+            logger.error("请正确设置企业微信群机器人的WebHook地址")
+            self.push_info_bar.emit(
+                "error",
+                "企业微信群机器人通知推送异常",
+                "请正确设置企业微信群机器人的WebHook地址",
+                -1,
+            )
+            return None
+
+        content = f"{title}\n{content}"
+        data = {"msgtype": "text", "text": {"content": content}}
+
+        for _ in range(3):
+            try:
+                response = requests.post(
+                    url=webhook_url,
+                    json=data,
+                    timeout=10,
+                )
+                info = response.json()
+                break
+            except Exception as e:
+                err = e
+                time.sleep(0.1)
+        else:
+            logger.error(f"推送企业微信群机器人时出错：{err}")
+            self.push_info_bar.emit(
+                "error",
+                "企业微信群机器人通知推送失败",
+                f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}',
+                -1,
+            )
+            return None
+
+        if info["errcode"] == 0:
+            logger.info("企业微信群机器人推送通知成功")
+            return True
+        else:
+            logger.error(f"企业微信群机器人推送通知失败：{info}")
+            self.push_info_bar.emit(
+                "error",
+                "企业微信群机器人通知推送失败",
+                f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}',
+                -1,
+            )
+            return f'使用企业微信群机器人推送通知时出错：{info["errmsg"]}'
 
     def send_test_notification(self):
         """发送测试通知到所有已启用的通知渠道"""