Add __all__ to top-level modules

refactor/__init__.py

@@ -8,6 +8,18 @@
 )
 from refactor.runner import run
 
+__all__ = [
+    "Configuration",
+    "Context",
+    "Representative",
+    "Action",
+    "NewStatementAction",
+    "ReplacementAction",
+    "Rule",
+    "Session",
+    "run",
+]
+
 
 def _check_asserts():
     import sys

refactor/ast.py

@@ -9,6 +9,15 @@
 
 from refactor import common
 
+__all__ = [
+    "BaseUnparser",
+    "Lines",
+    "PreciseUnparser",
+    "UNPARSER_BACKENDS",
+    "Unparser",
+    "split_lines",
+]
+
 
 @dataclass
 class Lines(UserList):

refactor/change.py

@@ -5,6 +5,8 @@
 
 from refactor.ast import split_lines
 
+__all__ = ["Change"]
+
 
 @dataclass
 class Change:

refactor/common.py

@@ -16,6 +16,23 @@
     cast,
 )
 
+__all__ = [
+    "Singleton",
+    "apply_condition",
+    "compare_ast",
+    "find_closest",
+    "find_indent",
+    "get_source_segment",
+    "has_positions",
+    "is_truthy",
+    "negate",
+    "pascal_to_snake",
+    "position_for",
+    "unpack_lhs",
+    "walk_scope",
+    "wrap_with_parens",
+]
+
 
 def negate(node: ast.expr) -> ast.UnaryOp:
     """Negate the given `node`."""

refactor/context.py

@@ -25,6 +25,18 @@
 import refactor.common as common
 from refactor.ast import UNPARSER_BACKENDS, BaseUnparser
 
+__all__ = [
+    "Ancestry",
+    "Configuration",
+    "Context",
+    "Dependable",
+    "Representative",
+    "Scope",
+    "ScopeInfo",
+    "ScopeType",
+    "resolve_dependencies",
+]
+
 
 @dataclass
 class Configuration:

refactor/core.py

@@ -19,6 +19,15 @@
     resolve_dependencies,
 )
 
+__all__ = [
+    "Action",
+    "NewStatementAction",
+    "ReplacementAction",
+    "Rule",
+    "Session",
+    "TargetedNewStatementAction",
+]
+
 
 @dataclass
 class Action:

refactor/runner.py

@@ -18,6 +18,8 @@
 
 from refactor.core import Session
 
+__all__ = ["dump_stats", "expand_paths", "run", "run_files", "unbound_main"]
+
 _DEFAULT_WORKERS = object()
 
 

refactor/validate_inputs.py

@@ -1,6 +1,8 @@
 from argparse import Namespace
 from pathlib import Path
 
+__all__ = ["validate_main_inputs"]
+
 _DEFAULT_FILES = [Path("refactors.py"), Path(".refactors/__init__.py")]