aboutsummaryrefslogtreecommitdiffstats
path: root/arthur/extensions.py
diff options
context:
space:
mode:
Diffstat (limited to 'arthur/extensions.py')
-rw-r--r--arthur/extensions.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/arthur/extensions.py b/arthur/extensions.py
new file mode 100644
index 0000000..2170d68
--- /dev/null
+++ b/arthur/extensions.py
@@ -0,0 +1,15 @@
+"""Utilities for working with extensions."""
+from pathlib import Path
+from typing import Generator
+
+
+def find_extensions() -> Generator[tuple[Path, str], None, None]:
+ """Search the exts directory to find cogs to load."""
+ for path in Path("arthur/exts").rglob("**/*.py"):
+ # Convert a path like "arthur/exts/foo/bar.py" to "arthur.exts.foo.bar"
+ yield path, path_to_module(path)
+
+
+def path_to_module(path: Path) -> str:
+ """Convert a path like "arthur/exts/foo/bar.py" to "arthur.exts.foo.bar"."""
+ return str(path.parent.as_posix()).replace("/", ".") + f".{path.stem}"