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): 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='AccessKeySecret', ) 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