|
|
|
Celery Background Tasks
|
|
|
|
=======================
|
|
|
|
|
|
|
|
If your application has a long running task, such as processing some uploaded
|
|
|
|
data or sending email, you don't want to wait for it to finish during a
|
|
|
|
request. Instead, use a task queue to send the necessary data to another
|
|
|
|
process that will run the task in the background while the request returns
|
|
|
|
immediately.
|
|
|
|
|
|
|
|
Celery is a powerful task queue that can be used for simple background tasks
|
|
|
|
as well as complex multi-stage programs and schedules. This guide will show you
|
|
|
|
how to configure Celery using Flask, but assumes you've already read the
|
|
|
|
`First Steps with Celery <https://celery.readthedocs.io/en/latest/getting-started/first-steps-with-celery.html>`_
|
|
|
|
guide in the Celery documentation.
|
|
|
|
|
|
|
|
Install
|
|
|
|
-------
|
|
|
|
|
|
|
|
Celery is a separate Python package. Install it from PyPI using pip::
|
|
|
|
|
|
|
|
$ pip install celery
|
|
|
|
|
|
|
|
Configure
|
|
|
|
---------
|
|
|
|
|
|
|
|
The first thing you need is a Celery instance, this is called the celery
|
|
|
|
application. It serves the same purpose as the :class:`~flask.Flask`
|
|
|
|
object in Flask, just for Celery. Since this instance is used as the
|
|
|
|
entry-point for everything you want to do in Celery, like creating tasks
|
|
|
|
and managing workers, it must be possible for other modules to import it.
|
|
|
|
|
|
|
|
For instance you can place this in a ``tasks`` module. While you can use
|
|
|
|
Celery without any reconfiguration with Flask, it becomes a bit nicer by
|
|
|
|
subclassing tasks and adding support for Flask's application contexts and
|
|
|
|
hooking it up with the Flask configuration.
|
|
|
|
|
|
|
|
This is all that is necessary to properly integrate Celery with Flask::
|
|
|
|
|
|
|
|
from celery import Celery
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
class ContextTask(celery.Task):
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
|
|
with app.app_context():
|
|
|
|
return self.run(*args, **kwargs)
|
|
|
|
|
|
|
|
celery.Task = ContextTask
|
|
|
|
return celery
|
|
|
|
|
|
|
|
The function creates a new Celery object, configures it with the broker
|
|
|
|
from the application config, updates the rest of the Celery config from
|
|
|
|
the Flask config and then creates a subclass of the task that wraps the
|
|
|
|
task execution in an application context.
|
|
|
|
|
|
|
|
An example task
|
|
|
|
---------------
|
|
|
|
|
|
|
|
Let's write a task that adds two numbers together and returns the result. We
|
|
|
|
configure Celery's broker and backend to use Redis, create a ``celery``
|
|
|
|
application using the factor from above, and then use it to define the task. ::
|
|
|
|
|
|
|
|
from flask import Flask
|
|
|
|
|
|
|
|
flask_app = Flask(__name__)
|
|
|
|
flask_app.config.update(
|
|
|
|
CELERY_BROKER_URL='redis://localhost:6379',
|
|
|
|
CELERY_RESULT_BACKEND='redis://localhost:6379'
|
|
|
|
)
|
|
|
|
celery = make_celery(flask_app)
|
|
|
|
|
|
|
|
@celery.task()
|
|
|
|
def add_together(a, b):
|
|
|
|
return a + b
|
|
|
|
|
|
|
|
This task can now be called in the background::
|
|
|
|
|
|
|
|
result = add_together.delay(23, 42)
|
|
|
|
result.wait() # 65
|
|
|
|
|
|
|
|
Run a worker
|
|
|
|
------------
|
|
|
|
|
|
|
|
If you jumped in and already executed the above code you will be
|
|
|
|
disappointed to learn that ``.wait()`` will never actually return.
|
|
|
|
That's because you also need to run a Celery worker to receive and execute the
|
|
|
|
task. ::
|
|
|
|
|
|
|
|
$ celery -A your_application.celery worker
|
|
|
|
|
|
|
|
The ``your_application`` string has to point to your application's package
|
|
|
|
or module that creates the ``celery`` object.
|
|
|
|
|
|
|
|
Now that the worker is running, ``wait`` will return the result once the task
|
|
|
|
is finished.
|