Initial YakPanel commit

mod/base/msg/mail_msg.py

@@ -0,0 +1,158 @@
+#coding: utf-8
+# +-------------------------------------------------------------------
+# | yakpanel
+# +-------------------------------------------------------------------
+# | Copyright (c) 2015-2020 yakpanel(http://www.yakpanel.com) All rights reserved.
+# +-------------------------------------------------------------------
+# | Author: 沐落 <cjx@yakpanel.com>
+# | Author: lx
+# | 消息通道邮箱模块
+# +-------------------------------------------------------------------
+
+import smtplib
+import traceback
+from email.mime.text import MIMEText
+from email.utils import formataddr
+from typing import Tuple, Union, Optional
+import public
+
+from mod.base.msg.util import write_push_log, write_mail_push_log, get_test_msg
+
+
+class MailMsg:
+
+    def __init__(self, mail_data):
+        self.id = mail_data["id"]
+        self.config = mail_data["data"]
+
+    @classmethod
+    def check_args(cls, args: dict) -> Tuple[bool, Union[dict, str]]:
+        if "send" not in args or "receive" not in args or len(args["receive"]) < 1:
+            return False, "Incomplete information, there must be a sender and at least one receiver"
+
+        if "title" not in args:
+            return False, "There is no necessary remark information"
+
+        title = args["title"]
+        if len(title) > 15:
+            return False, 'Note names cannot be longer than 15 characters'
+
+        send_data = args["send"]
+        send = {}
+        for i in ("qq_mail", "qq_stmp_pwd", "hosts", "port"):
+            if i not in send_data:
+                return False, "The sender configuration information is incomplete"
+            send[i] = send_data[i].strip()
+
+        receive_data = args["receive"]
+        if isinstance(receive_data, str):
+            receive_list = [i.strip() for i in receive_data.split("\n") if i.strip()]
+        else:
+            receive_list = [i.strip() for i in receive_data if i.strip()]
+
+        data = {
+            "send": send,
+            "title": title,
+            "receive": receive_list,
+        }
+
+        test_obj = cls({"data": data, "id": None})
+        test_msg = {
+            "msg_list": ['>configuration state: Success<br>']
+        }
+
+        test_task = get_test_msg("Message channel configuration reminders")
+
+        res = test_obj.send_msg(
+            test_task.to_mail_msg(test_msg, test_task.the_push_public_data()),
+            "Message channel configuration reminders"
+        )
+        if res is None or res.find("Failed to send mail to some recipients") != -1:
+            return True, data
+
+        return False, res
+
+    def send_msg(self, msg: str, title: str):
+        """
+        邮箱发送
+        @msg 消息正文
+        @title 消息标题
+        """
+        if not self.config:
+            return public.lang('Mailbox information is not configured correctly')
+
+        if 'port' not in self.config['send']: 
+            self.config['send']['port'] = 465
+
+        receive_list = self.config['receive']
+
+        error_list, success_list = [], []
+        error_msg_dict = {}
+        for email in receive_list:
+            if not email.strip():
+                continue
+            server = None
+            try:
+                data = MIMEText(msg, 'html', 'utf-8')
+                data['From'] = formataddr((self.config['send']['qq_mail'], self.config['send']['qq_mail']))
+                data['To'] = formataddr((self.config['send']['qq_mail'], email.strip()))
+                data['Subject'] = title
+
+                port = int(self.config['send']['port'])
+                host = str(self.config['send']['hosts'])
+                user = self.config['send']['qq_mail']
+                pwd = self.config['send']['qq_stmp_pwd']
+
+                if port == 465:
+                    # SSL direct connection
+                    server = smtplib.SMTP_SSL(host, port, timeout=10)
+                else:
+                    # Standard connection, possibly with STARTTLS
+                    server = smtplib.SMTP(host, port, timeout=10)
+                    try:
+                        # Attempt to upgrade to a secure connection
+                        server.starttls()
+                    except smtplib.SMTPNotSupportedError:
+                        # The server does not support STARTTLS, proceed with an insecure connection
+                        pass
+
+                server.login(user, pwd)
+                server.sendmail(user, [email.strip(), ], data.as_string())
+                success_list.append(email)
+            except:
+                err_msg = traceback.format_exc()
+                # public.print_log(f"邮件发送失败: {err_msg}")
+                error_list.append(email)
+                error_msg_dict[email] = err_msg
+            finally:
+                if server:
+                    server.quit()
+
+        if not error_list and not success_list:
+            return public.lang('The receiving mailbox is not configured')
+        if not error_list:
+            write_push_log("mail", True, title, success_list)
+            return None
+        if not success_list:
+            write_push_log("mail", False, title, error_list)
+            first_error_msg = list(error_msg_dict.values())[0]
+            # 修复 IndexError
+            return public.lang('Failed to send message, Recipient of failed to send:{}, Error: {}', error_list, first_error_msg)
+
+        write_mail_push_log(title, error_list, success_list)
+
+        return public.lang('Failed to send mail to some recipients, including:{}',error_list)
+
+    def test_send_msg(self) -> Optional[str]:
+        test_msg = {
+            "msg_list": ['>configuration state: <font color=#20a53a> Success </font>\n\n']
+        }
+        test_task = get_test_msg("Message channel configuration reminders")
+        res = self.send_msg(
+            test_task.to_mail_msg(test_msg, test_task.the_push_public_data()),
+            "Message channel configuration reminders"
+        )
+        if res is None:
+            return None
+        return res
+