python-injection 0.9.1__py3-none-any.whl → 0.9.2__py3-none-any.whl

injection/common/invertible.py

@@ -15,7 +15,7 @@ class Invertible[T](Protocol):
 
 @dataclass(repr=False, eq=False, frozen=True, slots=True)
 class SimpleInvertible[T](Invertible[T]):
-    callable: Callable[[], T]
+    callable: Callable[..., T]
 
     def __invert__(self) -> T:
         return self.callable()

injection/common/lazy.py

@@ -9,7 +9,7 @@ __all__ = ("Lazy", "LazyMapping")
 class Lazy[T](Invertible[T]):
     __slots__ = ("__cache", "__is_set")
 
-    def __init__(self, factory: Callable[[], T]):
+    def __init__(self, factory: Callable[..., T]):
         self.__setup_cache(factory)
 
     def __invert__(self) -> T:
@@ -19,7 +19,7 @@ class Lazy[T](Invertible[T]):
     def is_set(self) -> bool:
         return self.__is_set
 
-    def __setup_cache(self, factory: Callable[[], T]):
+    def __setup_cache(self, factory: Callable[..., T]):
         def cache_generator() -> Iterator[T]:
             nonlocal factory
             cached = factory()

injection/core/module.py

@@ -17,7 +17,7 @@ from dataclasses import dataclass, field
 from enum import StrEnum
 from functools import partialmethod, singledispatchmethod, update_wrapper
 from inspect import Signature, isclass
-from queue import Queue
+from queue import Empty, Queue
 from types import MethodType, UnionType
 from typing import (
     Any,
@@ -161,7 +161,7 @@ class Injectable[T](Protocol):
 
 @dataclass(repr=False, frozen=True, slots=True)
 class BaseInjectable[T](Injectable[T], ABC):
-    factory: Callable[[], T]
+    factory: Callable[..., T]
 
 
 class NewInjectable[T](BaseInjectable[T]):
@@ -476,7 +476,7 @@ class Module(EventListener, Broker):
 
             function = InjectedFunction(wp)
 
-            @function.on_setup(block=False)
+            @function.on_setup
             def listen():
                 function.update(self)
                 self.add_listener(function)
@@ -723,10 +723,8 @@ class InjectedFunction(EventListener):
         update_wrapper(self, wrapped, updated=())
         self.__dependencies = Dependencies.empty()
         self.__owner = None
-
-        queue = Queue[Callable[[], Any]]()
-        queue.put_nowait(self.__set_signature)
-        self.__setup_queue = queue
+        self.__setup_queue = Queue[Callable[..., Any]](maxsize=2)
+        self.on_setup(self.__set_signature)
 
     def __repr__(self) -> str:  # pragma: no cover
         return repr(self.wrapped)
@@ -735,13 +733,7 @@ class InjectedFunction(EventListener):
         return str(self.wrapped)
 
     def __call__(self, /, *args, **kwargs) -> Any:
-        queue = self.__setup_queue
-
-        while not queue.empty():
-            setup = queue.get()
-            setup()
-            queue.task_done()
-
+        self.__setup()
         arguments = self.bind(args, kwargs)
         return self.wrapped(*arguments.args, **arguments.kwargs)
 
@@ -796,15 +788,15 @@ class InjectedFunction(EventListener):
         self.__dependencies = Dependencies.resolve(self.signature, module, self.__owner)
         return self
 
-    def on_setup(self, wrapped: Callable[[], Any] = None, /, *, block: bool = True):
+    def on_setup(self, wrapped: Callable[..., Any] = None, /):
         def decorator(wp):
-            self.__setup_queue.put(wp, block=block)
+            self.__setup_queue.put_nowait(wp)
             return wp
 
         return decorator(wrapped) if wrapped else decorator
 
     @singledispatchmethod
-    def on_event(self, event: Event, /) -> None:  # type: ignore
+    def on_event(self, event: Event, /) -> ContextManager | None:  # type: ignore
         return None
 
     @on_event.register
@@ -813,6 +805,20 @@ class InjectedFunction(EventListener):
         yield
         self.update(event.on_module)
 
+    def __setup(self):
+        queue = self.__setup_queue
+
+        while True:
+            try:
+                task = queue.get_nowait()
+            except Empty:
+                break
+
+            task()
+            queue.task_done()
+
+        queue.join()
+
     def __set_signature(self) -> Self:
         self.__signature__ = inspect.signature(self.wrapped, eval_str=True)
         return self

injection/testing/__init__.py

@@ -33,6 +33,8 @@ def use_test_injectables(*, on: Module = None, test_module: Module = None):
     for module in (on, test_module):
         module.unlock()
 
+    del module
+
     with on.use_temporarily(test_module, priority=ModulePriority.HIGH):
         yield
         on.unlock()

injection/utils.py

@@ -1,3 +1,4 @@
+from collections.abc import Callable
 from importlib import import_module
 from pkgutil import walk_packages
 from types import ModuleType
@@ -5,9 +6,10 @@ from types import ModuleType
 __all__ = ("load_package",)
 
 
-def load_package(package: ModuleType | str):
+def load_package(package: ModuleType | str, predicate: Callable[[str], bool] = None):
     """
     Function for importing all modules in a Python package.
+    Pass the `predicate` parameter if you want to filter the modules to be imported.
     """
 
     if isinstance(package, str):
@@ -20,8 +22,15 @@ def load_package(package: ModuleType | str):
             "Package has no `__path__` attribute, as it's probably a module."
         ) from exc
 
+    if predicate is None:
+
+        def predicate(_: str) -> bool:
+            return True
+
     for info in walk_packages(path=path, prefix=f"{package.__name__}."):
-        if info.ispkg:
+        name = info.name
+
+        if info.ispkg or not predicate(name):
             continue
 
-        import_module(info.name)
+        import_module(name)

{python_injection-0.9.1.dist-info → python_injection-0.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-injection
-Version: 0.9.1
+Version: 0.9.2
 Summary: Fast and easy dependency injection framework.
 Home-page: https://github.com/100nm/python-injection
 License: MIT

{python_injection-0.9.1.dist-info → python_injection-0.9.2.dist-info}/RECORD

@@ -2,20 +2,20 @@ injection/__init__.py,sha256=LhfwYFMBw6tJ7XA_C353w61OPt4IuQQgs3zIjrwMIz8,654
 injection/__init__.pyi,sha256=dfIQAR8L8Yr7dM-DODR4czwUd2a7zzFQHzb8mr1Q6P0,6235
 injection/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 injection/common/event.py,sha256=5Rdb2m3vAMCic8cQAVkStJDbrDrW_lk6kav8wYwmexM,1283
-injection/common/invertible.py,sha256=noNcmJ96IQi0XJms0dyfrx_AvKZnQM0sZyxhc2l6qo0,527
-injection/common/lazy.py,sha256=FEas6ewwOGWvRR8cflmyVvSLZd_-Fxd0QeIdvAM5I9c,1313
+injection/common/invertible.py,sha256=a-fht4TxMnki-oFFaZrDz52X_LWx0NU60KQlw72pzs4,528
+injection/common/lazy.py,sha256=SpQhlGBpMpeD9R5R-CdIInSVDv1XZWvnkERp1J6wsus,1315
 injection/common/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 injection/common/tools/threading.py,sha256=HlvP6k_-eZaK8JbB2b9PP171IZVe_0W2oMYsw3ebdKA,187
 injection/common/tools/type.py,sha256=ThM3Z1_gHmsdT4Jp7PlZgJY0lsufc4InPO65485Ugsg,1739
 injection/core/__init__.py,sha256=zuf0ubI2dHnbjn1059eduhS-ACIkkROa6-dhp10krh0,22
-injection/core/module.py,sha256=3FqU4XQgTNLI-VwpX1bKcRsLOaPeKSQ0sQuVsL_27I4,21625
+injection/core/module.py,sha256=5Hqdll6gxiaHQCZl9OVWKojmzj36azzOPvIgi-6PGdg,21723
 injection/exceptions.py,sha256=RsWWiWwKSMU0vxXQqQSn6CKHLMrGu4SSzYUAy9OJRXk,626
 injection/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 injection/integrations/blacksheep.py,sha256=82P_owhF3FKIJxxalnSic3AJnoOr1mkojkWMefpO6QI,731
 injection/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-injection/testing/__init__.py,sha256=xglKmIJgg2j1wt8qWJ_TtGfFyKmn7xE7b8kQhD2yEkQ,864
+injection/testing/__init__.py,sha256=cr4e3VnoKbWzkc6uyC03sffT59deKgLN-SKiZjhzooI,880
 injection/testing/__init__.pyi,sha256=_u95cJVHBkt69HUvnu2bbfYWmi7GOH_1qyFyvC1PxBk,518
-injection/utils.py,sha256=_79aiciimZpuoUTz5lojKySUMMzpkU-e7SotiHIFTI8,676
-python_injection-0.9.1.dist-info/METADATA,sha256=6vDRDhsWCG-FbwiHs-pvslfia8Fpe2n7hWJPoxJJbhg,3572
-python_injection-0.9.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-python_injection-0.9.1.dist-info/RECORD,,
+injection/utils.py,sha256=a_H6RC6cNWbG1NQ-zvtrlgwnfS5Jfu0X6VJwIUIHlSU,974
+python_injection-0.9.2.dist-info/METADATA,sha256=UkMpmL3GAAQ5nqRyf-3-j-_hB8pJrXLW7D3E9JSEsE0,3572
+python_injection-0.9.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+python_injection-0.9.2.dist-info/RECORD,,