在另一台服务器上获取芹菜任务的状态

时间:2017-05-12 14:01:16

标签: python django rabbitmq celery worker

我有两台运行芹菜的服务器。

1 - Web服务器 2 - Workers Server

Web服务器负责将任务发送到工作服务器。工人们执行这些任务。

我想在Web服务器上监视这些任务的状态。有什么想法吗?

views.py

@user_passes_test(lambda u: u.is_superuser)
def taskCompute(request, uuid):
    """
    Now we have the uuid that identifies the image over we want to apply the process as an input.
    To trigger this view, the url has an uuid, and we receive it to receive the image we are working with
    """

    doc = AnnotationDocument.objects.get(uuid=uuid)
    image_name = doc.name_id
    doc.status = 'Q'
    doc.save()

    image_processing.delay(uuid, image_name)

    return HttpResponseRedirect(reverse('list_admin'))

tasks.py

    from __future__ import absolute_import, unicode_literals
    import os
    from celery import Celery
    import subprocess as sub
    from PIL import Image

    app = Celery('tasks',
                 broker='amqp://xxxxx',
                 backend='rpc://', )

    NFS_PATH = '/home/administrator/nfs'


    @app.task
    def image_processing(uuid, image_name):
        """
        WORKERS job: - Get the images from the NFS pre-processing path
                     - Move them to the worker path
                     - Process the images
                     - Create the PDF
                     - Leave the Results in the NFS post-processing path
        """

        current_working_path = os.path.join(NFS_PATH, uuid)
        local_filename_image = os.path.join(current_working_path, image_name)
        local_filename_annotations = os.path.join(current_working_path, "annotations.json")

        if os.path.isfile(local_filename_annotations):
            local_filename_annotations = local_filename_annotations

        else:

            local_filename_annotations = "None"

        cmd = '/home/administrator/Envs/CESSOR-env/bin/python' \
              + ' /home/administrator/CESSOR/compute.py' \
              + ' --iImage=' + local_filename_image \

        print "Command:", cmd

        p = sub.Popen(cmd.split(), stdout=sub.PIPE, stderr=sub.PIPE)
        output, errors = p.communicate()
        print output

        # Get thumbnail of the uploaded image to render it in the detail.html template:
        image_path = os.path.join(current_working_path, 'process_result.jpg')
        img = Image.open(image_path)
        w, h = img.size
        a = 301.0
        b = 200.0
        ptgx = a / w
        ptgy = b / h
        w2 = w * ptgx
        h2 = h * ptgy
        w2 = int(w2)
        h2 = int(h2)
        img_thumbnail = img.resize((w2, h2), Image.ANTIALIAS)
        # Save the thumbnail:
        input_thumbnail_filename = 'process_result_thnl.png'
        img_thumbnail.save(os.path.join(current_working_path, input_thumbnail_filename))
        return uuid

1 个答案:

答案 0 :(得分:1)

您需要配置Celery以将任务结果存储在Django数据库中。为此,请使用django-celery-results库。请参阅configuring the Django backend上的Celery文档。

完成此操作后,您可以编写一个视图来查询TaskResult模型,就像任何其他Django数据一样。