Python time profiling

benchmark.py

#!python
# -*- coding: utf-8 -*-
import importlib
import os
from timeit import default_timer as timer
import logging

from my_module import config

class benchmark(object):
    logger = config.getLogger()

    def __init__(self, msg, fmt="%0.3g"):
        self.msg = msg
        self.fmt = fmt

    def __enter__(self):
        self.start = timer()
        return self

    def __exit__(self, *args):
        t = timer() - self.start
        logger.info(("%s : " + self.fmt + " seconds") % (self.msg, t))
        self.time = t

if __name__ == '__main__':
    with benchmark('time executing my function') as b:
        foo()