mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.7 KiB
73 lines
2.7 KiB
# -*- 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.tasks import get_email_connection as get_connection |
|
from common.utils import get_logger |
|
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)})
|
|
|