ml_switcheroo.core.import_fixer

Import Fixer Package.

This package provides the ImportFixer class, a LibCST transformer responsible for: 1. Injection: Adding required imports for the target framework. 2. Pruning: Removing unused imports from the source framework. 3. Refinement: Collapsing aliases and cleaning up re-exports.

It uses ImportResolver to pre-calculate necessary changes before transformation.

Submodules

• ml_switcheroo.core.import_fixer.attributes_mixin
• ml_switcheroo.core.import_fixer.base
• ml_switcheroo.core.import_fixer.imports_mixin
• ml_switcheroo.core.import_fixer.injection_mixin
• ml_switcheroo.core.import_fixer.resolution
• ml_switcheroo.core.import_fixer.utils

Classes

ImportResolver
ResolutionPlan	The strategy for the ImportFixer to execute.
ImportReq	Represents a normalized import requirement.
ImportFixer	Composite Transformer for managing imports and namespacing.

Package Contents

class ml_switcheroo.core.import_fixer.ImportResolver(semantics: ml_switcheroo.semantics.manager.SemanticsManager)

semantics

resolve(tree: libcst.Module, target_fw: str) → ResolutionPlan

class ml_switcheroo.core.import_fixer.ResolutionPlan

The strategy for the ImportFixer to execute.

required_imports: List[ImportReq] = []

mappings: Dict[str, ImportReq]

path_to_alias: Dict[str, str]

class ml_switcheroo.core.import_fixer.ImportReq

Represents a normalized import requirement. Can represent import module as alias or from module import sub as alias.

module: str

subcomponent: str | None = None

alias: str | None = None

property signature: str

Unique signature for deduplication.

class ml_switcheroo.core.import_fixer.ImportFixer

Bases: attributes_mixin.AttributeMixin, imports_mixin.ImportMixin, injection_mixin.InjectionMixin, base.BaseImportFixer

Composite Transformer for managing imports and namespacing.

Inherits functionality from: - AttributeMixin: simplifying dotted attribute access. - ImportMixin: rewriting import statements using the plan. - InjectionMixin: injecting missing top-level imports from the plan. - BaseImportFixer: State management and configuration.