在Flask框架中将应用程序上下文传递给celery

Question

我试图将芹菜添加到现有的烧瓶项目中。 添加后，运行时出现“在应用程序上下文之外工作”错误。 芹菜工作者似乎缺少我的应用程序上下文。 但是在这种情况下，我不确定在哪里将申请上下文传递给芹菜工作者。

这是我当前的结构（我尝试遵循带有蓝图和api文档的工厂模式）：

-run.py
-app
    -module1
      -controller.py
      -model.py
      -service.py
    -__init__.py
    -config.py

---

对于init.py

# __init__.py
import os
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_cors import CORS
from app.config import Config
from flask_restplus import Api
from celery import Celery

cors = CORS()
db = SQLAlchemy()
api = Api()
celery = Celery(__name__, broker=Config.CELERY_BROKER_URL, include=["app.module1.service"])

def create_app(config_class = Config):
    app = Flask(__name__, static_url_path='')
    app.config.from_object(Config)

    cors.init_app(app)
    db.init_app(app)
    api.init_app(app=app)
    celery.conf.update(app.config)

    from app.module1.controller import blueprint
    from app.module1.controller import ns

    app.register_blueprint(blueprint)
    api.add_namespace(ns)

    return app

对于run.py

from app import create_app
app = create_app()
if __name__ == '__main__':
    app.run(threaded=True, debug=True)

对于service.py

from app import db, celery
@celery.task(bind=True)
def service1(self):
    # do somethigng & return

对于controller.py

from flask import Blueprint
from flask_restplus import Api, Resouce
blueprint = Blueprint('service', __name__)
apis = Api(app = blueprint)
ns = apis.namespace('service', 'service description')
@ns.route("/")
class SomeList(Resource):
    def get(self):
        service1.apply_async()
        # return

Answer 1

我认为这种混淆是基于您试图将应用程序上下文“传递”给Celery worker的事实。 实际上，Flask流程无法将上下文传递给工作人员，因为它们是不同的流程。 Celery工作进程需要通过调用create_app()创建自己的Flask应用程序实例，以便在需要时可以推送自己的应用程序上下文。

因此，例如，在您的service1任务中：

from app import db, celery, create_app

@celery.task(bind=True)
def service1(self):
    app = create_app()
    with app.app_context():
        # do somethigng & return

为了提高效率，您可以创建一个由所有任务共享的全局app ：

from app import db, celery, create_app
app = create_app()

@celery.task(bind=True)
def service1(self):
    with app.app_context():
        # do somethigng & return