47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from celery import Celery
|
|
|
|
|
|
celery = Celery(__name__)
|
|
|
|
|
|
def update_celery(celery, app):
|
|
celery.conf.update(app.config)
|
|
celery.conf.CELERYBEAT_SCHEDULE = {
|
|
"beat-dispatch_provision_portfolio": {
|
|
"task": "atat.jobs.dispatch_provision_portfolio",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_application": {
|
|
"task": "atat.jobs.dispatch_create_application",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_environment": {
|
|
"task": "atat.jobs.dispatch_create_environment",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_user": {
|
|
"task": "atat.jobs.dispatch_create_user",
|
|
"schedule": 60,
|
|
},
|
|
"beat-dispatch_create_environment_role": {
|
|
"task": "atat.jobs.dispatch_create_environment_role",
|
|
"schedule": 60,
|
|
},
|
|
"beat-send_task_order_files": {
|
|
"task": "atat.jobs.send_task_order_files",
|
|
"schedule": 60,
|
|
},
|
|
"beat-create_billing_instruction": {
|
|
"task": "atat.jobs.create_billing_instruction",
|
|
"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
|