Merge pull request #7 from taskiq-python/feature/class-deps

Author: s3rius

taskiq_dependencies/graph.py

@@ -100,9 +100,13 @@ def _build_graph(self) -> None:  # noqa: C901, WPS210
                 # If this is a class, we need to get signature of
                 # an __init__ method.
                 hints = get_type_hints(dep.dependency.__init__)  # noqa: WPS609
-            else:
-                # If this is function, we get it's type hints.
+            elif inspect.isfunction(dep.dependency):
+                # If this is function or an instance of a class, we get it's type hints.
                 hints = get_type_hints(dep.dependency)
+            else:
+                hints = get_type_hints(
+                    dep.dependency.__call__,  # type: ignore # noqa: WPS609
+                )
 
             # Now we need to iterate over parameters, to
             # find all parameters, that have TaskiqDepends as it's

tests/test_graph.py

@@ -290,3 +290,23 @@ def target(info: ParamInfo = Depends()) -> None:
     info: ParamInfo = kwargs["info"]
     assert info.name == ""
     assert info.definition is None
+
+
+def test_class_based_dependencies() -> None:
+    """Tests that if ParamInfo is used on the target, no error is raised."""
+
+    class TeClass:
+        def __init__(self, return_val: str) -> None:
+            self.return_val = return_val
+
+        def __call__(self) -> str:
+            return self.return_val
+
+    def target(class_val: str = Depends(TeClass("tval"))) -> None:
+        return None
+
+    with DependencyGraph(target=target).sync_ctx() as g:
+        kwargs = g.resolve_kwargs()
+
+    info: str = kwargs["class_val"]
+    assert info == "tval"