blob: cc5aff29814ff5922c12ce1965106287c4e1ef56 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
from __future__ import annotations
from contextvars import ContextVar
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from ._structures import JobInfo
from .schedulers.async_ import AsyncScheduler
from .schedulers.sync import Scheduler
from .workers.async_ import AsyncWorker
from .workers.sync import Worker
#: The currently running (local) scheduler
current_scheduler: ContextVar[Scheduler | AsyncScheduler | None] = ContextVar(
"current_scheduler", default=None
)
#: The worker running the current job
current_worker: ContextVar[Worker | AsyncWorker | None] = ContextVar(
"current_worker", default=None
)
#: Metadata about the current job
current_job: ContextVar[JobInfo] = ContextVar("job_info")
|