使用celery异步发送短信

1. 在celery_task/main.py 中添加发送短信函数

# celery 项目中的所有导包地址， 都是以CELERY_BASE_DIR为基准设定
# 执行celery命令时，也需要进入CELERY_BASE_DIR目录执行
CELERY_BASE_DIR = os.path.dirname(os.path.abspath(__file__))
​
​
@app.task(bind=True)
def send_sms_code(self, mobile, datas):
    sys.path.insert(0, os.path.join(CELERY_BASE_DIR, '../syl'))
    # 在方法中导包
    from libs.rl_sms import send_message
    # time.sleep(5)
    try:
        # 用res接收发送结果, 成功是：0, 失败是：-1
        res = send_message(mobile, datas)
    except Exception as e:
        res = '-1'
        
    if res == '-1':
        # 如果发送结果是 -1 就重试
        self.retry(countdown=5, max_retries=3, exc=Exception('发送失败'))

2. 在verifications/view.py 中添加celery发送短信视图函数

class SmsCodeView(APIView):
    '''使用apiview限流'''
    # 1. 所有人可以访问
    permission_classes = (AllowAny,)
    
    def post(self, request):
        # 1.获取参数
        phone = request.data.get('phone')   
        image_code = request.data.get('image_code') # 图片验证码
        image_code_uuid = request.data.get('image_code_uuid') # 前端生成的uuid
        
        # 2.检查参数
        if not all([phone, image_code, image_code_uuid]):
            return Response({"code":999, "msg":"参数不全"})
        if not re.match(r'^1[3456789]\d{9}$', phone):
            return Response({"code":999, "msg":"手机号码不正确"})
        
        # 3.检查是否发送
        redis_client = get_redis_connection('img_code')
        phone_exissts = redis_client.get(phone)
        if phone_exists:
            return Response({"code":999, "msg":"频繁发送，请稍后再试"})
        
        # 验证图形验证码
        redis_image_code = redis_client.get(image_code_uuid) # bytes
        if redis_image_code:
            # bytes 转成 string
            redis_image_code = redis_image_code.decode()
            
        # 比较用户提供的图片内容是否和redis中保存的一致
        if image_code.upper() != redis_image_code:
            return Response({"code":999, "msg":"图片验证码不正确"})
        
        # 4. 发送
        code = '%06d' % random.randint(0,999999) # 随机6位验证码
        
        from syl.settings import BASE_DIR
        sys.path.insert(0, os.path.join(BASE_DIR, '../celery_task'))
        from main import send_sms_code  # 必须这么写
        
        send_sms_code.delay(phone, (code, "5"))
        print(code)
        
        # 5. 使用pipeline批量操作
        pl = redis_client.pipeline()        # 实例化pipeline对象
        pl.setex(phone, 60*5, code)         # 存储phone：code，5分钟有效
        pl.delete(image_code_uuid)          # 从redis中删除，以防再次使用
        pl.execute()
    
        # 6. 返回结果
        return Response({"code":0, "msg":"短信发送成功"})

3. 添加路由

urlpatterns = [
    path('sms_codes/', views.SmsCodeView.as_view()),
]

4. 测试接口

 url = http://192.168.56.100:8888/user/sms_codes/
 
{   # 以实际图片为准
    "phone": 18538752511,
    "image_code":"aed3",                                         # 前端生成的 图形验证码
    "image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008"     # 前端生成的uuid
}