34 lines
986 B
Python
34 lines
986 B
Python
from celery import Celery
|
|
|
|
celery = Celery(__name__)
|
|
|
|
|
|
def update_celery(celery, app):
|
|
celery.conf.update(app.config)
|
|
celery.conf.CELERYBEAT_SCHEDULE = {
|
|
"beat-dispatch_create_environment": {
|
|
"task": "atst.jobs.dispatch_create_environment",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_atat_admin_user": {
|
|
"task": "atst.jobs.dispatch_create_atat_admin_user",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_environment_baseline": {
|
|
"task": "atst.jobs.dispatch_create_environment_baseline",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_provision_user": {
|
|
"task": "atst.jobs.dispatch_provision_user",
|
|
"schedule": 60,
|
|
},
|
|
}
|
|
|
|
class ContextTask(celery.Task):
|
|
def __call__(self, *args, **kwargs):
|
|
with app.app_context():
|
|
return self.run(*args, **kwargs)
|
|
|
|
celery.Task = ContextTask
|
|
return celery
|