Immediately access django-celery-results TaskResult after starting

Question

I have several Celery tasks I'm executing within a Django view (more specifically within Django Rest Framework's perform_create method).

What I'm trying to achieve is to immediately (that is, as soon as the task has an id/is in the results backend) access the TaskResult object and do something with it, like this:

tasks = [do_something.s(a) for a in (1, 2, 3, 4,)]
results = group(*tasks).apply_async()

for result in results.children:
    task = TaskResult.objects.get(task_id=result.task_id)
    do_something_with_task_object(task)

Now, this fails with django_celery_results.models.DoesNotExist: TaskResult matching query does not exist.

I did not yet try it, but I could make this work with something like the following snippet. But that strikes me as plain wrong and ugly, also does it wait until the tasks are finished:

while not all([TaskResult.objects.filter(task_id=t.task_id).exists() for t in results.children]):
    pass        

Is there some way to make this work in a nice and clean fashion?

Answer 1

It turns out that a) the moment you ask a question on StackOverflow, you're able to answer it yourself and b) Django transaction management does everything you need.

If you wrap the call to task.apply_async in an atomic wrapper all is fine, e.g.

with transactions.atomic():
    results = group(*tasks).apply_async()

TaskResult.objects.get(task_id=results.children[0].task_id)