from __future__ import absolute_import, unicode_literals
import os
from celery import Celery
# 設(shè)置環(huán)境變量
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'meiduo.settings')
# 注冊(cè)Celery的APP
app = Celery('meiduo')
# 綁定配置文件
app.config_from_object('django.conf:settings', namespace='CELERY')
# 自動(dòng)發(fā)現(xiàn)各個(gè)app下的tasks.py文件
app.autodiscover_tasks()

CELERY_BROKER_URL = 'redis://127.0.0.1:6379/'
CELERY_RESULT_BACKEND = 'redis://127.0.0.1:6379/'
CELERY_RESULT_SERIALIZER = 'json'

from __future__ import absolute_import, unicode_literals
from .celery import app as celery_app
__all__ = ['celery_app']

from celery.task import task
from .comm import send_message

# 定義發(fā)送郵件的方法
@task
def mail(mobile,code):
    send_message(mobile,code,5)
    

from utils.tasks import mail
import random
class SendMes(APIView):
    # 短信驗(yàn)證
    def get(self,request):
        # 接收客戶(hù)端發(fā)送的數(shù)據(jù)
        imagecode = request.query_params.get('imagecode')
        print(imagecode)
        mobile = request.query_params.get('mobile')
        print(mobile)
        uuid = request.query_params.get('uuid')
        print(uuid)
        if not all([imagecode,mobile]):
            return Response({'msg':'沒(méi)有獲取到'})
        # 驗(yàn)證圖片驗(yàn)證碼
        conn =get_redis_conn()
        # redis 中取驗(yàn)證碼
        code = conn.get(uuid)
        print(code)
        if code:
            code = str(code,encoding='utf8')
            # 圖片驗(yàn)證碼對(duì)比
            if imagecode.lower() == code.lower():
                # 驗(yàn)證通過(guò)后調(diào)用發(fā)送短信接口
                sms_code = random.randint(10000,99999)
                # 重點(diǎn) 重點(diǎn) 重點(diǎn)！?。。。。。?
                result = mail.delay(mobile,sms_code,1)
               # 加入短信嗎發(fā)送成功
                if result:
                    # redis中要存短信驗(yàn)證嗎
                    conn.setex(mobile,60,sms_code)
                    # 把圖片驗(yàn)證碼從redis中刪除
                    conn.delete(uuid)
                    return Response({'msg':sms_code})
                else:
                    return ({'msg':'發(fā)送失敗'})
            else:
                return Response({'msg':'驗(yàn)證碼不正確'})
        return Response('ok')

celery worker -A meiduo --loglevel=info --pool=solo --autoscale=50,5