Python -- 装饰器工厂

说明

• 装饰器工厂就是让装饰器可以携带参数
  
  

使用演示

不加装饰器

def demo():
    print('demo执行')


demo()

加上装饰器

• 函数执行前后分别多输出两行

@dazzling(n=10)  # 输出10个=
def demo():
    print('demo执行')


demo()

• 函数执行异常则自动重试

@auto_retry(times=3)  # 最多重试3次, 函数会执行4次
def demo():
    raise Exception('自定义错误')


demo()

• 记录函数执行时间

@record_time(keep=4)  # 保留4位小数
def demo():
    print('哈哈哈')
    time.sleep(1)


demo()

实现过程

# 耀眼显示, 默认输出100个=
def dazzling(n=100):
    def outer(func):
        def inner(*args, **kwargs):
            print('\n')
            print('=' * n)
            result = func(*args, **kwargs)
            print('=' * n)
            print('\n')
            return result

        return inner

    return outer


# 自动重试, 默认重试2次
def auto_retry(times=2):
    def outer(func):
        def inner(*args, **kwargs):
            for i in range(times + 1):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    print('ERROR  {}执行错误  {}'.format(func.__name__, e))

        return inner

    return outer


# 记录消耗的时间, 默认保留小数点后两位
def record_time(keep=2):
    def outer(func):
        def inner(*args, **kwargs):
            t1 = time.time()
            result = func(*args, **kwargs)
            print(f'耗时{time.time() - t1:.{keep}f}秒')
            return result

        return inner

    return outer