pybind11-stubgen

import os import subprocess import shutil # To generate stubs for a pybind11 module named 'my_module': # 1. Ensure 'my_module' is installed and importable in your Python environment. # 2. This example demonstrates generating stubs for the 'numpy' module. # (Requires numpy to be installed: `pip install numpy`) # Create a temporary directory for output output_dir = "temp_stubs" os.makedirs(output_dir, exist_ok=True) try: print(f"Generating stubs for 'numpy' into '{output_dir}'...") # Execute the command-line tool result = subprocess.run( ["pybind11-stubgen", "numpy", "-o", output_dir], capture_output=True, text=True, check=True ) print("STDOUT:", result.stdout) print("STDERR:", result.stderr) print(f"\nStubs generated in '{output_dir}'. Contents:") for root, _, files in os.walk(output_dir): for f in files: print(os.path.join(root, f)) except subprocess.CalledProcessError as e: print(f"Error generating stubs: {e}") print("STDOUT:", e.stdout) print("STDERR:", e.stderr) except FileNotFoundError: print("Error: 'pybind11-stubgen' command not found. Is it installed and in your PATH?") except Exception as e: print(f"An unexpected error occurred: {e}") finally: # Clean up the temporary directory if os.path.exists(output_dir): print(f"\nCleaning up '{output_dir}'...") shutil.rmtree(output_dir)