worker.py

"""Run the expense audit worker against a local Temporal dev server."""

import asyncio
import logging

from temporalio.client import Client
from temporalio.worker import Worker

from activities import (
    fetch_transactions,
    review_large_payments,
    review_recurring_payments,
)
from workflows import ExpenseAuditWorkflow

TASK_QUEUE = "expense-audit"
TARGET = "localhost:7233"
NAMESPACE = "default"


async def main() -> None:
    logging.basicConfig(level=logging.INFO)
    client = await Client.connect(TARGET, namespace=NAMESPACE)
    worker = Worker(
        client,
        task_queue=TASK_QUEUE,
        workflows=[ExpenseAuditWorkflow],
        activities=[
            fetch_transactions,
            review_recurring_payments,
            review_large_payments,
        ],
    )
    print(f"Worker up on {TARGET} / {NAMESPACE} / {TASK_QUEUE}. Ctrl-C to stop.")
    await worker.run()


if __name__ == "__main__":
    asyncio.run(main())