chore: add more __all__/__dir__

[henryiii]

Followup to #1011, filling out more all/dir.

I was going to do this one at a time, but quite a few already had __all__, so I did it all at once. Here's the script I used to help:

import argparse
import ast
from pathlib import Path


def generate_all_from_ast(path: Path) -> list[str]:
    tree = ast.parse(path.read_text())
    names: set[str] = set()

    for node in tree.body:
        match node:
            # def foo(...) or class Foo:
            case (
                ast.FunctionDef(name=name)
                | ast.AsyncFunctionDef(name=name)
                | ast.ClassDef(name=name)
            ) if not name.startswith("_"):
                names.add(name)

            # x = ...
            case ast.Assign(targets=targets) | ast.AnnAssign(targets=targets):
                for target in targets:
                    match target:
                        case ast.Name(id=name) if not name.startswith("_"):
                            names.add(name)
            case (
                ast.Assign(target=ast.Name(id=name))
                | ast.AnnAssign(target=ast.Name(id=name))
            ) if not name.startswith("_"):
                names.add(name)

            # from module import a, b as c
            case ast.ImportFrom(module=_, level=0, names=imported):
                for entry in imported:
                    match entry:
                        case ast.alias(name=name, asname=asname) if (
                            asname is not None and not asname.startswith("_")
                        ):
                            names.add(asname)

            # Ignore everything else
            case _:
                pass

    return sorted(names)


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Generate a __all__ list for a Python source file"
    )
    parser.add_argument(
        "file",
        type=Path,
        help="Python source file to analyze",
    )
    args = parser.parse_args()

    names = generate_all_from_ast(args.file)

    print("__all__ = [")
    for name in names:
        print(f'    "{name}",')
    print("]")


if __name__ == "__main__":
    main()

[henryiii]

This was noticed by pylance, not a directly related change, though.