如何装饰从文件导入的所有函数？

Question

我已经创建了许多功能，这些功能被划分到不同的文件中，现在我想为所有功能应用相同的装饰器，而不需要修改文件，也不需要一个一个地应用装饰器。

我试过用delnan写的this explanation，但是导入函数没有成功。

关于装饰器，每次使用函数参数和值执行类中的函数时，它必须更新一个列表，就像我问的this other question一样。

有什么建议可以帮助我解决这个问题吗？ 谢谢

Answer 1

一点点内省 (dir()) 和 getattr() 和 setattr() 的动态查找。

首先我们遍历模块和check for objects that look like functions 中找到的所有名称。之后，我们只需将旧功能重新分配给修饰过的功能。

ma​​in.py:

import types
import functools

def decorate_all_in_module(module, decorator):
    for name in dir(module):
        obj = getattr(module, name)
        if isinstance(obj, types.FunctionType):
            setattr(module, name, decorator(obj))

def my_decorator(f):
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        print(f)
        return f(*args, **kwargs)
    return wrapper

import mymod1
decorate_all_in_module(mymod1, decorator)

mymod1.py：

def f(x):
    print(x)


def g(x, y):
    print(x + y)

输出：

<function f at 0x101e309d8>
2
<function g at 0x101e30a60>
7

如果您使用星号导入 (from mymod import *)，流程不会那么顺利。原因很简单——因为所有的名字都放在一个大袋子里，而且它们的来源没有区别，你需要很多额外的技巧来找到你想要修补的东西。但是，这就是我们使用命名空间的原因 - because they are one honking great idea。