py-spy 0.4.2

Sampling profiler for Python programs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use log::error;

// Simple example of showing how to use the rust API to
// print out stack traces from a python program

fn print_python_stacks(pid: remoteprocess::Pid) -> Result<(), anyhow::Error> {
    // Create a new PythonSpy object with the default config options
    let config = py_spy::Config::default();
    let mut process = py_spy::PythonSpy::new(pid, &config)?;

    // get stack traces for each thread in the process
    let traces = process.get_stack_traces()?;

    // Print out the python stack for each thread
    for trace in traces {
        println!("Thread {:#X} ({})", trace.thread_id, trace.status_str());
        for frame in &trace.frames {
            println!("\t {} ({}:{})", frame.name, frame.filename, frame.line);
        }
    }
    Ok(())
}

fn main() {
    env_logger::init();
    let args: Vec<String> = std::env::args().collect();
    let pid = if args.len() > 1 {
        args[1].parse().expect("invalid pid")
    } else {
        error!("you must specify a pid!");
        return;
    };

    if let Err(e) = print_python_stacks(pid) {
        error!("failed to print stack traces: {:?}", e);
    }
}