python-injection 0.2.1__tar.gz → 0.2.3__tar.gz

{python_injection-0.2.1 → python_injection-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-injection
-Version: 0.2.1
+Version: 0.2.3
 Summary: Fast and easy dependency injection framework.
 Home-page: https://github.com/soon-app/injection
 License: MIT

{python_injection-0.2.1 → python_injection-0.2.3}/injection/core.py

@@ -3,7 +3,7 @@ from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
 from functools import cached_property, wraps
 from inspect import Parameter
-from typing import Callable, Generic, TypeVar
+from typing import Callable, Generic, Iterable, TypeVar
 
 from injection.exceptions import NoInjectable
 
@@ -40,21 +40,28 @@ class UniqueInjectable(Injectable[T]):
 class InjectionManager:
     __container: dict[type, Injectable] = field(default_factory=dict, init=False)
 
-    def __getitem__(self, reference: type) -> Injectable:
+    def get(self, reference: type) -> Injectable:
         try:
             return self.__container[reference]
         except KeyError as exc:
             raise NoInjectable(f"No injectable for {reference.__name__}.") from exc
 
-    def __setitem__(self, reference: type, injectable: Injectable):
+    def set_multiple(self, references: Iterable[type], injectable: Injectable):
+        def reference_parser():
+            for reference in references:
+                self.check_if_exists(reference)
+                yield reference, injectable
+
+        new_values = reference_parser()
+        self.__container.update(new_values)
+
+    def check_if_exists(self, reference: type):
         if reference in self.__container:
             raise RuntimeError(
                 f"An injectable already exists for the "
                 f"reference class `{reference.__name__}`."
             )
 
-        self.__container[reference] = injectable
-
 
 _manager = InjectionManager()
 
@@ -70,16 +77,19 @@ class Decorator:
 
     def __call__(self, wp=None, /, **kwargs):
         def decorator(wrapped):
-            injectable = self.__injectable_class(wrapped)
+            def iter_references():
+                if isinstance(wrapped, type):
+                    yield wrapped
 
-            if isinstance(wrapped, type):
-                _manager[wrapped] = injectable
+                if reference := kwargs.pop("reference", None):
+                    yield reference
 
-            if reference := kwargs.pop("reference", None):
-                _manager[reference] = injectable
+                for reference in kwargs.pop("references", ()):
+                    yield reference
 
-            for reference in kwargs.pop("references", ()):
-                _manager[reference] = injectable
+            references = iter_references()
+            injectable = self.__injectable_class(wrapped)
+            _manager.set_multiple(references, injectable)
 
             return wrapped
 
@@ -93,7 +103,7 @@ del Decorator
 
 
 def get_instance(reference: type[T]) -> T:
-    return _manager[reference].get_instance()
+    return _manager.get(reference).get_instance()
 
 
 def inject(fn=None):
@@ -107,7 +117,7 @@ def inject(fn=None):
 
             for name, parameter in signature.parameters.items():
                 try:
-                    value = arguments[name]
+                    value = arguments.pop(name)
                 except KeyError:
                     try:
                         value = get_instance(parameter.annotation)
@@ -120,7 +130,7 @@ def inject(fn=None):
                     case Parameter.VAR_POSITIONAL:
                         args.extend(value)
                     case Parameter.VAR_KEYWORD:
-                        kwargs |= value
+                        kwargs.update(value)
                     case _:
                         kwargs[name] = value
 

{python_injection-0.2.1 → python_injection-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-injection"
-version = "0.2.1"
+version = "0.2.3"
 description = "Fast and easy dependency injection framework."
 authors = ["remimd"]
 keywords = ["dependencies", "inject", "injection"]