celery异步发送短信

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 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/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_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()),
]

3.测试接口

1. 接口URL

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

2. 请求携带参数

{
	"phone": 18538752511,
	"image_code":"aed3", # 前端生成的 图形验证码
	"image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008" # 前端生成的uuid
}