click-extended 0.4.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

click_extended/core/other/context.py

@@ -0,0 +1,496 @@
+"""Context with a unified all contextual information across the context."""
+
+# pylint: disable=too-many-instance-attributes
+# pylint: disable=too-many-public-methods
+# pylint: disable=import-outside-toplevel
+# pylint: disable=protected-access
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, overload
+
+if TYPE_CHECKING:
+    from click import Context as ClickContext
+
+    from click_extended.core.decorators.argument import Argument
+    from click_extended.core.decorators.env import Env
+    from click_extended.core.decorators.option import Option
+    from click_extended.core.decorators.tag import Tag
+    from click_extended.core.nodes._root_node import RootNode
+    from click_extended.core.nodes.child_node import ChildNode
+    from click_extended.core.nodes.node import Node
+    from click_extended.core.nodes.parent_node import ParentNode
+
+
+@dataclass(frozen=True)
+class Context:
+    """
+    Context with a unified all contextual information across the context.
+
+    Attributes:
+      root (RootNode):
+        The root command node of the entire CLI tree.
+      current (Node | None):
+        The current node being processed.
+      parent (ParentNode | Tag):
+        The parent node (Option, Argument, Env, or Tag) that contains
+        the current child.
+      click_context (ClickContext):
+        The Click context object.
+      nodes (dict[str, Node]):
+        All registered nodes in the tree.
+      parents (dict[str, ParentNode]):
+        All parent nodes (Option/Argument/Env).
+      tags (dict[str, Tag]):
+        All tag instances by name.
+      children (dict[str, ChildNode]):
+        All child node instances.
+      data (dict[str, Any]):
+        Shared data store accessible across all nodes. Use this to pass
+        custom data between nodes.
+      debug (bool):
+        Debug mode flag. When `True`, handler exceptions show full tracebacks.
+        Set via `@debug()` decorator.
+    """
+
+    root: "RootNode"
+    current: "Node | None"
+    parent: "ParentNode | Tag | None"
+    click_context: "ClickContext"
+    nodes: dict[str, "Node"]
+    parents: dict[str, "ParentNode"]
+    tags: dict[str, "Tag"]
+    children: dict[str, "ChildNode"]
+    data: dict[str, Any]
+    debug: bool = False
+
+    def is_root(self) -> bool:
+        """
+        Check if the current node is a `RootNode` instance.
+
+        Returns:
+            bool:
+                `True` if current node is a `RootNode`, `False` otherwise.
+        """
+        from click_extended.core.nodes._root_node import RootNode
+
+        return isinstance(self.current, RootNode)
+
+    def is_parent(self) -> bool:
+        """
+        Check if the current node is a `ParentNode` instance.
+
+        Returns:
+            bool:
+                `True` if current node is a `ParentNode`, `False` otherwise.
+        """
+        from click_extended.core.nodes.parent_node import ParentNode
+
+        return isinstance(self.current, ParentNode)
+
+    def is_tag(self) -> bool:
+        """
+        Check if the parent node is a `Tag` instance.
+
+        Returns:
+            bool:
+                `True` if parent is a `Tag`, `False` otherwise.
+        """
+        from click_extended.core.decorators.tag import Tag
+
+        return isinstance(self.parent, Tag)
+
+    def is_child(self) -> bool:
+        """
+        Check if the current node is a `ChildNode` instance.
+
+        Returns:
+            bool:
+                `True` if current node is a `ChildNode`, `False` otherwise.
+        """
+        from click_extended.core.nodes.child_node import ChildNode
+
+        return isinstance(self.current, ChildNode)
+
+    def is_argument(self) -> bool:
+        """
+        Check if the current node is an `Argument` instance.
+
+        Returns:
+            bool:
+                `True` if current node is an `Argument`, `False` otherwise.
+        """
+        from click_extended.core.decorators.argument import Argument
+
+        return isinstance(self.current, Argument)
+
+    def is_option(self) -> bool:
+        """
+        Check if the current node is an `Option` instance.
+
+        Returns:
+            bool:
+                `True` if current node is an `Option`, `False` otherwise.
+        """
+        from click_extended.core.decorators.option import Option
+
+        return isinstance(self.current, Option)
+
+    def is_env(self) -> bool:
+        """
+        Check if the current node is an `Env` instance.
+
+        Returns:
+            bool:
+                `True` if current node is an `Env`, `False` otherwise.
+        """
+        from click_extended.core.decorators.env import Env
+
+        return isinstance(self.current, Env)
+
+    def is_tagged(self) -> bool:
+        """
+        Check if the current instance is tagged.
+
+        Returns:
+            bool:
+                `True` if current node has tags, `False` otherwise.
+        """
+        from click_extended.core.nodes.parent_node import ParentNode
+
+        if isinstance(self.current, ParentNode):
+            return len(self.current.tags) > 0
+        return False
+
+    def get_root(self) -> "RootNode":
+        """
+        Get the root node.
+
+        Returns:
+            RootNode:
+                The root node of the tree.
+        """
+        return self.root
+
+    def get_children(self, name: str | None = None) -> list["ChildNode"]:
+        """
+        Get a list of all children defined under the same parent.
+
+        Args:
+            name (str | None, optional):
+                The parent name to get children from. If `None`,
+                uses the current parent.
+
+        Returns:
+            list[ChildNode]:
+                List of child nodes under the specified parent.
+        """
+        from click_extended.core.decorators.tag import Tag
+        from click_extended.core.nodes.child_node import ChildNode
+        from click_extended.core.nodes.parent_node import ParentNode
+
+        if name is not None:
+            parent: "ParentNode | Tag | None" = self.get_parent(name)
+            if parent is None:
+                return []
+        elif isinstance(self.parent, (ParentNode, Tag)):
+            parent = self.parent
+        else:
+            return []
+
+        if not parent.children:
+            return []
+
+        return [
+            child
+            for child in parent.children.values()
+            if isinstance(child, ChildNode)
+        ]
+
+    def get_siblings(self) -> list["ChildNode"]:
+        """
+        Get a list of all siblings in the current parent, excluding the
+        current child.
+
+        Returns:
+            list[ChildNode]:
+                List of sibling child nodes.
+        """
+        from click_extended.core.nodes.child_node import ChildNode
+
+        if not isinstance(self.current, ChildNode):
+            return []
+
+        all_children = self.get_children()
+        return [child for child in all_children if child is not self.current]
+
+    def get_parent(self, name: str) -> "ParentNode | None":
+        """
+        Get a parent node by name.
+
+        Args:
+            name (str):
+                The parent node name to retrieve.
+
+        Returns:
+            ParentNode | None:
+                The parent node if found, `None` otherwise.
+        """
+        return self.parents.get(name)
+
+    def get_node(self, name: str) -> "Node | None":
+        """
+        Get any node by name.
+
+        Args:
+            name (str):
+                The node name to retrieve.
+
+        Returns:
+            Node | None:
+                The node if found, `None` otherwise.
+        """
+        return self.nodes.get(name)
+
+    def get_tag(self, name: str) -> "Tag | None":
+        """
+        Get a tag by name.
+
+        Args:
+            name (str):
+                The tag name to retrieve.
+
+        Returns:
+            Tag | None:
+                The tag if found, `None` otherwise.
+        """
+        return self.tags.get(name)
+
+    @overload
+    def get_tagged(self) -> dict[str, list["ParentNode"]]: ...
+
+    @overload
+    def get_tagged(self, name: str) -> list["ParentNode"]: ...
+
+    def get_tagged(
+        self, name: str | None = None
+    ) -> dict[str, list["ParentNode"]] | list["ParentNode"]:
+        """
+        Get tagged parent nodes, either all tags or a specific tag.
+
+        Args:
+            name (str | None, optional):
+                The tag name to get parents for. If `None`, returns all tags.
+
+        Returns:
+            dict[str, list[ParentNode]] | list[ParentNode]:
+                If `name` is `None`, returns a dictionary mapping tag names to
+                lists of parent nodes. If `name` is provided, returns a list
+                of parent nodes with that tag.
+        """
+        result: dict[str, list["ParentNode"]] = {}
+
+        for parent in self.parents.values():
+            for tag in parent.tags:
+                if tag not in result:
+                    result[tag] = []
+                result[tag].append(parent)
+
+        if name is None:
+            return result
+
+        return result.get(name, [])
+
+    def get_values(self) -> dict[str, Any]:
+        """
+        Get the processed value of all source nodes.
+
+        Returns:
+            dict[str, Any]:
+                Dictionary mapping parent names to their processed values.
+        """
+        return {
+            name: parent.get_value() for name, parent in self.parents.items()
+        }
+
+    def get_provided_arguments(self) -> list["Argument"]:
+        """
+        Get all provided positional arguments.
+
+        Returns:
+            list[Argument]:
+                List of provided argument nodes.
+        """
+        from click_extended.core.decorators.argument import Argument
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Argument) and parent.was_provided
+        ]
+
+    def get_provided_options(self) -> list["Option"]:
+        """
+        Get all provided keyword arguments.
+
+        Returns:
+            list[Option]:
+                List of provided option nodes.
+        """
+        from click_extended.core.decorators.option import Option
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Option) and parent.was_provided
+        ]
+
+    def get_provided_envs(self) -> list["Env"]:
+        """
+        Get all provided environment variables.
+
+        Returns:
+            list[Env]:
+                List of provided env nodes.
+        """
+        from click_extended.core.decorators.env import Env
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Env) and parent.was_provided
+        ]
+
+    def get_provided_value(self, name: str) -> Any:
+        """
+        Get the provided raw value of a parent node (before processing).
+
+        Args:
+            name (str):
+                The parent node name.
+
+        Returns:
+            Any:
+                The raw value if parent exists and was provided, `None`
+                otherwise.
+        """
+        parent = self.get_parent(name)
+        if parent is not None and parent.was_provided:
+            return parent.raw_value  # type: ignore
+        return None
+
+    def get_provided_values(self) -> dict[str, Any]:
+        """
+        Get the provided raw values in the context (before processing).
+
+        Returns:
+            dict[str, Any]:
+                The provided raw values in the context.
+        """
+        provided: dict[str, Any] = {}
+        for name, parent in self.parents.items():
+            if parent.was_provided:
+                provided[name] = parent.raw_value
+        return provided
+
+    def get_missing_arguments(self) -> list["Argument"]:
+        """
+        Get all missing positional arguments.
+
+        Returns:
+            list[Argument]:
+                List of all argument nodes.
+        """
+        from click_extended.core.decorators.argument import Argument
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Argument) and not parent.was_provided
+        ]
+
+    def get_missing_options(self) -> list["Option"]:
+        """
+        Get all missing keyword arguments.
+
+        Returns:
+            list[Option]:
+                List of all option nodes.
+        """
+        from click_extended.core.decorators.option import Option
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Option) and not parent.was_provided
+        ]
+
+    def get_missing_envs(self) -> list["Env"]:
+        """
+        Get all missing environment variables.
+
+        Returns:
+            list[Env]:
+                List of all env nodes.
+        """
+        from click_extended.core.decorators.env import Env
+
+        return [
+            parent
+            for parent in self.parents.values()
+            if isinstance(parent, Env) and not parent.was_provided
+        ]
+
+    def get_current_tags(self) -> list[str]:
+        """
+        Get a list of the tags of the current node.
+
+        Returns:
+            list[str]:
+                List of tag names for the current node.
+        """
+        from click_extended.core.nodes.parent_node import ParentNode
+
+        if isinstance(self.current, ParentNode):
+            return list(self.current.tags)
+        return []
+
+    def get_current_parent_as_parent(self) -> "ParentNode":
+        """
+        Get the current parent node as a `ParentNode`.
+
+        Returns:
+            ParentNode:
+                The current parent node.
+
+        Raises:
+            RuntimeError:
+                If called outside child node context or the parent is a `Tag`.
+        """
+        from click_extended.core.decorators.tag import Tag
+
+        if self.parent is None:
+            raise RuntimeError("No parent node in current context")
+        if isinstance(self.parent, Tag):
+            raise RuntimeError("Parent node is not a ParentNode instance.")
+        return self.parent
+
+    def get_current_parent_as_tag(self) -> "Tag":
+        """
+        Get the current parent node as a `Tag`.
+
+        Returns:
+            Tag:
+                The current parent node.
+
+        Raises:
+            RuntimeError:
+                If called outside child node context or
+                the parent is a `ParentNode`.
+        """
+        from click_extended.core.nodes.parent_node import ParentNode
+
+        if self.parent is None:
+            raise RuntimeError("No parent node in current context")
+        if isinstance(self.parent, ParentNode):
+            raise RuntimeError("Parent node is not a Tag instance.")
+        return self.parent

click_extended/decorators/__init__.py

@@ -0,0 +1,29 @@
+"""Initialization file for the `click_extended.decorators` module."""
+
+from click_extended.decorators.check import *
+from click_extended.decorators.check import __all__ as check_all
+from click_extended.decorators.compare import *
+from click_extended.decorators.compare import __all__ as compare_all
+from click_extended.decorators.convert import *
+from click_extended.decorators.convert import __all__ as convert_all
+from click_extended.decorators.load import *
+from click_extended.decorators.load import __all__ as load_all
+from click_extended.decorators.math import *
+from click_extended.decorators.math import __all__ as math_all
+from click_extended.decorators.misc import *
+from click_extended.decorators.misc import __all__ as misc_all
+from click_extended.decorators.random import *
+from click_extended.decorators.random import __all__ as random_all
+from click_extended.decorators.transform import *
+from click_extended.decorators.transform import __all__ as transform_all
+
+__all__ = [
+    *check_all,
+    *compare_all,
+    *convert_all,
+    *load_all,
+    *math_all,
+    *misc_all,
+    *random_all,
+    *transform_all,
+]  # type: ignore

click_extended/decorators/check/__init__.py

@@ -0,0 +1,57 @@
+"""Initialization file for the `click_extended.decorators.check` module."""
+
+from click_extended.decorators.check.conflicts import conflicts
+from click_extended.decorators.check.contains import contains
+from click_extended.decorators.check.dependencies import dependencies
+from click_extended.decorators.check.divisible_by import divisible_by
+from click_extended.decorators.check.ends_with import ends_with
+from click_extended.decorators.check.exclusive import exclusive
+from click_extended.decorators.check.falsy import falsy
+from click_extended.decorators.check.is_email import is_email
+from click_extended.decorators.check.is_hex_color import is_hex_color
+from click_extended.decorators.check.is_hostname import is_hostname
+from click_extended.decorators.check.is_ipv4 import is_ipv4
+from click_extended.decorators.check.is_ipv6 import is_ipv6
+from click_extended.decorators.check.is_json import is_json
+from click_extended.decorators.check.is_mac_address import is_mac_address
+from click_extended.decorators.check.is_negative import is_negative
+from click_extended.decorators.check.is_non_zero import is_non_zero
+from click_extended.decorators.check.is_port import is_port
+from click_extended.decorators.check.is_positive import is_positive
+from click_extended.decorators.check.is_url import is_url
+from click_extended.decorators.check.is_uuid import is_uuid
+from click_extended.decorators.check.length import length
+from click_extended.decorators.check.not_empty import not_empty
+from click_extended.decorators.check.regex import regex
+from click_extended.decorators.check.requires import requires
+from click_extended.decorators.check.starts_with import starts_with
+from click_extended.decorators.check.truthy import truthy
+
+__all__ = [
+    "conflicts",
+    "contains",
+    "dependencies",
+    "divisible_by",
+    "ends_with",
+    "exclusive",
+    "falsy",
+    "is_email",
+    "is_hex_color",
+    "is_hostname",
+    "is_ipv4",
+    "is_ipv6",
+    "is_json",
+    "is_mac_address",
+    "is_negative",
+    "is_non_zero",
+    "is_port",
+    "is_positive",
+    "regex",
+    "is_url",
+    "is_uuid",
+    "length",
+    "not_empty",
+    "requires",
+    "starts_with",
+    "truthy",
+]