1.使用celery异步发送短信
1.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 = '-1'
if res == '-1' :
#如果发送结果-1 就是重试。
self.retry(countdown=5,max_retries=3,exc=Exception('短信发送失败'))
1.2在vifications/views.py中添加celery中发送短信视图函数
class SmsCodeView(APIView):
"""使用apiview的限流"""
# 1.所有人可以访问
permission_classes = (AllowAny,)
def post(self,request):
# 1.获取参数
phone = request.data.get('phone') # 手机号
imgage_code = request.data.get('image_code') # 图片验证码
imgage_code_uuid = request.data.get('imgage_code_uuid') #前端生成的uuid
#2.检查参数
if not all([phone,image_code,image_code_uuid]):
return Response({'code':999,"msg":"参数不全"})
if not re.macth(r"^1[3456789]\d{9}$",phone)
return Response({"code":999,"msg":"手机号码不正确"})
# 3.检查是否发送
redis_clent = get_redis_connection("img_code")
phone_exists = redis_celent.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_imgage_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 # 必须这么写,从main中导包
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":"短信发送成"})
1.3 添加路由
urlpatterns = [
path('sms_codes/', views.SmsCodeView.as_view()),
]
2.测试接口
http://192.168.56.100:8888/user/sms_codes/
{
"phone": 18538752511,
"image_code":"aed3", # 前端生成的 图形验证码
"image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008" # 前端生成的uuid
}