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
26
27
28
29
30
31
32
33
@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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
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, []

    located = _find_import_line(text, old_module)
    line_no = located[0] if located else 1
    old_line = located[1].strip() if located else ""

    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)

    new_stmt = _format_new_import_stmt(symbols, collector.matched_names, new_module)

    rewrite = CallerRewrite(file="", line=line_no, old=old_line, new=new_stmt)
    return final_tree.code, [rewrite]