Comment ajouter un décorateur à une méthode d'objet existante?

Question

Si j'utilise un module/classe sur lequel je n'ai aucun contrôle, comment pourrais-je décorer l'une des méthodes?

Je comprends que je peux: my_decorate_method(target_method)() mais je cherche à ce que cela se produise où target_method est appelé sans avoir à faire une recherche/remplacement.

Est-ce encore possible?

Answer 1

Oui c'est possible, mais il y a plusieurs problèmes. Tout d'abord, vous obtenez la méthode de la classe de façon évidente, vous obtenez un objet Warper mais pas la fonction elle-même.

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) 

Deuxièmement, je ne sais pas si les paramètres nouvelle méthode fonctionnera toujours:

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

Answer 2

Ne faites pas cela.

Utilisez l'héritage.

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. 

Maintenant, vous fixer le programme principal d'utiliser MyVersionOfAClass au lieu de some_module.AClass.