（生鲜项目）17. drf实现发送短信验证码功能

import re
from datetime import datetime, timedelta
from rest_framework import serializers
from django.contrib.auth import get_user_model  

from MxShop.settings import REGEX_MOBILE
from .models import VerifyCode

User = get_user_model() # 可以获取数据库的userprofile表


# 验证手机号码
class SmsSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)

    # 单独对mobile字段进行验证
    def validate_mobile(self, mobile):
        # 手机号是否注册
        if User.objects.filter(mobile=mobile).count():
            raise serializers.ValidationError("用户已经存在")

        # 手机号码是否合法
        if not re.match(REGEX_MOBILE, mobile):
            raise serializers.ValidationError("手机号码非法")

        # 验证发送频率
        one_minute_ago = datetime.now() - timedelta(hours=0, minutes=1, seconds=0)
        if VerifyCode.objects.filter(add_time__gt=one_minute_ago, mobile=mobile).count():
            raise serializers.ValidationError("距离上一次发送未超过60s")

        # 验证通过
        return mobile

from django.shortcuts import render
from django.contrib.auth.backends import ModelBackend
from django.contrib.auth import get_user_model
from django.db.models import Q
from rest_framework.mixins import CreateModelMixin
from rest_framework import viewsets
from rest_framework.response import Response
from rest_framework import status
from random import choice

from .serializers import SmsSerializer
from utils.yunpian import YunPian
from MxShop.settings import APIKEY
from .models import VerifyCode

User = get_user_model()# 发送短信验证码
class SmsCodeViewset(CreateModelMixin,viewsets.GenericViewSet):
    serializer_class = SmsSerializer

    # 生成随机验证码的函数
    def generate_code(self):
        seeds="1234567890"
        code_lst=[]
        for i in range(4):
            code_lst.append(choice(seeds))
        return "".join(code_lst)


    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True) # 如果serializer不通过, 那么后面就不会执行,而且drf会自动封装400的状态码

        # 上面的mobile字段校验成功后,获取mobile, 并对其发送验证码
        mobile=serializer.validated_data["mobile"] # validated_data就是serializer返回的数据, 是一个字典
        yunpian = YunPian(APIKEY)
        code=self.generate_code()
        sms_status = yunpian.send_sms(code=code,mobile=mobile)

        # 如果发送失败,把错误信息填充给mobile,同时封装一个400的状态码
        if sms_status["code"] != 0:
            return Response(
                {"mobile":sms_status["msg"]},
                status=status.HTTP_400_BAD_REQUEST
            )
        else: # 如果发送成功就先保存code到数据库, 然后返回手机号码,并封装201状态码
            code_record=VerifyCode(code=code,mobile=mobile)
            code_record.save()
            return Response(
                {"mobile":mobile},
                status=status.HTTP_201_CREATED
            )

# 手机号码正则表达式
REGEX_MOBILE="^1[358]\d{9}$|^147\d{8}$|^176\d{8}$"

# 云片网设置
APIKEY = "cdb6f4b3860c552f790176a9e4f3fd85"

# 发送验证码的接口
router.register(r‘code‘, SmsCodeViewset, basename="code")