LineProfiler decorator

profile.py

from functools import wraps

from line_profiler import LineProfiler


def profile(follow=[]):
    def decorator(func):
        @wraps(func)
        def profiled_func(*args, **kwargs):
            try:
                profiler = LineProfiler()
                profiler.add_function(func.__wrapped__)
                for f in follow:
                    profiler.add_function(f)
                profiler.enable_by_count()
                return func(*args, **kwargs)
            finally:
                profiler.print_stats()
                profiler.dump_stats("my_profiling_results.lprof")

        return profiled_func

    return decorator