如何將裝飾器添加到現有的對象方法？

Question

如果我正在使用模塊/類，我無法控制，我將如何修飾其中一種方法？

我知道我可以：my_decorate_method(target_method)()但我希望有這種情況發生，無需進行搜索/替換即可調用target_method。

這有可能嗎？

Answer 1

是的，這是可能的，但有幾個問題。 首先，你明白地從類中得到一個方法，你得到一個修飾器對象，但不是函數本身。

class X(object): 
    def m(self,x): 
     print x 

print X.m   #>>> <unbound method X.m> 
print vars(X)['m'] #>>> <function m at 0x9e17e64> 

def increase_decorator(function): 
    return lambda self,x: function(self,x+1) 

其次，我不知道，如果設置新的方法將總是工作：

x = X() 
x.m(1)   #>>> 1 
X.m = increase_decorator(vars(X)['m']) 
x.m(1)   #>>> 2 

Answer 2

不要這樣做。

使用繼承。

import some_module 

class MyVersionOfAClass(some_module.AClass): 
    def someMethod(self, *args, **kwargs): 
     # do your "decoration" here. 
     super(MyVersionOfAClass, self). someMethod(*args, **kwargs) 
     # you can also do "decoration" here. 

現在，收拾你的主程序使用MyVersionOfAClass，而不是some_module.AClass。