import asyncio
import logging
from datetime import datetime, timezone
from .dependencies import (
CurrentDocket,
CurrentExecution,
CurrentWorker,
Retry,
TaskLogger,
)
from .docket import Docket, TaskCollection
from .execution import Execution
from .worker import Worker
async def trace(
message: str,
logger: "logging.LoggerAdapter[logging.Logger]" = TaskLogger(),
docket: Docket = CurrentDocket(),
worker: Worker = CurrentWorker(),
execution: Execution = CurrentExecution(),
) -> None:
logger.info(
"%s: %r added to docket %r %s ago now running on worker %r",
message,
execution.key,
docket.name,
(datetime.now(timezone.utc) - execution.when),
worker.name,
)
async def fail(
message: str,
docket: Docket = CurrentDocket(),
worker: Worker = CurrentWorker(),
execution: Execution = CurrentExecution(),
retry: Retry = Retry(attempts=2),
) -> None:
raise Exception(
f"{message}: {execution.key} added to docket "
f"{docket.name} {datetime.now(timezone.utc) - execution.when} "
f"ago now running on worker {worker.name}"
)
async def sleep(
seconds: float, logger: "logging.LoggerAdapter[logging.Logger]" = TaskLogger()
) -> None:
logger.info("Sleeping for %s seconds", seconds)
await asyncio.sleep(seconds)
standard_tasks: TaskCollection = [
trace,
fail,
sleep,
]
