Python能否像Java ByteBuddy一样实现运行时指定方法钩子？

一、直接引用方法的钩子实现 #

Python里函数是一等对象，直接替换原函数的指向就能实现运行时钩子，完全不用装饰器。针对你给出的示例，我们可以实现set_hook函数：

def method(s):  # 避免用str做参数名，和内置类型冲突
    print(s)
    
def new_method(origin_method, *args, **kwargs):
    print("new_method")
    origin_method(*args, **kwargs)

def set_hook(target_func, hook_func):
    # 保存原函数的引用
    original = target_func
    # 定义包装函数
    def wrapper(*args, **kwargs):
        hook_func(original, *args, **kwargs)
    # 替换原函数的指向
    globals()[target_func.__name__] = wrapper

if __name__ == '__main__':
    set_hook(method, new_method)
    method("a")

运行后输出：

new_method
a

原理就是把method变量重新指向包装后的wrapper函数，后续调用method时会先执行钩子逻辑，再调用原函数。

二、通过函数名称实现钩子 #

如果无法直接引用方法，可通过解析名称字符串定位到目标函数/方法，再替换其引用：

class User:
    def method(self):
        print("User.method called")

def method(s):
    print(s)

def new_method(origin_method, *args, **kwargs):
    print("new_method")
    origin_method(*args, **kwargs)

def setting_hook(name_str, hook_func):
    # 拆分名称，处理类方法和模块函数
    if "." in name_str:
        class_name, method_name = name_str.split(".", 1)
        # 从全局变量中获取类
        cls = globals().get(class_name)
        if not cls:
            raise ValueError(f"Class {class_name} not found")
        # 获取原方法
        original_method = getattr(cls, method_name)
        # 定义包装函数
        def wrapper(*args, **kwargs):
            hook_func(original_method, *args, **kwargs)
        # 替换类的方法
        setattr(cls, method_name, wrapper)
    else:
        # 处理模块级函数
        original_func = globals().get(name_str)
        if not original_func:
            raise ValueError(f"Function {name_str} not found")
        def wrapper(*args, **kwargs):
            hook_func(original_func, *args, **kwargs)
        globals()[name_str] = wrapper

if __name__ == '__main__':
    setting_hook("User.method", new_method)
    setting_hook("method", new_method)
    
    User().method()
    method("a")

运行输出：

new_method
User.method called
new_method
a

这段代码会解析名称字符串，区分模块函数和类方法，替换对应的引用实现钩子。

三、字节码级别的修改（类似ByteBuddy） #

如果需要更底层的字节码修改（而非替换函数引用），Python也支持：

• 可以使用bytecode、codetransformer这类库直接操作函数的字节码对象
• 也可以通过inspect模块获取函数的__code__属性，手动修改字节码指令（但需要熟悉Python字节码指令集，复杂度较高）

比如简单修改字节码替换打印内容的例子（简化版）：

import bytecode

def original_func():
    print("original")

# 解析字节码
code = bytecode.Bytecode.from_code(original_func.__code__)
# 找到打印指令，替换字符串常量
for instr in code:
    if isinstance(instr, bytecode.Instr) and instr.name == "LOAD_CONST":
        if instr.arg == "original":
            instr.arg = "modified by bytecode"
# 替换原函数的字节码
original_func.__code__ = code.to_code()

original_func()  # 输出：modified by bytecode

这种方式直接修改函数的字节码，和ByteBuddy操作Java字节码的思路类似，但日常场景中替换函数引用的方式已经足够满足大部分需求，字节码修改更多用于底层工具开发。

内容的提问来源于stack exchange，提问作者xinlin hou