Run and wait (blocking) a celery workflow to complete

run_workflow.py

def run_workflow(title: str, workflow, timeout: int = 30, poll_interval: int = 1):
  start_time = datetime.now(timezone.utc)
  elapsed_time = 0

  result = workflow.delay()
  print(title)
  print(f"🔄 Workflow '{result}' created")   

  print(f"⏳ Waiting until result is available (timeout {timeout}s)...")
  while not result.ready():
    current_time = datetime.now(timezone.utc)
    elapsed_time = (current_time - start_time).total_seconds()
    print(f"🔍 Result not ready yet. Checking again in {poll_interval}s...")

    if elapsed_time > timeout:
      print("⏰ Timeout reached. Workflow did not complete in time. 🛑")
      return

    time.sleep(poll_interval)

  if result.successful():
    workflow_result = result.get()
    print(f"✅ Workflow '{result}' completed successfully in {elapsed_time:.2f}s. Result:\n{workflow_result}")
  else:
    print("❌ Workflow '{result}' failed. 😞")
    print(f"📝 Traceback:\n{result.traceback}")
  print('=' * 100)