#!/usr/bin/env python3

"""Starter for the batch iterator sample."""

import asyncio

import datetime

import logging

from temporalio.client import Client

from batch_iterator.shared import ProcessBatchInput

from batch_iterator.workflows import IteratorBatchWorkflow

async def main():

"""Start the IteratorBatchWorkflow."""

# Set up logging

logging.basicConfig(level=logging.INFO)

# Create client

client = await Client.connect("localhost:7233")

# Create unique workflow ID with timestamp

timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")

workflow_id = f"batch_iterator_example_{timestamp}"

# Define workflow input

workflow_input = ProcessBatchInput(

record_count=100,

page_size=10,

)

print(f"Starting workflow with ID: {workflow_id}")

print(f"Input: {workflow_input}")

# Start workflow

handle = await client.start_workflow(

IteratorBatchWorkflow.run,

workflow_input,

id=workflow_id,

task_queue="batch_iterator_task_queue",

)

print(f"Workflow started with ID: {handle.id}")

print(f"Waiting for workflow to complete...")

# Wait for result

try:

result = await handle.result()

print(f"Workflow completed successfully!")

print(f"Total records processed: {result}")

except Exception as e:

print(f"Workflow failed with error: {e}")

raise

if __name__ == "__main__":

asyncio.run(main())