timeit_decorator.py

import time
from functools import wraps

def timeit(comment=None):
    """
        auto establish and close mongo connection
    """
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            begintime = time.time()
            ret = func(*args,**kwargs)
            duration = (time.time() - begintime).total_seconds()/60
            msg = '[{}] {}: duration (min) = {}'.format(begintime.strftime('%Y-%m-%d'), func.__name__, duration)
            
            if comment is not None:
                print '<{}>'.format(comment)
            print msg
            return ret
        return wrapper
    return decorator