click-extended 1.0.4__tar.gz → 1.0.6__tar.gz

{click_extended-1.0.4/click_extended.egg-info → click_extended-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: click_extended
-Version: 1.0.4
+Version: 1.0.6
 Summary: An extension to Click with additional features like automatic async support, aliasing and a modular decorator system.
 Author-email: Marcus Fredriksson <marcus@marcusfredriksson.com>
 License: MIT License

{click_extended-1.0.4 → click_extended-1.0.6}/click_extended/core/nodes/child_validation_node.py

@@ -3,6 +3,7 @@ Child validation node module for nodes that can
 act as both child and validation nodes.
 """
 
+import asyncio
 from abc import ABC
 from functools import wraps
 from typing import TYPE_CHECKING, Any, Callable, ParamSpec, TypeVar
@@ -88,8 +89,20 @@ class ChildValidationNode(ChildNode, ValidationNode, ABC):
             instance = cls(name=name, process_args=args, process_kwargs=kwargs)
             Tree.queue_child_validation(instance)
 
+            if asyncio.iscoroutinefunction(func):
+
+                @wraps(func)
+                async def async_wrapper(
+                    *call_args: P.args, **call_kwargs: P.kwargs
+                ) -> T:
+                    """Async wrapper that preserves the original function."""
+                    return await func(*call_args, **call_kwargs)  # type: ignore
+
+                return async_wrapper  # type: ignore
+
             @wraps(func)
             def wrapper(*call_args: P.args, **call_kwargs: P.kwargs) -> T:
+                """Wrapper that preserves the original function."""
                 return func(*call_args, **call_kwargs)
 
             return wrapper

{click_extended-1.0.4 → click_extended-1.0.6}/click_extended/core/nodes/validation_node.py

@@ -1,5 +1,6 @@
 """ValidationNode class for global validation logic."""
 
+import asyncio
 from abc import ABC
 from functools import wraps
 from typing import TYPE_CHECKING, Any, Callable, ParamSpec, TypeVar
@@ -111,6 +112,18 @@ class ValidationNode(Node, ABC):
             instance = cls(name=name, process_args=args, process_kwargs=kwargs)
             Tree.queue_validation(instance)
 
+            if asyncio.iscoroutinefunction(func):
+
+                @wraps(func)
+                async def async_wrapper(
+                    *call_args: P.args,
+                    **call_kwargs: P.kwargs,
+                ) -> T:
+                    """Async wrapper that preserves the original function."""
+                    return await func(*call_args, **call_kwargs)  # type: ignore
+
+                return async_wrapper  # type: ignore
+
             @wraps(func)
             def wrapper(*call_args: P.args, **call_kwargs: P.kwargs) -> T:
                 """Wrapper that preserves the original function."""

{click_extended-1.0.4 → click_extended-1.0.6}/click_extended/core/other/get_context.py

@@ -92,14 +92,14 @@ def get_context() -> Context:
         parent = parent_node
 
     return Context(
-        root=cast(RootNode, root_node),
-        current=cast(Node | None, current),
-        parent=cast(ParentNode | Tag | None, parent),
+        root=cast("RootNode", root_node),
+        current=cast("Node | None", current),
+        parent=cast("ParentNode | Tag | None", parent),
         click_context=click_context,
-        nodes=cast(dict[str, Node], all_nodes),
-        parents=cast(dict[str, ParentNode], parents),
-        tags=cast(dict[str, Tag], tags),
-        children=cast(dict[str, ChildNode], children),
+        nodes=cast("dict[str, Node]", all_nodes),
+        parents=cast("dict[str, ParentNode]", parents),
+        tags=cast("dict[str, Tag]", tags),
+        children=cast("dict[str, ChildNode]", children),
         data=meta.get("data", {}),
         debug=bool(meta.get("debug", False)),
     )

{click_extended-1.0.4 → click_extended-1.0.6/click_extended.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: click_extended
-Version: 1.0.4
+Version: 1.0.6
 Summary: An extension to Click with additional features like automatic async support, aliasing and a modular decorator system.
 Author-email: Marcus Fredriksson <marcus@marcusfredriksson.com>
 License: MIT License

{click_extended-1.0.4 → click_extended-1.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "click_extended"
-version = "1.0.4"
+version = "1.0.6"
 description = "An extension to Click with additional features like automatic async support, aliasing and a modular decorator system."
 authors = [
     { name = "Marcus Fredriksson", email = "marcus@marcusfredriksson.com" },