python - 如何包装从模块导入的所有函数？

Question

是否可以将代码从导入的模块应用到导入它的模块？例如，我有一个调试模块，其中定义了一些用于调试的装饰器，例如:

def debug_func(f):
    def wrapper(*func_args, **func_kwargs):
        print(f(*func_args, **func_kwargs))
    return wrapper

想法是什么:如果我能的话会很有用

import Debug 

当前模块中的所有函数都将用装饰器包装。可能吗？

Answer 1

在 Debug.py 中:

import functools
from types import FunctionType

def wrap_functions(module_dict):
    for k, f in module_dict.items():
        if not isinstance(f, FunctionType):
            continue
        def get_wrapper(f):
            def wrapper(*func_args, **func_kwargs):
                print(f(*func_args, **func_kwargs))
            return functools.wraps(f)(wrapper)
        module_dict[k] = get_wrapper(f)

在您希望调试的模块的底部:

import Debug
Debug.wrap_functions(globals())

感谢 (+1) 所有提供建议的评论者。