AiDA_Python/app/service/design_batch/utils/MQ.py

import json

import pika

from app.core.config import RABBITMQ_PARAMS


def publish_status(task_id, progress, result):
    connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
    # connection = pika.BlockingConnection(pika.ConnectionParameters('10.1.2.190'))
    channel = connection.channel()
    channel.queue_declare(queue='DesignBatch', durable=True)
    message = {'task_id': task_id, 'progress': progress, "result": result}
    channel.basic_publish(exchange='',
                          routing_key='DesignBatch',
                          body=json.dumps(message),
                          properties=pika.BasicProperties(
                              delivery_mode=2,
                          ))
    connection.close()


if __name__ == '__main__':
    publish_status("1", "1", "1")