1 つまたは複数の Celery タスクを開始するカスタム管理コマンドを作成しようとしています。
このために、ヘルパー関数を定義しました。
from celery import chord
from front import models
from worker.tasks import process_account, notify # The celery tasks
def register_check(account_ids, user):
if not account_ids:
return
tasks = [process_account.si(user.pk, acc) for acc in account_ids]
maintask = chord(tasks)(notify.s())
maintask.track_started = True
kwargs = {
'task_id': maintask.id,
'user': user,
'accounts': map(int, account_ids),
}
user_job = models.UserJob.objects.create(**kwargs)
return user_job
ビューからこの方法でタスクを開始するとすべてうまくいきますが、カスタム管理コマンドから同じ関数を呼び出すとうまくいきません:
import sys
from django.core.management.base import BaseCommand
from front.utils import register_check
from front.models import User
class Command(BaseCommand):
help = 'Run recurring checks. Use --daily, --weekly and/or --monthly options.'
def handle(self, *args, **options):
users = User.objects.filter(condition='value')
for user in users:
# fetch account_ids
user_job = register_check(account_ids=account_ids, user=user)
sys.stdout.write('Done, processed {} users.\n'.format(users.count()))
代わりに、次のトレースバックを取得します。
# (...)
File "/home/danilo/.virtualenvs/radar/lib/python2.7/site-packages/amqp/connection.py", line 136, in __init__
self.transport = create_transport(host, connect_timeout, ssl)
File "/home/danilo/.virtualenvs/radar/lib/python2.7/site-packages/amqp/transport.py", line 264, in create_transport
return TCPTransport(host, connect_timeout)
File "/home/danilo/.virtualenvs/radar/lib/python2.7/site-packages/amqp/transport.py", line 99, in __init__
raise socket.error(last_err)
error: [Errno 111] Connection refused
管理コマンドでセロリを初期化する必要がありますか? または、Celery タスクをカスタム管理コマンドから機能させるために有効にする必要がある設定はありますか?
(関係ないと思いますが、Redis と Postgres をブローカー/バックエンドとして使用しています。)