python-injection 0.14.6.post0__tar.gz → 0.14.6.post2__tar.gz

{python_injection-0.14.6.post0 → python_injection-0.14.6.post2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-injection
-Version: 0.14.6.post0
+Version: 0.14.6.post2
 Summary: Fast and easy dependency injection framework.
 Project-URL: Repository, https://github.com/100nm/python-injection
 Author: remimd

{python_injection-0.14.6.post0 → python_injection-0.14.6.post2}/injection/_core/common/invertible.py

@@ -6,6 +6,8 @@ from typing import Protocol, runtime_checkable
 
 @runtime_checkable
 class Invertible[T](Protocol):
+    __slots__ = ()
+
     @abstractmethod
     def __invert__(self) -> T:
         raise NotImplementedError

{python_injection-0.14.6.post0 → python_injection-0.14.6.post2}/injection/_core/module.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import threading
 from abc import ABC, abstractmethod
 from collections import OrderedDict, deque
 from collections.abc import (
@@ -27,7 +28,6 @@ from inspect import (
 )
 from inspect import signature as inspect_signature
 from logging import Logger, getLogger
-from threading import Lock
 from types import MethodType
 from typing import (
     Any,
@@ -982,7 +982,7 @@ class InjectMetadata[**P, T](Caller[P, T], EventListener):
 
     def __init__(self, wrapped: Callable[P, T], /, threadsafe: bool) -> None:
         self.__dependencies = Dependencies.empty()
-        self.__lock = Lock() if threadsafe else nullcontext()
+        self.__lock = threading.Lock() if threadsafe else nullcontext()
         self.__owner = None
         self.__tasks = deque()
         self.__wrapped = wrapped

{python_injection-0.14.6.post0 → python_injection-0.14.6.post2}/injection/_core/scope.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
+import itertools
 from abc import ABC, abstractmethod
 from collections import defaultdict
-from collections.abc import AsyncIterator, Iterator, MutableMapping
+from collections.abc import AsyncIterator, Iterator, Mapping, MutableMapping
 from contextlib import AsyncExitStack, ExitStack, asynccontextmanager, contextmanager
 from contextvars import ContextVar
 from dataclasses import dataclass, field
-from types import TracebackType
+from types import EllipsisType, TracebackType
 from typing import (
     Any,
     AsyncContextManager,
@@ -15,6 +16,7 @@ from typing import (
     NoReturn,
     Protocol,
     Self,
+    overload,
     runtime_checkable,
 )
 
@@ -26,18 +28,32 @@ from injection.exceptions import (
 )
 
 
-@dataclass(repr=False, slots=True)
-class _ScopeState:
-    # Shouldn't be instantiated outside `__SCOPES`.
+@runtime_checkable
+class ScopeState(Protocol):
+    __slots__ = ()
+
+    @property
+    @abstractmethod
+    def active_scopes(self) -> Iterator[Scope]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def bind(self, scope: Scope) -> ContextManager[None]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_scope(self) -> Scope | None:
+        raise NotImplementedError
+
+
+@dataclass(repr=False, frozen=True, slots=True)
+class _ContextualScopeState(ScopeState):
+    # Shouldn't be instantiated outside `__CONTEXTUAL_SCOPES`.
 
     __context_var: ContextVar[Scope] = field(
         default_factory=lambda: ContextVar(f"scope@{new_short_key()}"),
         init=False,
     )
-    __default: Scope | None = field(
-        default=None,
-        init=False,
-    )
     __references: set[Scope] = field(
         default_factory=set,
         init=False,
@@ -45,13 +61,10 @@ class _ScopeState:
 
     @property
     def active_scopes(self) -> Iterator[Scope]:
-        yield from self.__references
-
-        if default := self.__default:
-            yield default
+        return iter(self.__references)
 
     @contextmanager
-    def bind_contextual_scope(self, scope: Scope) -> Iterator[None]:
+    def bind(self, scope: Scope) -> Iterator[None]:
         self.__references.add(scope)
         token = self.__context_var.set(scope)
 
@@ -61,26 +74,38 @@ class _ScopeState:
             self.__context_var.reset(token)
             self.__references.remove(scope)
 
-    @contextmanager
-    def bind_shared_scope(self, scope: Scope) -> Iterator[None]:
-        if next(self.active_scopes, None):
-            raise ScopeError(
-                "A shared scope can't be defined when one or more contextual scopes "
-                "are defined on the same name."
-            )
+    def get_scope(self) -> Scope | None:
+        return self.__context_var.get(None)
+
+
+@dataclass(repr=False, slots=True)
+class _SharedScopeState(ScopeState):
+    __scope: Scope | None = field(default=None)
+
+    @property
+    def active_scopes(self) -> Iterator[Scope]:
+        if scope := self.__scope:
+            yield scope
 
-        self.__default = scope
+    @contextmanager
+    def bind(self, scope: Scope) -> Iterator[None]:
+        self.__scope = scope
 
         try:
             yield
         finally:
-            self.__default = None
+            self.__scope = None
 
     def get_scope(self) -> Scope | None:
-        return self.__context_var.get(self.__default)
+        return self.__scope
 
 
-__SCOPES: Final[defaultdict[str, _ScopeState]] = defaultdict(_ScopeState)
+__CONTEXTUAL_SCOPES: Final[Mapping[str, ScopeState]] = defaultdict(
+    _ContextualScopeState,
+)
+__SHARED_SCOPES: Final[Mapping[str, ScopeState]] = defaultdict(
+    _SharedScopeState,
+)
 
 
 @asynccontextmanager
@@ -98,36 +123,52 @@ def define_scope(name: str, *, shared: bool = False) -> Iterator[None]:
 
 
 def get_active_scopes(name: str) -> tuple[Scope, ...]:
-    state = __SCOPES.get(name)
+    active_scopes = (
+        state.active_scopes
+        for states in (__CONTEXTUAL_SCOPES, __SHARED_SCOPES)
+        if (state := states.get(name))
+    )
+    return tuple(itertools.chain.from_iterable(active_scopes))
+
 
-    if state is None:
-        return ()
+@overload
+def get_scope(name: str, default: EllipsisType = ...) -> Scope: ...
 
-    return tuple(state.active_scopes)
 
+@overload
+def get_scope[T](name: str, default: T) -> Scope | T: ...
 
-def get_scope(name: str) -> Scope:
-    state = __SCOPES.get(name)
 
-    if state is None or (scope := state.get_scope()) is None:
+def get_scope(name, default=...):  # type: ignore[no-untyped-def]
+    for states in (__CONTEXTUAL_SCOPES, __SHARED_SCOPES):
+        state = states.get(name)
+        if state and (scope := state.get_scope()):
+            return scope
+
+    if default is Ellipsis:
         raise ScopeUndefinedError(
             f"Scope `{name}` isn't defined in the current context."
         )
 
-    return scope
+    return default
 
 
 @contextmanager
 def _bind_scope(name: str, scope: Scope, shared: bool) -> Iterator[None]:
-    state = __SCOPES[name]
+    if shared:
+        is_already_defined = bool(get_active_scopes(name))
+        states = __SHARED_SCOPES
+
+    else:
+        is_already_defined = bool(get_scope(name, default=None))
+        states = __CONTEXTUAL_SCOPES
 
-    if state.get_scope():
+    if is_already_defined:
         raise ScopeAlreadyDefinedError(
             f"Scope `{name}` is already defined in the current context."
         )
 
-    strategy = state.bind_shared_scope if shared else state.bind_contextual_scope
-    with strategy(scope):
+    with states[name].bind(scope):
         yield
 
 

{python_injection-0.14.6.post0 → python_injection-0.14.6.post2}/pyproject.toml

@@ -25,7 +25,7 @@ test = [
 
 [project]
 name = "python-injection"
-version = "0.14.6.post0"
+version = "0.14.6.post2"
 description = "Fast and easy dependency injection framework."
 license = { text = "MIT" }
 readme = "README.md"