Dekorator modyfikujący tablicę z innej funkcji i jak ją zwrócić

Question

Hej, właśnie poznaję koncepcję dekoratora i nie jestem pewien jak zwrócić w nim listę otrzymaną z innej funkcji (którą chcę udekorować), tzn. funkcja A zwraca listę i w dekoratorze robię wrappera i tam umieszczam funkcje B modyfikującą listę, a następnie zwracam wrappera, ale otrzymuję wartość None.

def my_walking_decorator(fn):
    def wrapper():
        walk = fn()
        for step in walk:
            step += ' cross'
    return wrapper


def walking():
    steps = []
    for i in range(1,11):
        tmp = 'step ' + str(i)
        steps.append(tmp)
    return steps


walking_decorated = my_walking_decorator(walking)
print(walking_decorated())

Gdzie popełniam błąd? IMHO powinno zwracać listę z dopisanym "cross", bo jak np wypisuję jakiś tekst i dekorator tylko wypisuje jakiś tekst dodatkowo, to działa.

Answer 1

1) wrapper() nic nie zwraca, dlatego jest None

2) step += ' cross' nie robi tego co oczekujesz. String w pythonie jest immutable, więc tak naprawdę tworzysz tylko nowego stringa, a ten w liście jest niezmieniony

Comments

Oo, dzięki, rzeczywiście, nie zauważyłem, czyli raczej powinienem użyć walk.append(tmp), a za tmp stworzyć to step + ' cross'

Dzięki za czujność. I pewnie jeszcze return wrappera samego.

---

walk[i] += " cross" powinno wystarczyć

Answer 2

AFAIK do tego co już napisał @Whistleroosh, masz nie do końca prawidłowy zapis dekoratora, sprawdź np. taki zapis

def my_walking_decorator(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        walk = fn(*args, **kwargs)
        for i in range(len(walk)):
            walk[i] += ' cross'
        return walk
    return wrapper
 
@my_walking_decorator 
def walking():
    ...

 

no i taki zapis spowoduje podwójne wywołanie dekoratora

[ on-line ]

walking_decorated = my_walking_decorator(walking)

sprawdź

[ on-line ]

walking_decorated = walking

 

Comments

Wziąłem to z kursu, wiem, że ten napis z małpą też istnieje, ale wtedy chyba nie można użyć funkcji bez dekoratora? Skąd to @wraps(fn) ? W sensie funkcja wrapper, w środku my_walking_decorator też jest jakby dekorowana?

---

Skąd to @wraps(fn) ?

sprawdź np. tutaj.