mapper.py

import ast
import os
import argparse
from typing import Dict, List

class RepoAnalyzer(ast.NodeVisitor):
    def __init__(self):
        self.stats = {"classes": [], "functions": []}

    def visit_ClassDef(self, node: ast.ClassDef):
        self.stats["classes"].append(node.name)
        self.generic_visit(node)

    def visit_FunctionDef(self, node: ast.FunctionDef):
        # Ignore private methods/functions
        if not node.name.startswith('_'):
            self.stats["functions"].append(node.name)
        self.generic_visit(node)

def analyze_file(filepath: str) -> Dict[str, List[str]]:
    with open(filepath, "r", encoding="utf-8") as f:
        try:
            tree = ast.parse(f.read())
            analyzer = RepoAnalyzer()
            analyzer.visit(tree)
            return analyzer.stats
        except (SyntaxError, UnicodeDecodeError):
            return {"classes": [], "functions": []}

def run_mapper(target_dir: str):
    print(f"# Project Map: {os.path.abspath(target_dir)}\n")
    for root, _, files in os.walk(target_dir):
        for file in files:
            if file.endswith(".py"):
                path = os.path.join(root, file)
                rel_path = os.path.relpath(path, target_dir)
                data = analyze_file(path)
                
                if data["classes"] or data["functions"]:
                    print(f"### `{rel_path}`")
                    if data["classes"]:
                        print(f"- **Classes**: {', '.join(data['classes'])}")
                    if data["functions"]:
                        print(f"- **Public Methods**: {', '.join(data['functions'])}")
                    print()

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Generate a Markdown map of a Python repository.")
    parser.add_argument("path", nargs="?", default=".", help="Directory to analyze (default: current)")
    args = parser.parse_args()
    run_mapper(args.path)