Skip to content

Callers

callers

Caller rewriting: redirect from old_module import Symbol to new_module.

CallerRewrite dataclass

A single caller-import rewrite record for :class:MovePlan.

Source code in packages/axm-anvil/src/axm_anvil/core/callers.py
Python
22
23
24
25
26
27
28
29
@dataclass
class CallerRewrite:
    """A single caller-import rewrite record for :class:`MovePlan`."""

    file: str
    line: int
    old: str
    new: str

rewrite_caller_text(text, old_module, new_module, symbols)

Rewrite from old_module import <symbols> to new_module.

Returns (new_text, rewrites). When no matching import exists the original text and an empty list are returned unchanged.

Source code in packages/axm-anvil/src/axm_anvil/core/callers.py
Python
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
def rewrite_caller_text(
    text: str,
    old_module: str,
    new_module: str,
    symbols: Sequence[str],
) -> tuple[str, list[CallerRewrite]]:
    """Rewrite ``from old_module import <symbols>`` to ``new_module``.

    Returns ``(new_text, rewrites)``. When no matching import exists the
    original text and an empty list are returned unchanged.
    """
    moved = set(symbols)
    tree = cst.parse_module(text)

    collector = _CollectOldImport(old_module, moved)
    tree.visit(collector)
    if not collector.matched_names:
        return text, []

    new_tree = tree.visit(_RewriteOldImport(old_module, moved))

    context = _add_new_imports(symbols, collector.matched_names, new_module)
    final_tree = AddImportsVisitor(context).transform_module(new_tree)

    located_lines = _find_import_lines(text, old_module, moved)
    rewrites = _build_caller_rewrites(
        located_lines, symbols, collector.matched_names, new_module
    )
    return final_tree.code, rewrites