create a python script that rename import modules using AST

imch.py

import ast
import os

def rename_import_modules(source_file, old_module_name, new_module_name):
    # Parse the source code into an Abstract Syntax Tree (AST)
    with open(source_file) as f:
        source = f.read()
    tree = ast.parse(source)

    # Traverse the AST to find import statements and update the module name
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            for alias in node.names:
                if alias.name == old_module_name:
                    alias.name = new_module_name

    # Write the updated code to a new file
    new_file = os.path.splitext(source_file)[0] + '_new.py'
    with open(new_file, 'w') as f:
        f.write(ast.unparse(tree))

# Example usage:
rename_import_modules('example.py', 'old_module_name', 'new_module_name')

create a python script that rename import packages using AST

import ast
import os

def rename_import_packages(source_file, old_package_name, new_package_name):
    # Parse the source code into an Abstract Syntax Tree (AST)
    with open(source_file) as f:
        source = f.read()
    tree = ast.parse(source)

    # Traverse the AST to find import statements and update the package name
    for node in ast.walk(tree):
        if isinstance(node, ast.ImportFrom):
            if node.module.startswith(old_package_name):
                node.module = node.module.replace(old_package_name, new_package_name)

    # Write the updated code to a new file
    new_file = os.path.splitext(source_file)[0] + '_new.py'
    with open(new_file, 'w') as f:
        f.write(ast.unparse(tree))

# Example usage:
rename_import_packages('example.py', 'old_package_name', 'new_package_name')