03.celery发送短信接口

1.使用celery异步发送短信

 

1.1 在 celery_task/mian.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 utils.rl_sms import send_message

　　　　# time.sleep(5)

　　　　try:

　　　　　　# 用 res 接收发送结果, 成功是:０， 失败是：－１

　　　　　　res = send_message(mobile, datas)

　　　　except Exception as e:

　　　　　　res = '-1'

 

　　　　if res == '-1':

　　　　　　# 如果发送结果是 -1 就重试.

　　　　　　self.retry(countdown=5, max_retries=3, exc=Exception('短信发送失败'))

 

1.2 在 verifications/views.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_exists = redis_client.get(phone)

　　　　if phone_exists:

　　　　　　return Response({"code": 999, "msg": "频繁发送, 请稍后再试"})

 

　　　　# 4.检查验证图形验证码

　　　　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': '图片验证码不正确'})

 

　　　　# 5. 发送

　　　　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)

 

　　　　# 6.使用 pipeline 批量操作

　　　　pl = redis_client.pipeline() # 实例化pipeline对象

　　　　pl.setex(phone, 60 * 5, code) # 存储phone:code, 5分钟有效期

　　　　pl.delete(image_code_uuid) # 从redis中删除这个图片验证码, 以防再次被使用

　　　　pl.execute()

 

　　　　# 7. 返回结果

　　　　return Response({"code": 0, "msg": "短信发送成功"})

 

1.3 添加路由

 

urlpatterns = [

　　path('sms_codes/', views.SmsCodeView.as_view()),

]

 

2.测试接口

 

接口URL

http://192.168.56.100:8888/user/sms_codes/

 

请求携带参数

{

　　"phone": 18538752511,

　　"image_code":"aed3", 　　# 前端生成的 图形验证码

　　"image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008" # 前端生成的uuid

}