pyrasite script to dump thread tracebacks, merging those with identical tracebacks

dump_tracebacks.py

# pyrasite script to dump all thread tracebacks, grouping
# together threads with identical tracebacks.

#  pyrasite <pid> dump_tracebacks.py

# The report is appended to /tmp/report.txt

import os
import traceback
import sys
import threading

id2thread = {}
for thread in threading.enumerate():
    id2thread[thread.ident] = thread

traces = {}
for thread_id, frame in sys._current_frames().items():
    t = id2thread.get(thread_id, None)
    thread_name = t.name if t is not None else "UNKNOWN"
    thread_description = "Thread: %s (%d)" % (thread_name, thread_id)
    stack_trace = "".join(traceback.format_list(traceback.extract_stack(frame)))
    if stack_trace in traces:
        traces[stack_trace].append(thread_description)
    else:
        traces[stack_trace] = [thread_description]

with open("/tmp/report.txt", 'a') as s:
    def p(x):
        s.write(x)
        s.write("\n")
        return None
    p("Begin stack dump of process %d:" % os.getpid())
    p("Note: Threads with identical tracebacks are grouped together.")
    for trace, thread_descriptions in traces.iteritems():
        p("")
        map(p, thread_descriptions)
        p("")
        p(trace)
        p("")
    p("End stack dump of process %d" % os.getpid())