28 lines
994 B
Python
28 lines
994 B
Python
import json
|
|
|
|
import pika
|
|
|
|
from app.service.design_batch.design_batch import batch_design
|
|
|
|
|
|
def publish_status(task_id, progress, result):
|
|
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
|
|
channel = connection.channel()
|
|
channel.queue_declare(queue='DesignBatch', durable=True)
|
|
message = {'task_id': task_id, 'progress': progress, "result": result}
|
|
print(message)
|
|
channel.basic_publish(exchange='',
|
|
routing_key='DesignBatch',
|
|
body=json.dumps(message),
|
|
properties=pika.BasicProperties(
|
|
delivery_mode=2,
|
|
))
|
|
connection.close()
|
|
|
|
|
|
async def start_design_batch_generate(data, file):
|
|
generate_clothes_task = batch_design.delay(json.loads(file.decode())['objects'], data.total, data.tasks_id)
|
|
print(generate_clothes_task)
|
|
publish_status(data.tasks_id, "0/100", "")
|
|
return {"task_id": data.tasks_id}
|