mirror of https://github.com/jumpserver/jumpserver
90 lines
4.8 KiB
Python
90 lines
4.8 KiB
Python
from django.utils.translation import ugettext_lazy as _
|
||
from rest_framework import serializers
|
||
|
||
from common.drf.fields import EncryptedField
|
||
from common.validators import PhoneValidator
|
||
from common.sdk.sms import BACKENDS
|
||
|
||
__all__ = [
|
||
'SMSSettingSerializer', 'AlibabaSMSSettingSerializer', 'TencentSMSSettingSerializer',
|
||
'HuaweiSMSSettingSerializer', 'CMPP2SMSSettingSerializer'
|
||
]
|
||
|
||
|
||
class SMSSettingSerializer(serializers.Serializer):
|
||
SMS_ENABLED = serializers.BooleanField(default=False, label=_('Enable SMS'))
|
||
SMS_BACKEND = serializers.ChoiceField(
|
||
choices=BACKENDS.choices, default=BACKENDS.ALIBABA, label=_('SMS provider / Protocol')
|
||
)
|
||
|
||
|
||
class SignTmplPairSerializer(serializers.Serializer):
|
||
SIGN_NAME = serializers.CharField(max_length=256, required=True, label=_('Signature'))
|
||
TEMPLATE_CODE = serializers.CharField(max_length=256, required=True, label=_('Template code'))
|
||
|
||
|
||
class BaseSMSSettingSerializer(serializers.Serializer):
|
||
PREFIX_TITLE = _('SMS')
|
||
|
||
SMS_TEST_PHONE = serializers.CharField(
|
||
max_length=256, required=False, validators=[PhoneValidator(), ],
|
||
allow_blank=True, label=_('Test phone')
|
||
)
|
||
|
||
def to_representation(self, instance):
|
||
data = super().to_representation(instance)
|
||
# data['SMS_BACKEND'] = self.fields['SMS_BACKEND'].default
|
||
return data
|
||
|
||
|
||
class AlibabaSMSSettingSerializer(BaseSMSSettingSerializer):
|
||
ALIBABA_ACCESS_KEY_ID = serializers.CharField(max_length=256, required=True, label='AccessKeyId')
|
||
ALIBABA_ACCESS_KEY_SECRET = EncryptedField(
|
||
max_length=256, required=False, label='access_key_secret',
|
||
)
|
||
ALIBABA_VERIFY_SIGN_NAME = serializers.CharField(max_length=256, required=True, label=_('Signature'))
|
||
ALIBABA_VERIFY_TEMPLATE_CODE = serializers.CharField(max_length=256, required=True, label=_('Template code'))
|
||
|
||
|
||
class TencentSMSSettingSerializer(BaseSMSSettingSerializer):
|
||
TENCENT_SECRET_ID = serializers.CharField(max_length=256, required=True, label='Secret id')
|
||
TENCENT_SECRET_KEY = EncryptedField(max_length=256, required=False, label='Secret key')
|
||
TENCENT_SDKAPPID = serializers.CharField(max_length=256, required=True, label='SDK app id')
|
||
TENCENT_VERIFY_SIGN_NAME = serializers.CharField(max_length=256, required=True, label=_('Signature'))
|
||
TENCENT_VERIFY_TEMPLATE_CODE = serializers.CharField(max_length=256, required=True, label=_('Template code'))
|
||
|
||
|
||
class HuaweiSMSSettingSerializer(BaseSMSSettingSerializer):
|
||
HUAWEI_APP_KEY = serializers.CharField(max_length=256, required=True, label='App key')
|
||
HUAWEI_APP_SECRET = EncryptedField(max_length=256, required=False, label='App secret')
|
||
HUAWEI_SMS_ENDPOINT = serializers.CharField(max_length=1024, required=True, label=_('App Access Address'))
|
||
HUAWEI_SIGN_CHANNEL_NUM = serializers.CharField(max_length=1024, required=True, label=_('Signature channel number'))
|
||
HUAWEI_VERIFY_SIGN_NAME = serializers.CharField(max_length=256, required=True, label=_('Signature'))
|
||
HUAWEI_VERIFY_TEMPLATE_CODE = serializers.CharField(max_length=256, required=True, label=_('Template code'))
|
||
|
||
|
||
class CMPP2SMSSettingSerializer(BaseSMSSettingSerializer):
|
||
CMPP2_HOST = serializers.CharField(max_length=256, required=True, label=_('Host'))
|
||
CMPP2_PORT = serializers.IntegerField(default=7890, label=_('Port'))
|
||
CMPP2_SP_ID = serializers.CharField(max_length=128, required=True, label=_('Enterprise code(SP id)'))
|
||
CMPP2_SP_SECRET = EncryptedField(max_length=256, required=False, label=_('Shared secret(Shared secret)'))
|
||
CMPP2_SRC_ID = serializers.CharField(max_length=256, required=False, label=_('Original number(Src id)'))
|
||
CMPP2_SERVICE_ID = serializers.CharField(max_length=256, required=True, label=_('Business type(Service id)'))
|
||
CMPP2_VERIFY_SIGN_NAME = serializers.CharField(max_length=256, required=True, label=_('Signature'))
|
||
CMPP2_VERIFY_TEMPLATE_CODE = serializers.CharField(
|
||
max_length=69, required=True, label=_('Template'),
|
||
help_text=_('Template need contain {code} and Signature + template length does not exceed 67 words. '
|
||
'For example, your verification code is {code}, which is valid for 5 minutes. '
|
||
'Please do not disclose it to others.')
|
||
)
|
||
|
||
def validate(self, attrs):
|
||
sign_name = attrs.get('CMPP2_VERIFY_SIGN_NAME', '')
|
||
template_code = attrs.get('CMPP2_VERIFY_TEMPLATE_CODE', '')
|
||
if template_code.find('{code}') == -1:
|
||
raise serializers.ValidationError(_('The template needs to contain {code}'))
|
||
if len(sign_name + template_code) > 65:
|
||
# 保证验证码内容在一条短信中(长度小于70字), 签名两边的括号和空格占3个字，再减去2个即可(验证码占用4个但占位符6个
|
||
raise serializers.ValidationError(_('Signature + Template must not exceed 65 words'))
|
||
return attrs
|