simple profiling decorator with an optional comment string

profile_it.py

import functools
import logging
import time

logging.basicConfig(level=logging.INFO)


def profile(comment=""):
    def time_part(f):
        @functools.wraps(f)
        def w(*args, **kwargs):
            start = time.time()
            result = f(*args, **kwargs)
            dur = time.time() - start
            name = f.__name__
            logging.info(" {2} {0} : {1}".format(name, dur, comment))
            return result
        return w
    return time_part


@profile("this is a one")
def one():
    time.sleep(1)
    return 1


@profile("this is a two")
def two():
    time.sleep(2)
    return 2

one()
two()