Jak zarejestrować w dekoratorze wszystkie funkcje, które zdobi?

Question

Załóżmy, że masz następujący dekorator. W jaki sposób można go zmodyfikować, dodając do niektórych odnośników do wszystkich funkcji, które zdobi?

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

można łatwo przechowywać listę na obiekcie dekorator-function (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 

To może być używany jak:

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

print memoize.decorated 

Na marginesie, należy rozważ przechowywanie WeakRefs in the list, aby uniknąć wycieku pamięci lub uwolnienia się obiektów, gdy nie pozostawia się żadnych "prawdziwych" odniesień do nich.