Profiling

thread_profiling.py

import cProfile
import threading


def _run_profiler(run):
	"""Use as a decorator for Thread profiling ONLY"""
	def wrapper(self):
		profiler = cProfile.Profile()
		profiler.enable()
		run(self)
		profiler.disable()
		try:
			return profiler.runcall(threading.Thread.run, self)
		finally:
			profiler.dump_stats('myprofile-%d.profile' % (self.ident,))
	return wrapper

from functools import wraps

def profile(output_file="profile.prof"):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            profiler = cProfile.Profile()
            profiler.enable()

            result = func(*args, **kwargs)

            profiler.disable()
            profiler.dump_stats(output_file)

            return result
        return wrapper
    return decorator