refactoring: ♻️ Little things

Author: remimd

injection/core/module.py

@@ -21,7 +21,6 @@
     TypeVar,
     cast,
     final,
-    get_origin,
     runtime_checkable,
 )
 
@@ -93,7 +92,7 @@ def __iter__(self) -> Iterator[Event]:
 
     @property
     def origin(self) -> Event:
-        return next(self.__iter__())
+        return next(iter(self))
 
     @property
     def is_circular(self) -> bool:
@@ -224,10 +223,7 @@ def notify(self, event: Event):
 
     @staticmethod
     def __get_origin(cls: type) -> type:
-        if origin := get_origin(cls):
-            return origin
-
-        return cls
+        return getattr(cls, "__origin__", cls)
 
 
 """

tests/test_inject.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Generic, TypeVar
+from typing import Annotated, Generic, TypeVar
 
 import pytest
 
@@ -30,6 +30,13 @@ def my_function(instance: SomeInjectable):
 
         my_function()
 
+    def test_inject_with_annotated(self):
+        @inject
+        def my_function(instance: Annotated[SomeInjectable, "metadata"]):
+            assert isinstance(instance, SomeInjectable)
+
+        my_function()
+
     def test_inject_with_positional_only_parameter(self):
         @inject
         def my_function(instance: SomeInjectable, /, **kw):