中間還有重要的一步 沒有寫 就是 什么時候調用了 發(fā)送短信 這個方法
這幾篇寫的有些亂
首先寫 serializer 對 手機號進行驗證 驗證成功了 才調用 我們sms 功能 來發(fā)送短信
class SmsSerializer(serializers.Serializer):
mobile = serializers.CharField(max_length=11)
def validate_mobile(self, mobile):
"""
驗證手機號碼
:param data:
:return:
"""
#配置到settings中 更好 電話號碼正則表達式
REGEX_MOBILE = '^1[358]\d{9}$|^147\d{8}$|^176\d{8}$'
# 手機是否已經(jīng)注冊
if User.objects.filter(mobile=mobile).count():
raise serializers.ValidationError('用戶已經(jīng)存在')
# 驗證手機號碼是否合法
if not re.match(REGEX_MOBILE, mobile):
raise serializers.ValidationError('手機號碼非法')
# 驗證碼發(fā)送頻率 timedelata 是一分鐘前
one_mintes_ago = datetime.now() - timedelta(hours=0, minutes=1, seconds=0)
if VerifyCode.objects.filter(add_time__gt=one_mintes_ago, mobile=mobile).count():
raise serializers.ValidationError('距離上一次發(fā)送未超過60s')
return mobile
所以要排除 手機號碼已存在,手機號碼 非法 ,或者 驗證碼 發(fā)送申請過快這些情況
如果這些情況 都沒有 就返回mobile 進行短信發(fā)送
views
class SmscodeViewset(CreateModelMixin, viewsets.GenericViewSet):
"""
發(fā)送短信驗證碼
"""
serializer_class = SmsSerializer
def gennerate_code(self):
"""
生成四位數(shù)字的驗證碼
:return:
"""
seeds = "1234567890"
random_str = []
for i in range(4):
random_str.append(choices(seeds))
return "".join(random_str)
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
#從serializer的validated_data['mobile'] 獲取到mobile
mobile = serializer.validated_data['mobile']
#調用yunpian這個是自己寫的類 apikey 是自己的
yun_pian = YunPian(APIKEY)
code = self.gennerate_code() #驗證碼 就是上面我們寫的方法 生成一個四位數(shù)字
sms_status = yun_pian.send_sms(code=code,mobile=mobile) #然后 進行短信發(fā)送
#這里是云片網(wǎng) 文檔中告訴我們 成功 他返回了什么
if sms_status['code'] != 0 : #如果code 這個code跟我們上面的code不一樣 這是發(fā)送萬短信后的響應中的內容 如果code等于0 表示發(fā)送成功了 所以如果不等于0 就表示發(fā)送失敗 我們就可以返回一個 400 給他
return Response({
'mobile':sms_status['msg']
},status = status.HTTP_400_BAD_REQUEST)
else:
code_record = VerifyCode(code=code,mobile=mobile) #否則就是成功了 把code 和 mobile進行保存
code_record.save() #保存后 用戶進行驗證
return Response({
'mobile':sms_status['msg']
},status=status.HTTP_201_CREATED)
這里面主要的就是重寫 create方法 里面的解釋 我都寫在注釋中了 這樣更容易看懂