229 lines
8.5 KiB
Python
229 lines
8.5 KiB
Python
#coding: utf-8
|
||
# +-------------------------------------------------------------------
|
||
# | YakPanel
|
||
# +-------------------------------------------------------------------
|
||
# | Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
||
# +-------------------------------------------------------------------
|
||
# | Author: 沐落 <cjx@yakpanel.com>
|
||
# | Author: lx
|
||
# | 消息通道邮箱模块
|
||
# +-------------------------------------------------------------------
|
||
|
||
import os, sys, public, base64, json, re
|
||
import sys, os
|
||
panelPath = "/www/server/panel"
|
||
os.chdir(panelPath)
|
||
sys.path.insert(0,panelPath + "/class/")
|
||
import smtplib
|
||
from email.mime.text import MIMEText
|
||
from email.utils import formataddr
|
||
|
||
class mail_msg:
|
||
__module_name = None
|
||
__default_pl = "{}/data/default_msg_channel.pl".format(panelPath)
|
||
|
||
__mail_send_conf = '/www/server/panel/data/stmp_mail.json'
|
||
__mail_receive_conf = '/www/server/panel/data/mail_list.json'
|
||
__mail_config = None
|
||
def __init__(self):
|
||
self.__mail_config = {}
|
||
if os.path.exists('data/stmp_mail.json'):
|
||
self.__mail_config['send'] = json.loads(public.readFile(self.__mail_send_conf))
|
||
|
||
if os.path.exists('data/mail_list.json'):
|
||
self.__mail_config['receive'] = json.loads(public.readFile(self.__mail_receive_conf))
|
||
|
||
if not 'send' in self.__mail_config: self.__mail_config['send'] = {}
|
||
if not 'receive' in self.__mail_config: self.__mail_config['receive'] = {}
|
||
|
||
if 'qq_mail' not in self.__mail_config['send'] or 'qq_stmp_pwd' not in self.__mail_config['send'] or 'hosts' not in self.__mail_config['send']: self.__mail_config = None
|
||
|
||
self.__module_name = self.__class__.__name__.replace('_msg','')
|
||
|
||
def __get_default_channel(self):
|
||
"""
|
||
@获取默认消息通道
|
||
"""
|
||
try:
|
||
if public.readFile(self.__default_pl) == self.__module_name:
|
||
return True
|
||
except:pass
|
||
return False
|
||
|
||
def get_version_info(self,get):
|
||
"""
|
||
获取版本信息
|
||
"""
|
||
data = {}
|
||
data['ps'] = 'Email is used to receive panel message push'
|
||
data['version'] = '1.1'
|
||
data['date'] = '2022-08-10'
|
||
data['author'] = 'YakPanel'
|
||
data['title'] = 'Email'
|
||
data['help'] = 'http://www.yakpanel.com'
|
||
return data
|
||
|
||
def get_config(self,get):
|
||
"""
|
||
获取QQ邮箱配置
|
||
"""
|
||
data = {}
|
||
if self.__mail_config:
|
||
data = self.__mail_config
|
||
if "send" in data:
|
||
if not os.path.exists(self.__mail_send_conf):
|
||
public.writeFile(self.__mail_send_conf,json.dumps(data["send"]))
|
||
if "receive" in data:
|
||
if not os.path.exists(self.__mail_receive_conf):
|
||
public.writeFile(self.__mail_receive_conf, json.dumps(data["receive"]))
|
||
|
||
data['default'] = self.__get_default_channel()
|
||
return data
|
||
|
||
def set_config(self,get):
|
||
"""
|
||
设置邮箱配置
|
||
@send 带有此参数表示设置发送者设置
|
||
@qq_mail 发送者邮箱
|
||
@qq_stmp_pwd 发送者密码
|
||
@hosts 发送stmp
|
||
@发送端口port
|
||
|
||
@mails 接收者配置，一行一个（如：111@qq.com\n222@qq.com）
|
||
"""
|
||
|
||
if not self.__mail_config:
|
||
self.__mail_config = {'send':{},'receive':[]}
|
||
|
||
if hasattr(get, 'send'):
|
||
if not hasattr(get, 'qq_mail') or not hasattr(get, 'qq_stmp_pwd') or not hasattr(get, 'hosts') or not hasattr(get, 'port'): return public.returnMsg(False, public.lang("Please fill in the complete information"))
|
||
mail_config = {
|
||
"qq_mail": get.qq_mail.strip(),
|
||
"qq_stmp_pwd": get.qq_stmp_pwd.strip(),
|
||
"hosts": get.hosts.strip(),
|
||
"port": get.port
|
||
}
|
||
self.__mail_config['send'] = mail_config
|
||
|
||
else:
|
||
mails = ''
|
||
if hasattr(get, 'mails'):
|
||
mails = get.mails.strip()
|
||
|
||
arrs = []
|
||
for mail in mails.split('\n'):
|
||
if not mail.strip(): continue
|
||
arrs.append(mail.strip())
|
||
|
||
self.__mail_config['receive'] = arrs
|
||
|
||
#首次配置同步接收者配置
|
||
receive_list = self.__mail_config['receive']
|
||
mail_address = self.__mail_config['send']['qq_mail']
|
||
if not mail_address in receive_list and len(receive_list) == 0:
|
||
if not 'receive' in self.__mail_config:
|
||
self.__mail_config['receive'] = []
|
||
self.__mail_config['receive'].append(mail_address)
|
||
|
||
ret = self.send_msg("YakPanel 测试邮件")
|
||
if ret['status']:
|
||
|
||
if ret['success'] <= 0:
|
||
return public.returnMsg(False, public.lang("Sending failed, please check whether the sender configuration or receiver information is correct."))
|
||
|
||
public.writeFile(self.__mail_send_conf, json.dumps(self.__mail_config['send']))
|
||
public.writeFile(self.__mail_receive_conf,json.dumps(self.__mail_config['receive']))
|
||
|
||
return public.returnMsg(True, public.lang("successfully set"))
|
||
else:
|
||
return ret
|
||
|
||
|
||
def get_send_msg(self,msg):
|
||
"""
|
||
@name 处理md格式
|
||
"""
|
||
try:
|
||
title = None
|
||
if msg.find("####") >= 0:
|
||
try:
|
||
title = re.search(r"####(.+)\n", msg).groups()[0]
|
||
except:pass
|
||
msg = msg.replace("\n\n","<br>").strip()
|
||
|
||
except:pass
|
||
return msg,title
|
||
|
||
def send_msg(self,msg , title = 'YakPanel panel message push',to_email = None):
|
||
"""
|
||
邮箱发送
|
||
@msg 消息正文
|
||
@title 消息标题
|
||
@to_email 发送给谁，默认发送所有人
|
||
"""
|
||
if not self.__mail_config :
|
||
return public.returnMsg(False, public.lang("Email information is not configured correctly."))
|
||
|
||
if not 'port' in self.__mail_config['send']: self.__mail_config['send']['port'] = 465
|
||
|
||
receive_list = []
|
||
if to_email:
|
||
for x in to_email.split(','):
|
||
receive_list.append(x)
|
||
else:
|
||
receive_list = self.__mail_config['receive']
|
||
|
||
res = {}
|
||
ret_msg = {}
|
||
error ,sucess,total = 0,0,0
|
||
msg,n_title = self.get_send_msg(msg)
|
||
if n_title: title = n_title
|
||
|
||
for email in receive_list:
|
||
if not email: continue
|
||
|
||
try:
|
||
data = MIMEText(msg, 'html', 'utf-8')
|
||
data['From'] = formataddr([self.__mail_config['send']['qq_mail'], self.__mail_config['send']['qq_mail']])
|
||
data['To'] = formataddr([self.__mail_config['send']['qq_mail'], email.strip()])
|
||
data['Subject'] = title
|
||
if int(self.__mail_config['send']['port']) == 465:
|
||
server = smtplib.SMTP_SSL(str(self.__mail_config['send']['hosts']), str(self.__mail_config['send']['port']))
|
||
else:
|
||
server = smtplib.SMTP(str(self.__mail_config['send']['hosts']), str(self.__mail_config['send']['port']))
|
||
server.login(self.__mail_config['send']['qq_mail'], self.__mail_config['send']['qq_stmp_pwd'])
|
||
server.sendmail(self.__mail_config['send']['qq_mail'], [email.strip(), ], data.as_string())
|
||
server.quit()
|
||
sucess += 1
|
||
res[email] = 1
|
||
except :
|
||
error += 1
|
||
res[email] = 0
|
||
ret_msg[email] = public.get_error_info()
|
||
total += 1
|
||
|
||
try:
|
||
if res:
|
||
public.write_push_log(self.__module_name,title,res)
|
||
else:
|
||
public.WriteLog('Alarm notification','[Email] No sender configured, please configure first.')
|
||
except:pass
|
||
|
||
result = public.returnMsg(True,'The sending is complete, a total of [{}] items were sent, [{}] succeeded, and [{}] failed.'.format(total,sucess,error))
|
||
result['list'] = ret_msg
|
||
result['success'] = sucess
|
||
result['error'] = error
|
||
return result
|
||
|
||
def push_data(self,data):
|
||
to_email = data.get("to_email", None)
|
||
if 'to_user' in data:
|
||
to_email = data['to_user']
|
||
|
||
return self.send_msg(data['msg'],data['title'],to_email)
|
||
|
||
def uninstall(self):
|
||
if os.path.exists(self.__mail_send_conf):
|
||
os.remove(self.__mail_send_conf)
|
||
if os.path.exists(self.__mail_receive_conf):
|
||
os.remove(self.__mail_receive_conf) |