python操作带参的装饰器

python操作带参的装饰器
python操作带参的装饰器

1、说明

装饰函数的第一个参数是装饰func,和以前一样。

另一个参数timelimit是用位置参数写的,有默认值。

和原来一样使用了可变参数的写法。

2、实例

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

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注