starter.py

import asyncio
from typing import Optional

from temporalio.client import Client, WorkflowUpdateStage
from temporalio.envconfig import ClientConfig

from message_passing.introduction import TASK_QUEUE
from message_passing.introduction.workflows import (
    ApproveInput,
    GetLanguagesInput,
    GreetingWorkflow,
    Language,
    SetLanguageInput,
)


async def main(client: Optional[Client] = None):
    if not client:
        config = ClientConfig.load_client_connect_config()
        config.setdefault("target_host", "localhost:7233")
        client = await Client.connect(**config)

    wf_handle = await client.start_workflow(
        GreetingWorkflow.run,
        id="greeting-workflow-1234",
        task_queue=TASK_QUEUE,
    )

    # 👉 Send a Query
    supported_languages = await wf_handle.query(
        GreetingWorkflow.get_languages, GetLanguagesInput(include_unsupported=False)
    )
    print(f"supported languages: {supported_languages}")

    # 👉 Execute an Update
    previous_language = await wf_handle.execute_update(
        GreetingWorkflow.set_language, SetLanguageInput(language=Language.CHINESE)
    )
    assert await wf_handle.query(GreetingWorkflow.get_language) == Language.CHINESE
    print(f"language changed: {previous_language.name} -> {Language.CHINESE.name}")

    # 👉 Start an Update and then wait for it to complete
    update_handle = await wf_handle.start_update(
        GreetingWorkflow.set_language_using_activity,
        SetLanguageInput(language=Language.ARABIC),
        wait_for_stage=WorkflowUpdateStage.ACCEPTED,
    )
    previous_language = await update_handle.result()
    assert await wf_handle.query(GreetingWorkflow.get_language) == Language.ARABIC
    print(f"language changed: {previous_language.name} -> {Language.ARABIC.name}")

    # 👉 Send a Signal
    await wf_handle.signal(GreetingWorkflow.approve, ApproveInput(name=""))
    print(await wf_handle.result())


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