feat: add load_extensions method (#1530)

maksims8 · pre-commit-ci[bot] · web-flow · commit f3ba43dc3b6b · 2023-08-14T22:20:36.000-04:00
* feat: add load_extensions method

* chore: add documentation

* ci: correct from checks.

* refactor: change package parameter to *packages

* refactor: remove stray print statement

---------

Co-authored-by: pre-commit-ci[bot] &lt;66853113+pre-commit-ci[bot]@users.noreply.github.com&gt;
diff --git a/interactions/client/client.py b/interactions/client/client.py
@@ -1,9 +1,11 @@
 import asyncio
 import contextlib
 import functools
+import glob
 import importlib.util
 import inspect
 import logging
+import os
 import re
 import sys
 import time
@@ -2002,6 +2004,35 @@ def load_extension(
         module = importlib.import_module(module_name, package)
         self.__load_module(module, module_name, **load_kwargs)
 
+    def load_extensions(
+        self,
+        *packages: str,
+        recursive: bool = False,
+    ) -> None:
+        """
+        Load multiple extensions at once.
+
+        Removes the need of manually looping through the package
+        and loading the extensions.
+
+        Args:
+            *packages: The package(s) where the extensions are located.
+            recursive: Whether to load extensions from the subdirectories within the package.
+        """
+        if not packages:
+            raise ValueError("You must specify at least one package.")
+
+        for package in packages:
+            # If recursive then include subdirectories ('**')
+            # otherwise just the package specified by the user.
+            pattern = os.path.join(package, "**" if recursive else "", "*.py")
+
+            # Find all files matching the pattern, and convert slashes to dots.
+            extensions = [f.replace(os.path.sep, ".").replace(".py", "") for f in glob.glob(pattern, recursive=True)]
+
+            for ext in extensions:
+                self.load_extension(ext)
+
     def unload_extension(
         self, name: str, package: str | None = None, force: bool = False, **unload_kwargs: Any
     ) -> None:
