Python: programme pour obtenir le vrai nom de la fonction décorée

Question

Je veux écrire un décorateur pour montrer le vrai nom des fonctions décorées, mais quelque chose ne va pas quand j'écris de cette façon.

from functools import update_wrapper 

def decorator(d): 
    def _decorator(f): 
     update_wrapper(d(f), f) 
     return d(f) 
    return _decorator 

@decorator 
def add2(f): 
    def _add2(x,y): 
     return f(x,y)*2 
    return _add2 

@add2 
def add(x,y): return x+y 

Si je veux obtenir le nom d'ajouter (x, y), il montre comme ceci:

>>>print add.__name__ 
_add2 

Mais si je fais peu de changement au décorateur func comme ceci:

def decorator(d): 
    def _decorator(f): 
     return update_wrapper(d(f), f) 
    return _decorator 

Le résultat sera correct:

>>>print add.__name__ 
add 

Je suis vraiment confus à ce sujet. Dans mon esprit, cela ne devrait pas faire une telle différence parce qu'ils sont presque les mêmes.

Answer 1

Vous appelez d(f)deux fois, et chacun crée un nouvel objet fonction wrapper; ce sont des copies indépendantes et nouvelles. Vous n'appliquez que update_wrapper() au premier résultat et renvoyez le second.

Appelez-le une fois:

def decorator(d): 
    def _decorator(f): 
     wrapper = d(f) 
     update_wrapper(wrapper, f) 
     return wrapper 
    return _decorator