Python -- 装饰器工厂

原创已于 2022-06-22 15:37:23 修改 · 2.9w 阅读 ·

本内容遵循CC 4.0 BY-SA版权协议

是大嘟嘟呀

关注

标签

#python #开发语言 #django

分类数据科学与机器学习开发

代码可运行

于 2022-06-14 15:42:45 首次发布

Python高级专栏收录该内容

32 篇文章

订阅专栏

本文介绍了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