Auto-open HTML output in browser after generation

Automatically open flamegraph and heatmap HTML output in the default
browser after profiling completes. Add --no-browser flag to disable
this behavior.

Author: ivonastojanovic

Lib/profiling/sampling/cli.py

@@ -10,6 +10,7 @@
 import subprocess
 import sys
 import time
+import webbrowser
 from contextlib import nullcontext
 
 from .errors import SamplingUnknownProcessError, SamplingModuleNotFoundError, SamplingScriptNotFoundError
@@ -492,6 +493,11 @@ def _add_format_options(parser, include_compression=True, include_binary=True):
         help="Output path (default: stdout for pstats, auto-generated for others). "
         "For heatmap: directory name (default: heatmap_PID)",
     )
+    output_group.add_argument(
+        "--no-browser",
+        action="store_true",
+        help="Disable automatic browser opening for HTML output (flamegraph, heatmap)",
+    )
 
 
 def _add_pstats_options(parser):
@@ -591,6 +597,32 @@ def _generate_output_filename(format_type, pid):
     return f"{format_type}_{pid}.{extension}"
 
 
+def _open_in_browser(path):
+    """Open a file or directory in the default web browser.
+
+    Args:
+        path: File path or directory path to open
+
+    For directories (heatmap), opens the index.html file inside.
+    """
+    abs_path = os.path.abspath(path)
+
+    # For heatmap directories, open the index.html file
+    if os.path.isdir(abs_path):
+        index_path = os.path.join(abs_path, 'index.html')
+        if os.path.exists(index_path):
+            abs_path = index_path
+        else:
+            print(f"Warning: Could not find index.html in {path}")
+            return
+
+    file_url = f"file://{abs_path}"
+    try:
+        webbrowser.open(file_url)
+    except Exception as e:
+        print(f"Warning: Could not open browser: {e}")
+
+
 def _handle_output(collector, args, pid, mode):
     """Handle output for the collector based on format and arguments.
 
@@ -630,6 +662,10 @@ def _handle_output(collector, args, pid, mode):
             filename = args.outfile or _generate_output_filename(args.format, pid)
         collector.export(filename)
 
+        # Auto-open browser for HTML output unless --no-browser flag is set
+        if args.format in ('flamegraph', 'heatmap') and not getattr(args, 'no_browser', False):
+            _open_in_browser(filename)
+
 
 def _validate_args(args, parser):
     """Validate format-specific options and live mode requirements.
@@ -1153,6 +1189,10 @@ def progress_callback(current, total):
             filename = args.outfile or _generate_output_filename(args.format, os.getpid())
             collector.export(filename)
 
+            # Auto-open browser for HTML output unless --no-browser flag is set
+            if args.format in ('flamegraph', 'heatmap') and not getattr(args, 'no_browser', False):
+                _open_in_browser(filename)
+
         print(f"Replayed {count} samples")