Passare un parametro al decoratore in python

Question

Perché questo decoratore con un parametro non funziona?

def decAny(f0): 
    def wrapper(s0): 
     return "<%s> %s </%s>" % (any, f0(), any) 
    return wrapper 

@decAny('xxx') 
def test2(): 
    return 'test1XML' 

print(test2()) 

mi dà sempre un errore che dice "str non è callable" si sta cercando di eseguire la stringa di ritorno all'interno del involucro() invece di elaborarlo e restituire la stringa risultato

Answer 1

decoratori sono funzioni che funzioni di ritorno. Quando "passando un parametro al decoratore", ciò che stai facendo in realtà è chiamare una funzione che restituisce un decoratore. Quindi decAny() dovrebbe essere una funzione che restituisce una funzione che restituisce una funzione.

Sarebbe simile a questa:

import functools 

def decAny(tag): 
    def dec(f0): 
     @functools.wraps(f0) 
     def wrapper(*args, **kwargs): 
      return "<%s> %s </%s>" % (tag, f0(*args, **kwargs), tag) 
     return wrapper 
    return dec 

@decAny('xxx') 
def test2(): 
    return 'test1XML' 

Esempio:

>>> print(test2()) 
<xxx> test1XML </xxx> 

Nota che, oltre a risolvere il problema specifico che si stavano colpendo Ho anche migliorato il codice un po 'con l'aggiunta di *args e **kwargs come argomenti per la funzione avvolta e passandoli sulla chiamata f0 all'interno del decoratore. In questo modo è possibile decorare una funzione che accetta un numero qualsiasi di argomenti posizionali o denominati e funzionerà comunque correttamente.

È possibile leggere su functools.wraps() qui:
http://docs.python.org/2/library/functools.html#functools.wraps

Answer 2

V'è un buon esempio da "Mark Lutz - Learning Python" libro:

def timer(label=''): 
    def decorator(func): 
     def onCall(*args): # Multilevel state retention: 
      ...    # args passed to function 
      func(*args)  # func retained in enclosing scope 
      print(label, ... # label retained in enclosing scope 
     return onCall 
    return decorator   # Returns the actual decorator 

@timer('==>')    # Like listcomp = timer('==>')(listcomp) 
def listcomp(N): ...   # listcomp is rebound to new onCall 

listcomp(...)    # Really calls onCall