Trouver toutes les fonctions décorées dans un module

Question

Est-il possible de savoir si une fonction est décorée au moment de l'exécution? Par exemple pourrais-je trouver toutes les fonctions dans un module qui sont décorées par "exemple"?

@example 
def test1(): 
    print "test1" 

Answer 1

Dans le cas général, il est impossible, car le décorateur example pourrait ne pas laisser aucune trace qui est détectable lors de l'exécution - par exemple, il pourrait être

def example(f): 
    return f 

Si vous ne contrôlez la source pour example, il est bien sûr facile de faire marquer ou enregistrer les fonctions qu'il décore; Si vous ne le faites pas, il peut être tout à fait impossible de faire ce que vous voulez.

Answer 2

Je ne pense pas qu'il y ait une manière générale depuis un décorateur est un simple appel de fonction. Votre code est identique à ceci:

def test1(): 
    print "test1" 

test1 = example(test1) 

Vous pourriez probablement détecter décorateurs spécifiques en démontant et l'analyse (en utilisant le module dis). Ou vous pouvez simplement analyser le fichier source, même si c'est un peu moche.

Pourquoi voulez-vous les détecter en premier lieu?

Answer 3

Si vous pouvez, monkey-patch le décorateur lors de l'exécution pour attraper les fonctions sont transmises à elle:

decfuncs = [] 

def decpatch(dec): 
    def catchdec(func, *args, **kwargs): 
    decfuncs.append(func) 
    return dec(func, *args, **kwargs) 
    return catchdec 

somemodule.somedecorator = decpatch(somemodule.somedecorator) 

Answer 4

Puisque vous avez indiqué que vous avez le contrôle sur le code d'emballage, est un exemple ici:

def example(f): 
    f.wrapped = True 
    return f 

@example 
def test1(): 
    print "test1" 

def test2(): 
    print "test2" 


print test1.wrapped 
print hasattr(test2, 'wrapped') 

Answer 5

Vous trouverez peut-être cette question récente fournit des informations détaillées sur les 3 façons de le faire:

Howto get all methods of a python class with given decorator