python Monkey通过装饰器在函数中修补对象的方法

Question

我正在尝试学习一些高级装饰器的用法。具体来说，我正在尝试通过函数中的装饰器来修补类的方法。

这是一个基本示例来说明我正在尝试做的事情。我有一个函数something 可以做一些事情；在该函数中有一个类的实例。那个实例我想猴子补丁。

from functools import update_wrapper

class Foobar:
    def get_something(self):
        return "apple"

class FakeFoobar:
    def get_something(self):
        return "orange"

class my_decorator:
    def __init__(self, original_function):
        self._original_function = original_function
        update_wrapper(self, original_function)

    def __call__(self, *args, **kwargs):
        # some magic here?
        return self._original_function(*args, **kwargs)

@my_decorator
def something():
    f = Foobar()
    return f.get_something()

if __name__ == '__main__':
    print(something())

我正在尝试使用Foobar 到FakeFoobar 或猴子补丁Foobar 的get_something 方法对FakeFoobar 的get_something 方法进行一对一替换。

当我运行上面的代码时，我得到以下信息：

>>> something()
'apple'
>>>

我想找到一些方法来扩充Foobar 的get_something 方法，以便我们得到以下输出：

>>> something()
'orange'
>>>

unittests 库中有一个 mock 模块，但是，尚不清楚如何将其用于我的用例。正如mock 库中的许多示例所示，我非常赞同不将参数传递给装饰器或将额外参数传递给something 函数的想法。

我还注意到moto library 正在完成类似于我正在尝试做的事情。我尝试深入研究源代码，但对于我正在尝试做的事情来说似乎相当复杂。

Answer 1

更新函数的全局变量dict怎么样？

from functools import update_wrapper

class Foobar:
    def get_something(self):
        return "apple"

class FakeFoobar:
    def get_something(self):
        return "orange"

class my_decorator:
    def __init__(self, original_function):
        self._original_function = original_function
        update_wrapper(self, original_function)

    def __call__(self, *args, **kwargs):
        f = self._original_function
        restore_val = f.func_globals['Foobar']
        f.func_globals['Foobar'] = f.func_globals['FakeFoobar']
        # ^^^^^ This is your magic-line.
        try:
            return f(*args, **kwargs)
        except:
            raise
        finally:
            f.func_globals['Foobar'] = restore_val

@my_decorator
def something():
    f = Foobar()
    return f.get_something()

if __name__ == '__main__':
    print(something())   #Prints orange
    print(Foobar().get_something()) #Prints apple