about summary refs log tree commit diff
path: root/.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py')
-rw-r--r--.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py b/.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py
new file mode 100644
index 00000000..bfe63c4d
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/pydash/chaining/all_funcs.py
@@ -0,0 +1,44 @@
+from abc import ABC, abstractmethod
+import typing as t
+
+
+class AllFuncs(ABC):
+    """Exposing all of the exposed functions of a module through an class."""
+
+    module: t.Any
+    invalid_method_exception: t.Type[Exception]
+
+    @abstractmethod
+    def _wrap(self, func) -> t.Callable:
+        """Proxy attribute access to :attr:`module`."""
+        raise NotImplementedError()  # pragma: no cover
+
+    @classmethod
+    def get_method(cls, name: str) -> t.Callable:
+        """
+        Return valid :attr:`module` method.
+
+        Args:
+            name: Name of pydash method to get.
+
+        Returns:
+            :attr:`module` callable.
+
+        Raises:
+            InvalidMethod: Raised if `name` is not a valid :attr:`module` method.
+        """
+        method = getattr(cls.module, name, None)
+
+        if not callable(method) and not name.endswith("_"):
+            # Alias method names not ending in underscore to their underscore
+            # counterpart. This allows chaining of functions like "map_()"
+            # using "map()" instead.
+            method = getattr(cls.module, name + "_", None)
+
+        if not callable(method):
+            raise cls.invalid_method_exception(f"Invalid {cls.module.__name__} method: {name}")
+
+        return method
+
+    def __getattr__(self, name: str) -> t.Callable:
+        return self._wrap(self.get_method(name))