|
import os |
|
import ast |
|
import json |
|
import sys |
|
from rich.console import Console |
|
from rich.table import Table |
|
from rich.progress import Progress |
|
|
|
console = Console(stderr=True) |
|
|
|
|
|
def get_imported_modules(filename): |
|
with open(filename, "r") as file: |
|
tree = ast.parse(file.read()) |
|
|
|
imported_modules = [] |
|
for node in ast.walk(tree): |
|
if isinstance(node, ast.Import): |
|
imported_modules.extend( |
|
(alias.name, alias.name in sys.builtin_module_names) |
|
for alias in node.names |
|
) |
|
elif isinstance(node, ast.ImportFrom): |
|
if node.module: |
|
imported_modules.append( |
|
(node.module, node.module in sys.builtin_module_names) |
|
) |
|
|
|
return imported_modules |
|
|
|
|
|
def list_imported_modules(folder): |
|
modules = [] |
|
|
|
file_count = sum(len(files) for _, _, files in os.walk(folder)) |
|
progress = Progress() |
|
|
|
task = progress.add_task("[cyan]Scanning files...", total=file_count) |
|
|
|
for root, _, files in os.walk(folder): |
|
for file in files: |
|
if file.endswith(".py"): |
|
file_path = os.path.join(root, file) |
|
imported_modules = get_imported_modules(file_path) |
|
modules.extend(imported_modules) |
|
progress.update(task, advance=1) |
|
|
|
progress.stop() |
|
|
|
return modules |
|
|
|
|
|
if __name__ == "__main__": |
|
if len(sys.argv) < 2: |
|
console.print( |
|
"[bold red]Please provide the folder path as a command-line argument.[/bold red]" |
|
) |
|
sys.exit(1) |
|
|
|
|
|
|
|
|
|
|
|
folder_path = sys.argv[1] |
|
if not os.path.isdir(folder_path): |
|
console.print("[bold red]Invalid folder path![/bold red]") |
|
sys.exit(1) |
|
|
|
console.print("[bold green]=== Python Imported Modules ===[/bold green]\n") |
|
console.print(f"Scanning folder: [bold]{folder_path}[/bold]\n") |
|
|
|
imported_modules = list_imported_modules(folder_path) |
|
|
|
console.print(f"\n[bold green]Imported Modules:[/bold green]\n") |
|
|
|
table = Table(show_header=True, header_style="bold cyan") |
|
table.add_column("Module") |
|
table.add_column("Type") |
|
|
|
for module, is_builtin in imported_modules: |
|
module_type = "Built-in" if is_builtin else "External" |
|
table.add_row(module, module_type) |
|
|
|
console.print(table) |
|
|
|
json_data = json.dumps( |
|
[ |
|
{"module": module, "type": "Built-in" if is_builtin else "External"} |
|
for module, is_builtin in imported_modules |
|
], |
|
indent=4, |
|
) |
|
print(json_data) |
|
|
