calculate python function time

timeit.py

class Timeit(object):
    def __init__(self, tag):
        self.tag = tag

    def __enter__(self):
        self.start = time.time()

    def __exit__(self, *unused):
        self.cost = time.time() - self.start
        logger.info("%s took about: %ss" % (self.tag, self.cost))


def exec_profile(tag):
    def warpper(func):
        def new(*args, **kwds):
            with Timeit(tag):
                return func(*args, **kwds)

        return new

    return warpper
  
@exec_profile("sync device")
def syncer():
    with self.mc.cursor() as cursor:
        cursor.execute("select hash from waf where hash IS NOT NULL")
        self.devices = map(operator.itemgetter("hash"), cursor.fetchall())
    logger.info("current device number: {0}".format(self.devices))