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.
64 lines
2.4 KiB
64 lines
2.4 KiB
3 years ago
|
from collections import OrderedDict
|
||
|
|
|
||
|
|
from rest_framework.views import Response
|
||
|
|
from rest_framework.generics import GenericAPIView
|
||
|
|
from rest_framework.exceptions import APIException
|
||
|
|
from rest_framework import status
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
|
||
|
|
from common.message.backends.sms import SMS_MESSAGE
|
||
|
|
from common.message.backends.sms.tencent import TencentSMS
|
||
|
|
from settings.models import Setting
|
||
|
|
from common.permissions import IsSuperUser
|
||
|
|
from common.exceptions import JMSException
|
||
|
|
|
||
|
|
from .. import serializers
|
||
|
|
|
||
|
|
|
||
|
|
class TencentSMSTestingAPI(GenericAPIView):
|
||
|
|
permission_classes = (IsSuperUser,)
|
||
|
|
serializer_class = serializers.TencentSMSSettingSerializer
|
||
|
|
|
||
|
|
def post(self, request):
|
||
|
|
serializer = self.serializer_class(data=request.data)
|
||
|
|
serializer.is_valid(raise_exception=True)
|
||
|
|
|
||
|
|
tencent_secret_id = serializer.validated_data['TENCENT_SECRET_ID']
|
||
|
|
tencent_secret_key = serializer.validated_data.get('TENCENT_SECRET_KEY')
|
||
|
|
tencent_sms_sign_and_tmpl = serializer.validated_data['TENCENT_SMS_SIGN_AND_TEMPLATES']
|
||
|
|
tencent_sdkappid = serializer.validated_data.get('TENCENT_SDKAPPID')
|
||
|
|
|
||
|
|
test_phone = serializer.validated_data.get('SMS_TEST_PHONE')
|
||
|
|
|
||
|
|
if not test_phone:
|
||
|
|
raise JMSException(code='test_phone_required', detail=_('test_phone is required'))
|
||
|
|
|
||
|
|
if not tencent_secret_key:
|
||
|
|
secret = Setting.objects.filter(name='TENCENT_SECRET_KEY').first()
|
||
|
|
if secret:
|
||
|
|
tencent_secret_key = secret.cleaned_value
|
||
|
|
|
||
|
|
tencent_secret_key = tencent_secret_key or ''
|
||
|
|
|
||
|
|
try:
|
||
|
|
client = TencentSMS(
|
||
|
|
secret_id=tencent_secret_id,
|
||
|
|
secret_key=tencent_secret_key,
|
||
|
|
sdkappid=tencent_sdkappid
|
||
|
|
)
|
||
|
|
sign, tmpl = SMS_MESSAGE.VERIFICATION_CODE.get_sign_and_tmpl(tencent_sms_sign_and_tmpl)
|
||
|
|
|
||
|
|
client.send_sms(
|
||
|
|
phone_numbers=[test_phone],
|
||
|
|
sign_name=sign,
|
||
|
|
template_code=tmpl,
|
||
|
|
template_param=OrderedDict(code='test')
|
||
|
|
)
|
||
|
|
return Response(status=status.HTTP_200_OK, data={'msg': _('Test success')})
|
||
|
|
except APIException as e:
|
||
|
|
try:
|
||
|
|
error = e.detail['errmsg']
|
||
|
|
except:
|
||
|
|
error = e.detail
|
||
|
|
return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': error})
|