taskiq/docs/examples/state/generics_annot.py at develop · taskiq-python/taskiq

Repository navigation

from typing import Annotated, Generic, TypeVar

from taskiq import TaskiqDepends

T = TypeVar("T")

def default_dependency() -> int:

return 1

class GenericDep(Generic[T]):

def __init__(self, dep: Annotated[T, TaskiqDepends()]) -> None:

self.dep = dep

def print_dep(self) -> None:

print(self.dep)

class MyClassDep:

def __init__(self, i: Annotated[int, TaskiqDepends(default_dependency)]) -> None:

self.i = i

async def my_task(dep: Annotated[GenericDep[MyClassDep], TaskiqDepends()]) -> None:

assert isinstance(dep.dep, MyClassDep)

dep.print_dep()