KiJdZddlmZddlZddlmZmZmZddlm Z m Z m Z m Z erddl mZdd lmZdd lmZGd d e d Zd!dZGdde dZd"dZGdde dZd#dZGdde eZd$dZGdde eZd%d&dZGdde dZd'd Zy)(zMContextual dependencies for accessing current Docket, Worker, Execution, etc.) annotationsN) TYPE_CHECKINGAnycast) Dependencycurrent_docketcurrent_executioncurrent_worker)Docket) Execution)WorkerceZdZddZy)_CurrentWorkerc2KtjSwN)r getselfs k/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/docket/dependencies/_contextual.py __aenter__z_CurrentWorker.__aenter__!!##Nreturnr__name__ __module__ __qualname__rrrr$r"rrc*tdtS)zA dependency to access the current Worker. Example: ```python @task async def my_task(worker: Worker = CurrentWorker()) -> None: assert isinstance(worker, Worker) ``` r)rrr!r"r CurrentWorkerr% .* ++r"ceZdZddZy)_CurrentDocketc2KtjSwr)r rrs rrz_CurrentDocket.__aenter__$rrNrr rr!r"rr(r(#r#r"r(r c*tdtS)zA dependency to access the current Docket. Example: ```python @task async def my_task(docket: Docket = CurrentDocket()) -> None: assert isinstance(docket, Docket) ``` r )rr(r!r"r CurrentDocketr,(r&r"ceZdZddZy)_CurrentExecutionc2KtjSwr)r rrs rrz_CurrentExecution.__aenter__7s $$&&rNrrrr!r"rr.r.6s'r"r.rc*tdtS)zA dependency to access the current Execution. Example: ```python @task async def my_task(execution: Execution = CurrentExecution()) -> None: assert isinstance(execution, Execution) ``` r)rr.r!r"rCurrentExecutionr2;s  .0 11r"ceZdZddZy)_TaskKeycFKtjjSwr)r rkeyrs rrz_TaskKey.__aenter__Js $$&***s!Nrstrrr!r"rr4r4Is+r"r4c2tttS)zA dependency to access the key of the currently executing task. Example: ```python @task async def my_task(key: str = TaskKey()) -> None: assert isinstance(key, str) ``` )rr8r4r!r"rTaskKeyr:Ns XZ  r"c4eZdZUded<ded<dd dZd dZy) _TaskArgument str | None parameterbooloptionalNc ||_||_yrr>r@)rr>r@s r__init__z_TaskArgument.__init__`s"  r"cK|jJtj} |j|jS#t$r|j rYywxYwwr)r>r r get_argumentKeyErrorr@)r executions rrz_TaskArgument.__aenter__dsY~~)))%))+  ))$..9 9 }}  s(#AAAAAAANF)r>r=r@r?rNone)rr)rrr __annotations__rCrr!r"rr<r<\sN!r"r<c6ttt||S)aA dependency to access a argument of the currently executing task. This is often useful in dependency functions so they can access the arguments of the task they are injected into. Example: ```python async def customer_name(customer_id: int = TaskArgument()) -> str: ...look up the customer's name by ID... return "John Doe" @task async def greet_customer(customer_id: int, name: str = Depends(customer_name)) -> None: print(f"Hello, {name}!") ``` )rrr<rBs r TaskArgumentrLos" ]9h7 88r"ceZdZddZy) _TaskLoggercJKtj}tjd|j}tj |it jjtjj|jSw)Nz docket.task.) r rlogging getLogger function_name LoggerAdapterr labelsr specific_labels)rrGloggers rrz_TaskLogger.__aenter__s%))+ ""\)2I2I1J#KL$$   $$&--/  $$&--/ ++-   sB!B#Nr%logging.LoggerAdapter[logging.Logger]rr!r"rrNrNs  r"rNrXc*tdtS)aA dependency to access a logger for the currently executing task. The logger will automatically inject contextual information such as the worker and docket name, the task key, and the current execution attempt number. Example: ```python @task async def my_task(logger: "LoggerAdapter[Logger]" = TaskLogger()) -> None: logger.info("Hello, world!") ``` rX)rrNr!r"r TaskLoggerrZs 7 GGr"rr*r0r7rH)r>r=r@r?rrrW) __doc__ __future__rrPtypingrrr_baserr r r docketr rGrworkerrrr%r(r,r.r2r8r4r:r<rLrNrZr!r"rrasS"++PP%$Z)$ ,$Z)$ ,' ;/' 2+z#+ !JsO&9(  *DE   Hr"