utils.py

# -*- encoding: utf-8

import time, os, cPickle
from termcolor import colored

def chrono(method):

    def timed(*args, **kw):
        start = time.time()
        result = method(*args, **kw)

        print colored("  {} done in {:.3f} sec".format(method.__name__, time.time() - start), "cyan")

        return result

    return timed