Przekazywanie parametru do dekoratora w pythonie

Question

Dlaczego ten dekorator nie działa z parametrem?

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

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

print(test2()) 

zawsze daje mi błąd mówiąc „str nie jest wymagalne” stara się wykonać ciąg zwrotny wewnątrz owijki() zamiast przetwarzać go i zwróci ciąg wynik

Answer 1

dekoratorów funkcje te funkcje powrotu. Kiedy "przekazanie parametru do dekoratora" faktycznie wykonywane jest wywołanie funkcji, która zwraca dekorator. Tak więc decAny() powinna być funkcją, która zwraca funkcję, która zwraca funkcję.

To mogłoby wyglądać tak:

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' 

Przykład:

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

Należy pamiętać, że oprócz ustalenia konkretnego problemu ty uderzanie ja również poprawić swój kod nieco dodając *args i **kwargs jako argumenty funkcji opakowanej i przekazywanie ich do wywołania f0 wewnątrz dekoratora. Dzięki temu można udekorować funkcję, która akceptuje dowolną liczbę argumentów pozycyjnych lub nazwanych i nadal będzie działała poprawnie.

Można przeczytać o functools.wraps() tutaj:
http://docs.python.org/2/library/functools.html#functools.wraps

Answer 2

Jest to dobry przykład z "Mark Lutz - Learning Python" book:

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