BBS论坛（三十三）

from celery import Celery
from flask import Flask
from flask_mail import Message
from exts import mail,alidayu
import config

app=Flask(__name__)

app.config.from_object(config)
mail.init_app(app)

alidayu.init_app(app)


# 运行本文件：
# 在windows操作系统上：
# celery -A tasks.celery worker --pool=solo --loglevel=info
# 在linux操作系统上：
# celery -A tasks.celery worker --loglevel=info


def make_celery(app):
    celery = Celery(app.import_name, backend=app.config['CELERY_RESULT_BACKEND'],
                    broker=app.config['CELERY_BROKER_URL'])
    celery.conf.update(app.config)
    TaskBase = celery.Task
    class ContextTask(TaskBase):
        abstract = True
        def __call__(self, *args, **kwargs):
            with app.app_context():
                return TaskBase.__call__(self, *args, **kwargs)
    celery.Task = ContextTask
    return celery

celery = make_celery(app)

@celery.task
def send_mail(subject,recipients,body):
    message = Message(subject=subject,recipients=recipients,body=body)
    mail.send(message)

# @celery.task
# def send_sms_captcha(telephone,captcha):
#     alidayu.send_sms(telephone,code=captcha)

# celery的配置
CELERY_RESULT_BACKEND = "redis://127.0.0.1:6379/0"
CELERY_BROKER_URL = "redis://127.0.0.1:6379/0"

@bp.route('/email_captcha/')
def email_captcha():
    #获取要修改的邮箱
    email = request.args.get('email')
    if not email:
        return restful.params_error('请输入要修改的邮箱')
    #得到大小写字母的列表
    source = list(string.ascii_letters)
    #得到大小写字母的列表 + 0到9的数字字符串
    source.extend(map(lambda x: str(x), range(0, 10)))
    # 随机取六位作为验证码
    captcha = "".join(random.sample(source, 6))
    #给这个邮箱发送邮件验证码
    # message = Message(subject='derek论坛密码修改邮件发送', recipients=[email,], body='你的验证码是：%s'%captcha)
    # try:
    #     mail.send(message)
    # except:
    #     return restful.server_error()
    
    #celery异步发送邮件
    send_mail.delay('derek论坛密码修改邮件发送', [email], '你的验证码是:%s' % captcha)
    #把邮箱和验证码保存到memcached中
    zlcache.set(email,captcha)
    return restful.success()