aboutsummaryrefslogtreecommitdiffstats
path: root/arthur/extensions.py
blob: 2170d6891d2334848f2a42d7fd91fcb51c8f57f9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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}"