我可以在包装函数之前修补Python装饰器吗?

发布于 2021-01-29 19:35:16

我有一个带有装饰器的函数,我正在Python
Mock库的帮助下进行测试。我想mock.patch用一个仅调用函数的模拟“
bypass”装饰器代替真正的装饰器。

我不知道的是如何在真正的装饰器包装功能之前应用补丁。我在补丁目标上尝试了几种不同的变体,并对补丁和导入语句重新排序,但均未成功。有任何想法吗?

关注者
0
被浏览
94
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    装饰器在函数定义时应用。对于大多数功能,这是模块加载时的时间。(在其他函数中定义的函数会在每次调用封闭函数时应用装饰器。)

    因此,如果您想用猴子修补装饰器,您需要做的是:

    1. 导入包含它的模块
    2. 定义模拟装饰器功能
    3. 设置 例如 module.decorator = mymockdecorator
    4. 导入使用装饰器的模块,或在您自己的模块中使用它

    如果包含装饰器的模块也包含使用该装饰器的功能,则在您看到它们时它们已经被装饰了,您可能就是SOL

    自从我最初编写此代码以来,进行编辑以反映对Python的更改:如果装饰器使用functools.wraps()并且Python版本足够新,则可以使用__wrapped__属性来挖掘原始函数并重新装饰它,但这绝不是保证,并且您要替换的装饰器也可能不是唯一应用的装饰器。



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看