实现一个重运行的装饰器，可以用来装饰任何一个功能函数，只要被装饰的函数执行出现AssertionError，则重新执行该函数，同一个函数最多重运行三次

AllenBright

于 2023-04-24 21:07:22 发布

阅读量113

点赞数

分类专栏：试题练习文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_39033358/article/details/130352191

版权

试题练习专栏收录该内容

70 篇文章

订阅专栏

该代码定义了一个装饰器`rerun`，用于包装函数。如果被包装的函数在执行时抛出`AssertionError`异常，会在尝试3次后仍然失败时将异常抛出。在每次尝试之间，会打印执行出错啦，重新执行。当没有错误发生时，输出执行通过并返回函数结果。示例中，`work2`函数使用了这个装饰器，期望用户输入的结果为abc。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

def rerun(func):
    def wrapper(*args, **kwargs):
        for i in range(4):
            try:
                res = func(*args, **kwargs)
            except AssertionError as e:
                if i == 3:
                    raise e
                print("执行出错啦，重新执行")
            else:
                print('执行通过')
                return res
    return wrapper


@rerun
def work2():
    expected = 'abc'
    res = input('请输入实际结果：')
    assert expected == res


work2()