Merge pull request #3 from miguelteixeiraa/feat/make-decorator-return…

…-wrapped-function Make decorator return wrapped function
miguelteixeiraa · Dec 14, 2023 · 57ddb99 · 57ddb99
2 parents 715e091 + 40ab4e8
commit 57ddb99
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/crinkle/flow.py b/crinkle/flow.py
@@ -49,7 +49,7 @@ def decorator(
             function_processor: Union[
                 TypeFunctionProcessor, TypeFunctionProcessorAsync
             ],
-        ) -> None:
+        ) -> Union[TypeFunctionProcessor, TypeFunctionProcessorAsync]:
             if asyncio.iscoroutinefunction(function_processor):
                 self.add_processor(
                     processor=FunctionProcessorAsync(func=function_processor, name=name)
@@ -59,6 +59,7 @@ def decorator(
             self.add_processor(
                 processor=FunctionProcessor(func=function_processor, name=name)
             )
+            return function_processor
 
         return decorator
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "crinkle"
-version = "0.2.0"
+version = "0.2.1"
 description = ""
 authors = ["Miguel Teixeira <[email protected]>"]
 license = "MIT"