How to Generate a Dependency Graph for Python Projects

import os
import ast
from pathlib import Path
from collections import defaultdict

def get_imports(filepath):
    with open(filepath) as f:
        try:
            tree = ast.parse(f.read())
        except SyntaxError:
            return []
    imports = []
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            for alias in node.names:
                imports.append(alias.name.split('.')[0])
        elif isinstance(node, ast.ImportFrom):
            if node.module:
                imports.append(node.module.split('.')[0])
    return imports

def build_dependency_graph(project_dir):
    dependencies = defaultdict(set)
    project_path = Path(project_dir)
    
    for pyfile in project_path.rglob("*.py"):
        module = pyfile.stem
        for imp in get_imports(pyfile):
            dependencies[module].add(imp)
    
    for module, deps in dependencies.items():
        if deps:
            print(f"{module}: {', '.join(sorted(deps))}")

if __name__ == "__main__":
    build_dependency_graph(".")