Come registrare all'interno di un decoratore tutte le funzioni che decora?

Question

Supponiamo che tu abbia il seguente decoratore. Come può essere modificato per aggiungere un'appendice a qualche elenco di riferimenti a tutte le funzioni che decora?

def memoize(obj): 
    cache = obj.cache = {} 

    @functools.wraps(obj) 
    def memoizer(*args, **kwargs): 
     if args not in cache: 
      cache[args] = obj(*args, **kwargs) 
     return cache[args] 
    return memoizer 

@memoize 
def foo(bar): 
    return bar ** 3 

Answer 1

Si può facilmente memorizzare un elenco sull'oggetto decoratore-funzione (memoize.decorated):

_decorated = [] 

def memoize(obj): 
    cache = obj.cache = {} 

    # add to the decorated list 
    _decorated.append(obj) 

    @functools.wraps(obj) 
    def memoizer(*args, **kwargs): 
     if args not in cache: 
      cache[args] = obj(*args, **kwargs) 
     return cache[args] 
    return memoizer 

# make the list accessible from the decorator: 
memoize.decorated = _decorated 

Esso può essere utilizzato come:

@memoize 
def foo(bar): 
    return bar ** 3 

print memoize.decorated 

Come nota a margine, si dovrebbe considerare l'archiviazione di WeakRefs in the list, per evitare perdite di memoria o oggetti che non vengono liberati quando non vengono lasciati altri riferimenti "reali".