mirror of https://github.com/jumpserver/jumpserver
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from smtplib import SMTPSenderRefused
|
|
|
|
from django.conf import settings
|
|
from django.core.mail import send_mail
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework.views import Response, APIView
|
|
|
|
from common.utils import get_logger
|
|
from common.tasks import get_email_connection as get_connection
|
|
from .. import serializers
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
__all__ = ['MailTestingAPI']
|
|
|
|
|
|
class MailTestingAPI(APIView):
|
|
serializer_class = serializers.MailTestSerializer
|
|
success_message = _("Test mail sent to {}, please check")
|
|
rbac_perms = {
|
|
'POST': 'settings.change_email'
|
|
}
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
# 测试邮件时，邮件服务器信息从配置中获取
|
|
email_host = settings.EMAIL_HOST
|
|
email_port = settings.EMAIL_PORT
|
|
email_host_user = settings.EMAIL_HOST_USER
|
|
email_host_password = settings.EMAIL_HOST_PASSWORD
|
|
email_from = serializer.validated_data.get('EMAIL_FROM')
|
|
email_use_ssl = settings.EMAIL_USE_SSL
|
|
email_use_tls = settings.EMAIL_USE_TLS
|
|
email_recipient = serializer.validated_data.get('EMAIL_RECIPIENT')
|
|
|
|
# 设置 settings 的值，会导致动态配置在当前进程失效
|
|
# for k, v in serializer.validated_data.items():
|
|
# if k.startswith('EMAIL'):
|
|
# setattr(settings, k, v)
|
|
try:
|
|
subject = settings.EMAIL_SUBJECT_PREFIX or '' + "Test"
|
|
message = "Test smtp setting"
|
|
email_from = email_from or email_host_user
|
|
email_recipient = email_recipient or email_from
|
|
connection = get_connection(
|
|
host=email_host, port=email_port,
|
|
username=email_host_user, password=email_host_password,
|
|
use_tls=email_use_tls, use_ssl=email_use_ssl,
|
|
)
|
|
send_mail(
|
|
subject, message, email_from, [email_recipient],
|
|
connection=connection
|
|
)
|
|
except SMTPSenderRefused as e:
|
|
error = e.smtp_error
|
|
if isinstance(error, bytes):
|
|
for coding in ('gbk', 'utf8'):
|
|
try:
|
|
error = error.decode(coding)
|
|
except UnicodeDecodeError:
|
|
continue
|
|
else:
|
|
break
|
|
return Response({"error": str(error)}, status=400)
|
|
except Exception as e:
|
|
logger.error(e)
|
|
return Response({"error": str(e)}, status=400)
|
|
return Response({"msg": self.success_message.format(email_recipient)})
|