modern-di-taskiq 2.0.0__tar.gz

modern_di_taskiq-2.0.0/PKG-INFO

@@ -0,0 +1,60 @@
+Metadata-Version: 2.4
+Name: modern-di-taskiq
+Version: 2.0.0
+Summary: modern-di integration for taskiq
+Keywords: dependency-injection,di,ioc-container,modern-di,taskiq,python
+Author: Artur Shiriev
+Author-email: Artur Shiriev <me@shiriev.ru>
+License-Expression: MIT
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Typing :: Typed
+Classifier: Topic :: Software Development :: Libraries
+Requires-Dist: taskiq>=0.11,<0.13
+Requires-Dist: modern-di>=2.25,<3
+Requires-Python: >=3.10, <4
+Project-URL: Homepage, https://modern-di.modern-python.org
+Project-URL: Documentation, https://modern-di.modern-python.org
+Project-URL: Repository, https://github.com/modern-python/modern-di-taskiq
+Project-URL: Issues, https://github.com/modern-python/modern-di-taskiq/issues
+Project-URL: Changelog, https://github.com/modern-python/modern-di-taskiq/releases
+Description-Content-Type: text/markdown
+
+# modern-di-taskiq
+
+[Modern-DI](https://github.com/modern-python/modern-di) integration for [taskiq](https://taskiq-python.github.io).
+
+## Quickstart
+
+```python
+import typing
+
+from modern_di import Container, Group, Scope, providers
+from modern_di_taskiq import FromDI, setup_di
+from taskiq import InMemoryBroker
+
+
+class Settings:
+    def __init__(self) -> None:
+        self.greeting = "hello"
+
+
+class Dependencies(Group):
+    settings = providers.Factory(scope=Scope.APP, creator=Settings)
+
+
+broker = InMemoryBroker()
+setup_di(broker, Container(groups=[Dependencies], validate=True))
+
+
+@broker.task
+async def greet(name: str, settings: typing.Annotated[Settings, FromDI(Dependencies.settings)]) -> str:
+    return f"{settings.greeting}, {name}"
+```
+
+See the [documentation](https://modern-di.modern-python.org) for the full guide.

modern_di_taskiq-2.0.0/README.md

@@ -0,0 +1,33 @@
+# modern-di-taskiq
+
+[Modern-DI](https://github.com/modern-python/modern-di) integration for [taskiq](https://taskiq-python.github.io).
+
+## Quickstart
+
+```python
+import typing
+
+from modern_di import Container, Group, Scope, providers
+from modern_di_taskiq import FromDI, setup_di
+from taskiq import InMemoryBroker
+
+
+class Settings:
+    def __init__(self) -> None:
+        self.greeting = "hello"
+
+
+class Dependencies(Group):
+    settings = providers.Factory(scope=Scope.APP, creator=Settings)
+
+
+broker = InMemoryBroker()
+setup_di(broker, Container(groups=[Dependencies], validate=True))
+
+
+@broker.task
+async def greet(name: str, settings: typing.Annotated[Settings, FromDI(Dependencies.settings)]) -> str:
+    return f"{settings.greeting}, {name}"
+```
+
+See the [documentation](https://modern-di.modern-python.org) for the full guide.

modern_di_taskiq-2.0.0/modern_di_taskiq/__init__.py

@@ -0,0 +1,9 @@
+from modern_di_taskiq.main import FromDI, fetch_di_container, setup_di, taskiq_message_provider
+
+
+__all__ = [
+    "FromDI",
+    "fetch_di_container",
+    "setup_di",
+    "taskiq_message_provider",
+]

modern_di_taskiq-2.0.0/modern_di_taskiq/main.py

@@ -0,0 +1,64 @@
+import dataclasses
+import typing
+
+import taskiq
+from modern_di import Container, Scope, providers
+from taskiq import AsyncBroker, Context, TaskiqDepends, TaskiqEvents, TaskiqState
+
+
+taskiq_message_provider = providers.ContextProvider(taskiq.TaskiqMessage, scope=Scope.REQUEST)
+
+# The root container lives on ``broker.state`` under this named attribute. One
+# writer (``setup_di``), one reader (``fetch_di_container``) — naming it keeps
+# them in provable agreement instead of relying on a bare string literal.
+_ROOT_CONTAINER_ATTR = "modern_di_container"
+
+_CONNECTION_PROVIDERS = (taskiq_message_provider,)
+
+
+def setup_di(broker: AsyncBroker, container: Container) -> Container:
+    setattr(broker.state, _ROOT_CONTAINER_ATTR, container)
+    container.add_providers(*_CONNECTION_PROVIDERS)
+
+    def _open_root(_state: TaskiqState) -> None:
+        container.open()
+
+    async def _close_root(_state: TaskiqState) -> None:
+        await container.close_async()
+
+    broker.add_event_handler(TaskiqEvents.WORKER_STARTUP, _open_root)
+    broker.add_event_handler(TaskiqEvents.WORKER_SHUTDOWN, _close_root)
+    return container
+
+
+def fetch_di_container(broker: AsyncBroker) -> Container:
+    return typing.cast(Container, getattr(broker.state, _ROOT_CONTAINER_ATTR))
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+
+
+async def build_di_container(
+    context: typing.Annotated[Context, TaskiqDepends()],
+) -> typing.AsyncIterator[Container]:
+    container = fetch_di_container(context.broker).build_child_container(
+        scope=Scope.REQUEST, context={taskiq.TaskiqMessage: context.message}
+    )
+    try:
+        yield container
+    finally:
+        await container.close_async()
+
+
+@dataclasses.dataclass(slots=True, frozen=True)
+class Dependency(typing.Generic[T_co]):
+    dependency: providers.AbstractProvider[T_co] | type[T_co]
+
+    async def __call__(self, request_container: typing.Annotated[Container, TaskiqDepends(build_di_container)]) -> T_co:
+        return request_container.resolve_dependency(self.dependency)
+
+
+def FromDI(  # noqa: N802
+    dependency: providers.AbstractProvider[T_co] | type[T_co], *, use_cache: bool = True
+) -> T_co:
+    return typing.cast(T_co, TaskiqDepends(Dependency(dependency), use_cache=use_cache))

modern_di_taskiq-2.0.0/pyproject.toml

@@ -0,0 +1,84 @@
+[project]
+name = "modern-di-taskiq"
+description = "modern-di integration for taskiq"
+authors = [{ name = "Artur Shiriev", email = "me@shiriev.ru" }]
+requires-python = ">=3.10,<4"
+license = "MIT"
+readme = "README.md"
+keywords = ["dependency-injection", "di", "ioc-container", "modern-di", "taskiq", "python"]
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Typing :: Typed",
+    "Topic :: Software Development :: Libraries",
+]
+dependencies = ["taskiq>=0.11,<0.13", "modern-di>=2.25,<3"]
+version = "2.0.0"
+
+[project.urls]
+Homepage = "https://modern-di.modern-python.org"
+Documentation = "https://modern-di.modern-python.org"
+Repository = "https://github.com/modern-python/modern-di-taskiq"
+Issues = "https://github.com/modern-python/modern-di-taskiq/issues"
+Changelog = "https://github.com/modern-python/modern-di-taskiq/releases"
+
+[dependency-groups]
+dev = [
+    "pytest",
+    "pytest-cov",
+    "pytest-asyncio",
+]
+lint = [
+    "ty",
+    "ruff",
+    "eof-fixer",
+    "typing-extensions",
+]
+
+[build-system]
+requires = ["uv_build>=0.11,<1.0"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-name = "modern_di_taskiq"
+module-root = ""
+
+[tool.ruff]
+fix = false
+unsafe-fixes = true
+line-length = 120
+target-version = "py310"
+
+[tool.ruff.format]
+docstring-code-format = true
+
+[tool.ruff.lint]
+select = ["ALL"]
+ignore = [
+    "D1", # allow missing docstrings
+    "S101", # allow asserts
+    "TCH", # ignore flake8-type-checking
+    "FBT", # allow boolean args
+    "D203", # "one-blank-line-before-class" conflicting with D211
+    "D213", # "multi-line-summary-second-line" conflicting with D212
+    "COM812", # flake8-commas "Trailing comma missing"
+    "ISC001", # flake8-implicit-str-concat
+    "G004", # allow f-strings in logging
+]
+isort.lines-after-imports = 2
+isort.no-lines-before = ["standard-library", "local-folder"]
+
+[tool.pytest.ini_options]
+addopts = ""
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+
+[tool.coverage.report]
+exclude_also = [
+    "if typing.TYPE_CHECKING:",
+]