chore: snapshot workflow migration progress
This commit is contained in:
@@ -31,7 +31,13 @@ def _bbox_from_step_cadquery(step_path: str) -> dict | None:
|
||||
|
||||
|
||||
@celery_app.task(bind=True, name="app.tasks.step_tasks.process_step_file", queue="step_processing")
|
||||
def process_step_file(self, cad_file_id: str):
|
||||
def process_step_file(
|
||||
self,
|
||||
cad_file_id: str,
|
||||
workflow_run_id: str | None = None,
|
||||
workflow_node_id: str | None = None,
|
||||
**_: object,
|
||||
):
|
||||
"""Process a STEP file: extract objects, generate thumbnail, convert to glTF.
|
||||
|
||||
After processing completes, auto-populate cad_part_materials from Excel
|
||||
@@ -122,10 +128,24 @@ def process_step_file(self, cad_file_id: str):
|
||||
r.delete(lock_key) # always release on completion or unhandled error
|
||||
|
||||
pl.step_done("process_step_file")
|
||||
try:
|
||||
from app.domains.rendering.tasks import _update_workflow_run_status
|
||||
|
||||
# Queue thumbnail rendering on the dedicated single-concurrency worker
|
||||
from app.domains.pipeline.tasks.render_thumbnail import render_step_thumbnail
|
||||
render_step_thumbnail.delay(cad_file_id)
|
||||
_update_workflow_run_status(
|
||||
cad_file_id,
|
||||
"completed",
|
||||
workflow_run_id=workflow_run_id,
|
||||
workflow_node_id=workflow_node_id,
|
||||
)
|
||||
except Exception:
|
||||
logger.exception("Failed to update workflow state for process_step_file %s", cad_file_id)
|
||||
|
||||
# Legacy flow still auto-queues thumbnail generation here.
|
||||
# Graph-mode workflows dispatch explicit thumbnail save/render nodes instead.
|
||||
if workflow_run_id is None:
|
||||
from app.domains.pipeline.tasks.render_thumbnail import render_step_thumbnail
|
||||
|
||||
render_step_thumbnail.delay(cad_file_id)
|
||||
|
||||
|
||||
def _auto_populate_materials_for_cad(cad_file_id: str, tenant_id: str | None = None) -> None:
|
||||
|
||||
Reference in New Issue
Block a user