python - 装饰外部(库)函数的正确方法是什么？

Question

我在我的代码中多次使用了一个库函数来测试通过/失败条件并相应地执行不同的代码，但由于某种原因没有它找到的结果的返回值；我想用装饰器添加它，以便我可以在我的代码中调用它。

鉴于我无法编辑源文件，正确的做法是什么？

我应该这样做吗:

def test_pass(param1, param2):
    external_function(param1, param2)
    if(...):
        return False
    else:
        return True

或者有没有办法使用漂亮的@decorator 语法？

Answer 1

用@decorator 装饰是语法糖；函数对象被 decorator(orig_function) 调用返回的任何内容替换。

对于外部函数，您只需使用您编写的包装器即可；您只能在定义原始函数时使用装饰器语法。