python操作带参的装饰器

from decorator import decorator

@decorator

def warn_slow(func, timelimit=60, *args, **kw):

    t0 = time.time()

    result = func(*args, **kw)

    dt = time.time() - t0

    if dt > timelimit:

        logging.warn('%s took %d seconds', func.__name__, dt)

    else:

        logging.info('%s took %d seconds', func.__name__, dt)

    return result

@warn_slow(timelimit=600)  # warn if it takes more than 10 minutes

def run_calculation(tempdir, outdir):

    pass