aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

design design batch

Browse Source
This commit is contained in:
zhouchengrong
2024-12-11 11:35:41 +08:00
parent 7d54d398c5
commit 2a06825446
3 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/design_batch/design_batch_celery.py
View File

@@ -108,7 +108,7 @@ def batch_design(objects_data, tasks_id, json_name):
with lock: with lock:
object_response.append(items_response) object_response.append(items_response)
logger.info(items_response) # logger.info(items_response)
publish_status(tasks_id, step + 1, items_response) publish_status(tasks_id, step + 1, items_response)
active_threads -= 1 active_threads -= 1

2
app/service/design_batch/service.py
View File

@@ -5,7 +5,7 @@ from app.service.design_batch.utils.MQ import publish_status
async def start_design_batch_generate(data, file): async def start_design_batch_generate(data, file):
generate_clothes_task = batch_design.delay(json.loads(file.decode())['objects'], data.total, data.file_name) generate_clothes_task = batch_design(json.loads(file.decode())['objects'], data.total, data.file_name)
print(generate_clothes_task) print(generate_clothes_task)
publish_status(data.tasks_id, "0/100", "") publish_status(data.tasks_id, "0/100", "")
return {"task_id": data.tasks_id} return {"task_id": data.tasks_id}

4
app/service/design_batch/utils/save_json.py
View File

@@ -1,3 +1,4 @@
import io
import json import json
import logging import logging
import os import os
@@ -14,6 +15,7 @@ def oss_upload_json(oss_client, json_data, object_name):
file_path = os.path.join(save_dir, object_name) file_path = os.path.join(save_dir, object_name)
with open(file_path, 'w') as file: with open(file_path, 'w') as file:
json.dump(json_data, file, indent=4) json.dump(json_data, file, indent=4)
oss_client.fput_object("test", object_name, file_path) json_bytes = json.dumps(json_data).encode('utf-8')
oss_client.put_object("test", object_name, io.BytesIO(json_bytes), length=len(json_bytes), content_type="application/json")
except Exception as e: except Exception as e:
logger.warning(str(e)) logger.warning(str(e))