modern-di 0.1.0__py3-none-any.whl

modern_di/__init__.py

@@ -0,0 +1,11 @@
+from modern_di.container import Container
+from modern_di.graph import BaseGraph
+from modern_di.scope import Scope
+
+
+__all__ = [
+    "BaseGraph",
+    "Container",
+    "Scope",
+    "resolvers",
+]

modern_di/container.py

@@ -0,0 +1,109 @@
+import contextlib
+import enum
+import types
+import typing
+
+from modern_di.resolver_state import ResolverState
+
+
+if typing.TYPE_CHECKING:
+    import typing_extensions
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+
+
+class Container(contextlib.AbstractAsyncContextManager["Container"]):
+    __slots__ = "scope", "parent_container", "_is_async", "_resolver_states", "_overrides"
+
+    def __init__(self, *, scope: enum.IntEnum, parent_container: typing.Optional["Container"] = None) -> None:
+        if scope.value != 1 and parent_container is None:
+            msg = "Only first scope can be used without parent_container"
+            raise RuntimeError(msg)
+
+        self.scope = scope
+        self.parent_container = parent_container
+        self._is_async: bool | None = None
+        self._resolver_states: dict[str, ResolverState[typing.Any]] = {}
+        self._overrides: dict[str, typing.Any] = {}
+
+    def _exit(self) -> None:
+        self._is_async = None
+        self._resolver_states = {}
+
+    def _check_entered(self) -> None:
+        if self._is_async is None:
+            msg = "Enter the context first"
+            raise RuntimeError(msg)
+
+    def build_child_container(self) -> "typing_extensions.Self":
+        self._check_entered()
+
+        try:
+            new_scope = self.scope.__class__(self.scope.value + 1)
+        except ValueError as exc:
+            msg = f"Max scope is reached, {self.scope.name}"
+            raise RuntimeError(msg) from exc
+
+        return self.__class__(scope=new_scope, parent_container=self)
+
+    def find_container(self, scope: enum.IntEnum) -> "typing_extensions.Self":
+        container = self
+        while container.scope > scope and container.parent_container:
+            container = typing.cast("typing_extensions.Self", container.parent_container)
+        return container
+
+    def fetch_resolver_state(
+        self, resolver_id: str, is_async_resource: bool = False, is_lock_required: bool = False
+    ) -> ResolverState[typing.Any]:
+        self._check_entered()
+        if is_async_resource and self._is_async is False:
+            msg = "Resolving async resource in sync container is not allowed"
+            raise RuntimeError(msg)
+
+        if resolver_id not in self._resolver_states:
+            self._resolver_states[resolver_id] = ResolverState(is_lock_required=is_lock_required)
+
+        return self._resolver_states[resolver_id]
+
+    def override(self, resolver_id: str, override_object: object) -> None:
+        self._overrides[resolver_id] = override_object
+
+    def fetch_override(self, resolver_id: str) -> object | None:
+        return self._overrides.get(resolver_id)
+
+    def reset_override(self, resolver_id: str | None = None) -> None:
+        if resolver_id is None:
+            self._overrides = {}
+        else:
+            self._overrides.pop(resolver_id, None)
+
+    async def __aenter__(self) -> "Container":
+        self._is_async = True
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        traceback: types.TracebackType | None,
+    ) -> None:
+        self._check_entered()
+        for resolver_state in reversed(self._resolver_states.values()):
+            await resolver_state.async_tear_down()
+        self._exit()
+
+    def __enter__(self) -> "Container":
+        self._is_async = False
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: types.TracebackType | None,
+    ) -> None:
+        self._check_entered()
+        for resolver_state in reversed(self._resolver_states.values()):
+            resolver_state.sync_tear_down()
+        self._exit()

modern_di/graph.py

@@ -0,0 +1,39 @@
+import typing
+
+from modern_di import Container
+from modern_di.resolvers import AbstractResolver, BaseCreatorResolver
+
+
+if typing.TYPE_CHECKING:
+    import typing_extensions
+
+
+T = typing.TypeVar("T")
+P = typing.ParamSpec("P")
+
+
+class BaseGraph:
+    resolvers: dict[str, AbstractResolver[typing.Any]]
+
+    def __new__(cls, *_: typing.Any, **__: typing.Any) -> "typing_extensions.Self":  # noqa: ANN401
+        msg = f"{cls.__name__} cannot not be instantiated"
+        raise RuntimeError(msg)
+
+    @classmethod
+    def get_resolvers(cls) -> dict[str, AbstractResolver[typing.Any]]:
+        if not hasattr(cls, "resolvers"):
+            cls.resolvers = {k: v for k, v in cls.__dict__.items() if isinstance(v, AbstractResolver)}
+
+        return cls.resolvers
+
+    @classmethod
+    async def async_resolve_creators(cls, container: Container) -> None:
+        for resolver in cls.get_resolvers().values():
+            if isinstance(resolver, BaseCreatorResolver) and resolver.scope == container.scope:
+                await resolver.async_resolve(container)
+
+    @classmethod
+    def sync_resolve_creators(cls, container: Container) -> None:
+        for resolver in cls.get_resolvers().values():
+            if isinstance(resolver, BaseCreatorResolver) and resolver.scope == container.scope:
+                resolver.sync_resolve(container)

modern_di/resolver_state.py

@@ -0,0 +1,38 @@
+import asyncio
+import contextlib
+import typing
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+
+
+class ResolverState(typing.Generic[T_co]):
+    __slots__ = "context_stack", "instance", "resolver_lock"
+
+    def __init__(self, is_lock_required: bool) -> None:
+        self.context_stack: contextlib.AsyncExitStack | contextlib.ExitStack | None = None
+        self.instance: T_co | None = None
+        self.resolver_lock: typing.Final = asyncio.Lock() if is_lock_required else None
+
+    async def async_tear_down(self) -> None:
+        if self.context_stack is None:
+            return
+
+        if isinstance(self.context_stack, contextlib.AsyncExitStack):
+            await self.context_stack.aclose()
+        else:
+            self.context_stack.close()
+        self.context_stack = None
+        self.instance = None
+
+    def sync_tear_down(self) -> None:
+        if self.context_stack is None:
+            return
+
+        if isinstance(self.context_stack, contextlib.AsyncExitStack):
+            msg = "Cannot tear down async context in `sync_tear_down`"
+            raise RuntimeError(msg)  # noqa: TRY004
+
+        self.context_stack.close()
+        self.context_stack = None
+        self.instance = None

modern_di/resolvers/__init__.py

@@ -0,0 +1,11 @@
+from modern_di.resolvers.base import AbstractResolver, BaseCreatorResolver
+from modern_di.resolvers.factory import Factory
+from modern_di.resolvers.resource import Resource
+
+
+__all__ = [
+    "AbstractResolver",
+    "BaseCreatorResolver",
+    "Factory",
+    "Resource",
+]

modern_di/resolvers/base.py

@@ -0,0 +1,88 @@
+import abc
+import enum
+import itertools
+import typing
+import uuid
+
+from modern_di import Container
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+R = typing.TypeVar("R")
+P = typing.ParamSpec("P")
+
+
+class AbstractResolver(typing.Generic[T_co], abc.ABC):
+    BASE_SLOTS: typing.ClassVar = ["scope", "resolver_id"]
+
+    def __init__(self, scope: enum.IntEnum) -> None:
+        self.scope = scope
+        self.resolver_id: typing.Final = str(uuid.uuid4())
+
+    @abc.abstractmethod
+    async def async_resolve(self, container: Container) -> T_co:
+        """Resolve dependency asynchronously."""
+
+    @abc.abstractmethod
+    def sync_resolve(self, container: Container) -> T_co:
+        """Resolve dependency synchronously."""
+
+    def override(self, override_object: object, container: Container) -> None:
+        container.override(self.resolver_id, override_object)
+
+    def reset_override(self, container: Container) -> None:
+        container.reset_override(self.resolver_id)
+
+    @property
+    def cast(self) -> T_co:
+        return typing.cast(T_co, self)
+
+
+class BaseCreatorResolver(AbstractResolver[T_co], abc.ABC):
+    BASE_SLOTS: typing.ClassVar = [*AbstractResolver.BASE_SLOTS, "_args", "_kwargs", "_creator"]
+
+    def __init__(
+        self,
+        scope: enum.IntEnum,
+        creator: typing.Callable[P, typing.Any],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> None:
+        super().__init__(scope)
+
+        if any(x.scope > self.scope for x in itertools.chain(args, kwargs.values()) if isinstance(x, AbstractResolver)):
+            msg = "Scope of dependency cannot be more than scope of dependent"
+            raise RuntimeError(msg)
+
+        self._creator: typing.Final = creator
+        self._args: typing.Final = args
+        self._kwargs: typing.Final = kwargs
+
+    def _sync_build_creator(self, container: Container) -> typing.Any:  # noqa: ANN401
+        return self._creator(
+            *typing.cast(
+                P.args, [x.sync_resolve(container) if isinstance(x, AbstractResolver) else x for x in self._args]
+            ),
+            **typing.cast(
+                P.kwargs,
+                {
+                    k: v.sync_resolve(container) if isinstance(v, AbstractResolver) else v
+                    for k, v in self._kwargs.items()
+                },
+            ),
+        )
+
+    async def _async_build_creator(self, container: Container) -> typing.Any:  # noqa: ANN401
+        return self._creator(
+            *typing.cast(
+                P.args,
+                [await x.async_resolve(container) if isinstance(x, AbstractResolver) else x for x in self._args],
+            ),
+            **typing.cast(
+                P.kwargs,
+                {
+                    k: await v.async_resolve(container) if isinstance(v, AbstractResolver) else v
+                    for k, v in self._kwargs.items()
+                },
+            ),
+        )

modern_di/resolvers/factory.py

@@ -0,0 +1,46 @@
+import enum
+import typing
+
+from modern_di import Container
+from modern_di.resolvers import BaseCreatorResolver
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+P = typing.ParamSpec("P")
+
+
+class Factory(BaseCreatorResolver[T_co]):
+    __slots__ = [*BaseCreatorResolver.BASE_SLOTS, "_creator"]
+
+    def __init__(
+        self,
+        scope: enum.IntEnum,
+        creator: typing.Callable[P, T_co],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> None:
+        super().__init__(scope, creator, *args, **kwargs)
+
+    async def async_resolve(self, container: Container) -> T_co:
+        container = container.find_container(self.scope)
+        if (override := container.fetch_override(self.resolver_id)) is not None:
+            return typing.cast(T_co, override)
+
+        resolver_state = container.fetch_resolver_state(self.resolver_id)
+        if resolver_state.instance is not None:
+            return typing.cast(T_co, resolver_state.instance)
+
+        resolver_state.instance = typing.cast(T_co, await self._async_build_creator(container))
+        return resolver_state.instance
+
+    def sync_resolve(self, container: Container) -> T_co:
+        container = container.find_container(self.scope)
+        if (override := container.fetch_override(self.resolver_id)) is not None:
+            return typing.cast(T_co, override)
+
+        resolver_state = container.fetch_resolver_state(self.resolver_id)
+        if resolver_state.instance is not None:
+            return typing.cast(T_co, resolver_state.instance)
+
+        resolver_state.instance = self._sync_build_creator(container)
+        return typing.cast(T_co, resolver_state.instance)

modern_di/resolvers/resource.py

@@ -0,0 +1,95 @@
+import contextlib
+import enum
+import inspect
+import typing
+
+from modern_di import Container
+from modern_di.resolvers import BaseCreatorResolver
+
+
+T_co = typing.TypeVar("T_co", covariant=True)
+P = typing.ParamSpec("P")
+
+
+class Resource(BaseCreatorResolver[T_co]):
+    __slots__ = [*BaseCreatorResolver.BASE_SLOTS, "_creator", "_args", "_kwargs", "_is_async"]
+
+    def _is_creator_async(
+        self,
+        _: contextlib.AbstractContextManager[T_co] | contextlib.AbstractAsyncContextManager[T_co],
+    ) -> typing.TypeGuard[contextlib.AbstractAsyncContextManager[T_co]]:
+        return self._is_async
+
+    def __init__(
+        self,
+        scope: enum.IntEnum,
+        creator: typing.Callable[P, typing.Iterator[T_co] | typing.AsyncIterator[T_co]],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> None:
+        new_creator: typing.Any
+        if inspect.isasyncgenfunction(creator):
+            self._is_async = True
+            new_creator = contextlib.asynccontextmanager(creator)
+        elif inspect.isgeneratorfunction(creator):
+            self._is_async = False
+            new_creator = contextlib.contextmanager(creator)
+        else:
+            msg = "Unsupported resource type"
+            raise RuntimeError(msg)
+
+        super().__init__(scope, new_creator, *args, **kwargs)
+
+    async def async_resolve(self, container: Container) -> T_co:
+        container = container.find_container(self.scope)
+        if (override := container.fetch_override(self.resolver_id)) is not None:
+            return typing.cast(T_co, override)
+
+        resolver_state = container.fetch_resolver_state(
+            self.resolver_id, is_async_resource=self._is_async, is_lock_required=self._is_async
+        )
+        if resolver_state.instance is not None:
+            return typing.cast(T_co, resolver_state.instance)
+
+        if resolver_state.resolver_lock:
+            await resolver_state.resolver_lock.acquire()
+
+        try:
+            if resolver_state.instance is not None:
+                return typing.cast(T_co, resolver_state.instance)
+
+            _intermediate_ = await self._async_build_creator(container)
+
+            if self._is_creator_async(self._creator):  # type: ignore[arg-type]
+                resolver_state.context_stack = contextlib.AsyncExitStack()
+                resolver_state.instance = await resolver_state.context_stack.enter_async_context(_intermediate_)
+            else:
+                resolver_state.context_stack = contextlib.ExitStack()
+                resolver_state.instance = resolver_state.context_stack.enter_context(_intermediate_)
+        finally:
+            if resolver_state.resolver_lock:
+                resolver_state.resolver_lock.release()
+
+        return typing.cast(T_co, resolver_state.instance)
+
+    def sync_resolve(self, container: Container) -> T_co:
+        container = container.find_container(self.scope)
+        if (override := container.fetch_override(self.resolver_id)) is not None:
+            return typing.cast(T_co, override)
+
+        resolver_state = container.fetch_resolver_state(self.resolver_id)
+        if resolver_state.instance is not None:
+            return typing.cast(T_co, resolver_state.instance)
+
+        if self._is_async:
+            msg = "Async resource cannot be resolved synchronously"
+            raise RuntimeError(msg)
+
+        _intermediate_ = self._sync_build_creator(container)
+
+        resolver_state.context_stack = contextlib.ExitStack()
+        resolver_state.instance = resolver_state.context_stack.enter_context(
+            typing.cast(contextlib.AbstractContextManager[typing.Any], _intermediate_)
+        )
+
+        return typing.cast(T_co, resolver_state.instance)

modern_di/scope.py

@@ -0,0 +1,8 @@
+import enum
+
+
+class Scope(enum.IntEnum):
+    APP = 1
+    REQUEST = 2
+    ACTION = 3
+    STEP = 4

modern_di-0.1.0.dist-info/METADATA

@@ -0,0 +1,38 @@
+Metadata-Version: 2.3
+Name: modern-di
+Version: 0.1.0
+Summary: Simple Dependency Injection framework
+Project-URL: repository, https://github.com/modern-python/that-depends
+Project-URL: docs, https://that-depends.readthedocs.io
+Author-email: Artur Shiriev <me@shiriev.ru>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: dependency injector,di,ioc-container,mocks,python
+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: Topic :: Software Development :: Libraries
+Classifier: Typing :: Typed
+Requires-Python: <4,>=3.10
+Description-Content-Type: text/markdown
+
+"Modern-DI"
+==
+[![Test Coverage](https://codecov.io/gh/modern-python/modern-di/branch/main/graph/badge.svg)](https://codecov.io/gh/modern-python/modern-di)
+[![MyPy Strict](https://img.shields.io/badge/mypy-strict-blue)](https://mypy.readthedocs.io/en/stable/getting_started.html#strict-mode-and-configuration)
+[![Supported versions](https://img.shields.io/pypi/pyversions/modern-di.svg)](https://pypi.python.org/pypi/modern-di)
+[![downloads](https://img.shields.io/pypi/dm/modern-di.svg)](https://pypistats.org/packages/modern-di)
+[![GitHub stars](https://img.shields.io/github/stars/modern-python/modern-di)](https://github.com/modern-python/modern-di/stargazers)
+
+Dependency injection framework for Python inspired by `dependency-injector` and `dishka`.
+
+It is production-ready and gives you the following:
+- Simple DI framework with IOC-container.
+- Async and sync resolving.
+- Python 3.10-3.13 support.
+- Full coverage by types annotations (mypy in strict mode).
+- Overriding dependencies for tests.
+- Package with zero dependencies.
+
+📚 [Documentation](https://modern-di.readthedocs.io)

modern_di-0.1.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+modern_di/__init__.py,sha256=2xhgfqi4gpzPrlfam5HSwTFRroWmYIybDgsSYD9pPNg,194
+modern_di/container.py,sha256=o-6_eoSGpOrIZlAXDryISWmgnSLq7u0YAVRXHvkWU0s,3742
+modern_di/graph.py,sha256=V6LUbtcZb64jBD-GJB2bJijEcPBHSYEEI9WZGflrFpQ,1335
+modern_di/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+modern_di/resolver_state.py,sha256=rRIoyWPK7XtMeWPBFuEe8t81WdrHbj5-jtpOqZWf0jY,1206
+modern_di/scope.py,sha256=nnrTLnFT_dMDLVIwvJHcGlJsB2k1WIHusrRSGQIMEsg,97
+modern_di/resolvers/__init__.py,sha256=ubCeXD-44otfS1-AG5vRbTaqn8d6szY7cKQTprFT3Pk,271
+modern_di/resolvers/base.py,sha256=BMZbYfkIzAW8-8CQ9aRVIu5D8zw9owKv4nz-z0EMYmg,2897
+modern_di/resolvers/factory.py,sha256=CyMvl--t3bCapxARFnCKNJNgy1svCR5c9HcwHm85G8Q,1650
+modern_di/resolvers/resource.py,sha256=s0c2PQQSjffdKKXxZ6fJVKJaiqqBgQLZ8mTpzKjDb4Q,3688
+modern_di-0.1.0.dist-info/METADATA,sha256=t_wifprG9bqMg-H6l8kTRoWYFPZcbJm6bMtBjsaJLeM,1825
+modern_di-0.1.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+modern_di-0.1.0.dist-info/licenses/LICENSE,sha256=aA5_eJwDKqnGvL7PbkPb9x5f-VGIqZ9cvWWkeGqeD90,1070
+modern_di-0.1.0.dist-info/RECORD,,

modern_di-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.25.0
+Root-Is-Purelib: true
+Tag: py3-none-any

modern_di-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 modern-python
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.