#!/usr/bin/env python3

"""Worker for the batch iterator sample."""

import asyncio

import logging

from temporalio import worker

from temporalio.client import Client

from batch_iterator.activities import get_records

from batch_iterator.workflows import IteratorBatchWorkflow, RecordProcessorWorkflow

async def main():

"""Run the worker that registers workflows and activities."""

# Set up logging

logging.basicConfig(level=logging.INFO)

# Create client

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

# Create worker

temporal_worker = worker.Worker(

client,

task_queue="batch_iterator_task_queue",

workflows=[IteratorBatchWorkflow, RecordProcessorWorkflow],

activities=[get_records],

)

print("Starting worker for batch iterator...")

print("Worker will process workflows and activities")

print("Press Ctrl+C to stop")

# Run the worker

await temporal_worker.run()

if __name__ == "__main__":

asyncio.run(main())