Merge pull request #96 from fhdufhdu/master

Added context manager condition to attr function(Fix #94)
ivankorobkov · Mar 24, 2024 · e6bc868 · e6bc868
2 parents 80f1948 + 189c20d
commit e6bc868
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/inject/__init__.py b/src/inject/__init__.py
@@ -281,7 +281,14 @@ def __init__(self, cls: Binding) -> None:
         self._cls = cls
 
     def __get__(self, obj: Any, owner: Any) -> Injectable:
-        return instance(self._cls)
+        inst = instance(self._cls)
+        if isinstance(inst, contextlib._AsyncGeneratorContextManager):
+            raise InjectorException(
+                    'Fail to load _AsyncGeneratorContextManager, Use autoparams, param or params instead of attr funcion')
+        elif isinstance(inst, contextlib._GeneratorContextManager):
+            with contextlib.ExitStack() as sync_stack:
+                inst = sync_stack.enter_context(inst)
+        return inst
 
 
 class _AttributeInjectionDataclass(Generic[T]):