Find Unused Python Packages Automatically

import pkg_resources
import ast
import os
import sys
from pathlib import Path

def find_imports_in_project(project_dir="."):
    imports = set()
    for py_file in Path(project_dir).rglob("*.py"):
        try:
            with open(py_file, "r") as f:
                tree = ast.parse(f.read())
            for node in ast.walk(tree):
                if isinstance(node, ast.Import):
                    for alias in node.names:
                        imports.add(alias.name.split(".")[0])
                elif isinstance(node, ast.ImportFrom):
                    if node.module:
                        imports.add(node.module.split(".")[0])
        except (SyntaxError, UnicodeDecodeError):
            pass
    return imports

def get_installed_packages():
    return {pkg.key for pkg in pkg_resources.working_set}

def find_unused_packages(project_dir="."):
    installed = get_installed_packages()
    used = find_imports_in_project(project_dir)
    unused = installed - used
    return sorted(unused)

if __name__ == "__main__":
    unused = find_unused_packages()
    if unused:
        print("Unused packages found:")
        for pkg in unused:
            print(f"  - {pkg}")
    else:
        print("No unused packages detected.")