Celery 配置与基本使用 并且用celery 一步发送短信
1.1 安装celery
pip install celery @ https://github.com/celery/celery/tarball/master
1.2
# celery_task/main.py import os from celery import Celery # 定义celery实例, 需要的参数, 1, 实例名, 2, 任务发布位置, 3, 结果保存位置 app = Celery('mycelery', broker='redis://127.0.0.1:6379/14', # 任务存放的地方 backend='redis://127.0.0.1:6379/15') # 结果存放的地方 @app.task def add(x, y): return x + y
'''1.启动celery''' #1.1 单进程启动celery celery -A main worker -l INFO #1.2 celery管理 celery multi start celery_test -A celery_test -l debug --autoscale=50,5 # celery并发数:最多50个,最少5个 ps auxww|grep "celery worker"|grep -v grep|awk '{print $2}'|xargs kill -9 # 关闭所有celery进程
# 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('短信发送失败'))
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": "频繁发送, 请稍后再试"}) # 验证图形验证码 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 # 必须这么写, 从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)
1.3 添加路由
urlpatterns = [ path('sms_codes/', views.SmsCodeView.as_view()), ]
http://192.168.56.100:8888/user/sms_codes/
请求携带参数
{ "phone": 18538752511, "image_code":"aed3", # 前端生成的 图形验证码 "image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008" # 前端生成的uuid }
做最野的狼