Django-实现图形验证码、短信验证码功能（二）

接上文：Django-实现图形验证码、短信验证码功能（一）

短信验证码后端发送实现

在settings中添加redis短信验证码的缓存设置，选择redis的3号数据库为图形验证码储存位置。

CACHES = {
    "default": { # 默认
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "redis://127.0.0.1:6379/0",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },
    "session": { # session
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "redis://127.0.0.1:6379/1",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },
    "verify_code": { # 图形验证码
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "redis://127.0.0.1:6379/2",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },
    "sms_code": { # 短信验证码
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "redis://127.0.0.1:6379/3",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },

视图函数定义

依旧是标准的接收、校处理、响应四大步骤。

• 先接收前端传递的手机号、用户填写的图像验证码、uuid
• 对图像验证码进行校验，通过后再进行短信验证码发送
• 验证码需要设置过期时间
• 从容联云通讯提供的短信发送接口，复制代码，自行简单封装成工具包使用
• 发送短信功能使用celery异步方式执行

from django_redis import get_redis_connection
from django.views.generic.base import View
from . import constants
from django import http
from meiduo_mall.utils.response_code import RETCODE
import random
# 自定义异步处理短信验证码的模块
from celery_tasks.sms.tasks import send_sms


class SmsCodeView(View):
    def get(self,request,mobile):
        # 接收图像验证码
        image_code = request.GET.get("image_code")
        # 接收uuid
        uuid = request.GET.get("image_code_id")
        # 连接到存放短信验证码的redis数据库
        redis_cli2 = get_redis_connection("sms_code")
        # 判断数据库短信验证标识符是否存在未过期(防止恶意请求短信)
        if redis_cli2.get(mobile + "_flag") is not None:
            return http.JsonResponse({"code":RETCODE.SMSCODERR,"errmsg":"短信验证太频繁,稍后再试"})

        if not all([image_code,uuid]):
            return http.JsonResponse({"code":RETCODE.PARAMERR, "errmsg":"参数不完整"})
        # 校验图像验证码
        redis_cli = get_redis_connection("verify_code")
        image_code_redis = redis_cli.get(uuid)
        if image_code_redis is None:
            return http.JsonResponse({"code":RETCODE.IMAGECODEERR, "errmsg":"图形验证码已过期"})
        # 成功获取数据库验证码后,删除数据库中的验证码,防止被恶意重复使用
        redis_cli.delete(uuid)
        if image_code_redis.decode().lower() != image_code:
            return http.JsonResponse({"code":RETCODE.IMAGECODEERR, "errmsg":"图形验证码不正确"})

        # 图形验证码通过,开始发送短信验证码
        sms_code = "%6d" % random.randint(0,999999)
        # sms_code = random.randint(100000, 999999) 另外的简单实现6位数验证码方式

        # 使用管道操作数据库,减少交互次数
        redis_p = redis_cli2.pipeline()
        # 短信验证码存入数据库
        # constants.SMS_CODE_EXPIRES为自定义的过期时间
        redis_p.setex(mobile,constants.SMS_CODE_EXPIRES, sms_code)
        # 短信验证码标识, 设定60秒过期,防止频繁请求短信验证码
        redis_p.setex(mobile + "_flag",constants.SMS_CODE_EXPIRES, 1)
        redis_p.execute()

        # 异步执行短信发送任务
        # 通过delay调用，可以将任务加到队列中，交给celery去执行
        send_sms.delay(mobile,sms_code)

        # 后台打印查看验证码
        # print(sms_code)
        return http.JsonResponse({"code":RETCODE.OK,"errmsg":"OK"})

详解都在代码注释中，celery异步的实现见下篇：
Django-使用Celery异步发送短信验证码