Merge pull request #11 from nrbnlulu/support-batch-register

Support batch register

aioinject/containers.py

@@ -37,15 +37,18 @@ def _init_extensions(self, extensions: Sequence[Extension]) -> None:
 
     def register(
         self,
-        provider: Provider[Any],
+        *providers: Provider[Any],
     ) -> None:
-        if provider.type_ in self.providers:
-            msg = f"Provider for type {provider.type_} is already registered"
-            raise ValueError(msg)
+        for provider in providers:
+            if provider.type_ in self.providers:
+                msg = (
+                    f"Provider for type {provider.type_} is already registered"
+                )
+                raise ValueError(msg)
 
-        self.providers[provider.type_] = provider
-        if class_name := getattr(provider.type_, "__name__", None):
-            self.type_context[class_name] = provider.type_
+            self.providers[provider.type_] = provider
+            if class_name := getattr(provider.type_, "__name__", None):
+                self.type_context[class_name] = provider.type_
 
     def get_provider(self, type_: type[T]) -> Provider[T]:
         try:

benchmark/container.py

@@ -25,7 +25,5 @@
 
 def create_container() -> aioinject.Container:
     container = aioinject.Container()
-    for provider in providers:
-        container.register(provider=provider)
-
+    container.register(*providers)
     return container

docs/code/contextmanagers/03_implicity.py

@@ -1,5 +1,6 @@
 from types import TracebackType
-from typing import Self
+
+from typing_extensions import Self
 
 import aioinject
 from aioinject import Scoped

docs/code/example.py

@@ -24,8 +24,10 @@ def get(self, id: int) -> str:
 
 
 container = aioinject.Container()
-container.register(aioinject.Singleton(Database))
-container.register(aioinject.Singleton(UserService))
+container.register(
+    aioinject.Singleton(Database),
+    aioinject.Singleton(UserService),
+)
 
 with container.sync_context() as ctx:
     service = ctx.resolve(UserService)

tests/container/test_container.py

@@ -43,6 +43,14 @@ def test_can_register_single(container: Container) -> None:
     assert container.providers == expected
 
 
+def test_can_register_batch(container: Container) -> None:
+    provider1 = providers.Scoped(_ServiceA)
+    provider2 = providers.Scoped(_ServiceB)
+    container.register(provider1, provider2)
+    excepted = {_ServiceA: provider1, _ServiceB: provider2}
+    assert container.providers == excepted
+
+
 def test_cant_register_multiple_providers_for_same_type(
     container: Container,
 ) -> None:

tests/extensions/mypy_test.py

@@ -6,7 +6,6 @@
 
 
 class _TestExtension(SyncOnResolveExtension, OnResolveExtension):
-
     async def on_resolve(
         self,
         context: InjectionContext,