Python profiling decorator

profile_decorator.py

from cProfile import Profile
from functools import wraps
import pstats


def profiled(f):
    """
    Decorator for profiling a function

    To use:
    - decorate function
    - execute the code path
    - browse the results via `python -m pstats /tmp/weatherhub_profiling`
       (docs https://docs.python.org/3/library/profile.html ... try:
       `strip`, `sort time`, `stats 10`)
    """
    @wraps(f)
    def _wrapper(*args, **kwargs):
        pr = Profile()
        pr.enable()
        try:
            result = f(*args, **kwargs)
        except Exception:  # pylint: disable=broad-except
            raise
        finally:
            pr.disable()
        ps = pstats.Stats(pr)
        ps.dump_stats("/tmp/profiling")
        return result
    return _wrapper