A line by line profiler for Python

profiler.py

# Adapted from https://zapier.com/engineering/profiling-python-boss/
# Requires https://pypi.org/project/line_profiler/

def profile_func(follow=[]):
    from line_profiler import LineProfiler
    # simply decorate the function you want profiled
    # add any functions called you want followed to []
    def inner(func):
        def profiled_func(*args, **kwargs):
            profiler = LineProfiler()
            try:
                profiler.add_function(func)
                for f in follow:
                    profiler.add_function(f)
                profiler.enable_by_count()
                return func(*args, **kwargs)
            finally:
                profiler.print_stats()
        return profiled_func
    return inner