Processing generators

log.py

def log(f):
    def wrap(*args, **kwargs):
        total = 0
        t = time.time()
        r = f(*args, **kwargs)
        total += time.time() - t
        if isinstance(r, types.GeneratorType):
            try:
                while 1:
                    t = time.time()
                    yield r.next()
                    total += time.time() - t
            except StopIteration:
                pass
        print total
    return wrap