from datetime import datetime, timedelta, timezone

from hatchet_sdk import Hatchet

hatchet = Hatchet()


async def create_scheduled() -> None:
    # > Create
    scheduled_run = await hatchet.scheduled.aio_create(
        workflow_name="simple-workflow",
        trigger_at=datetime.now(tz=timezone.utc) + timedelta(seconds=10),
        input={
            "data": "simple-workflow-data",
        },
        additional_metadata={
            "customer_id": "customer-a",
        },
    )

    scheduled_run.metadata.id  # the id of the scheduled run trigger

    # > Delete
    await hatchet.scheduled.aio_delete(scheduled_id=scheduled_run.metadata.id)

    # > List
    await hatchet.scheduled.aio_list()

    # > Get
    scheduled_run = await hatchet.scheduled.aio_get(
        scheduled_id=scheduled_run.metadata.id
    )