Jak mogę uruchomić dekorator Pythona po zakończeniu dekorowanej funkcji?

Chcę użyć dekoratora do kontrolowania różnych funkcji (głównie funkcji widoku Django, ale nie wyłącznie). W tym celu chciałbym móc przeprowadzić audyt funkcji Po wykonaniu - tzn. funkcja działa normalnie, a jeśli powróci bez wyjątku, to dekorator rejestruje fakt.

Coś w stylu: 

@audit_action(action='did something')
def do_something(*args, **kwargs):
    if args[0] == 'foo':
        return 'bar'
    else:
        return 'baz'

Gdzie audit_action będzie działać dopiero po zakończeniu funkcji.

Author: Martijn Pieters, 2013-02-05
Source

2 answers

Dekoratory zwykle zwracają funkcję wrappera; po wywołaniu funkcji wrappera wystarczy umieścić swoją logikę w funkcji wrapper.

def audit_action(action):
    def decorator_func(func):
        def wrapper_func(*args, **kwargs):
            # Invoke the wrapped function first
            retval = func(*args, **kwargs)
            # Now do something here with retval and/or action
            print 'In wrapper_func, handling action {!r} after wrapped function returned {!r}'.format(action, retval)
            return retval
        return wrapper_func
    return decorator_func

Więc audit_action(action='did something') jest fabryką dekoratorów, która zwraca lunetę decorator_func, która służy do dekoracji Twojego do_something (do_something = decorator_func(do_something)).

Po zdobieniu, Twoje do_something odniesienie zostało zastąpione przez wrapper_func. Wywołanie wrapper_func() powoduje wywołanie oryginalnego do_something(), a następnie Twój kod w func może robić różne rzeczy.

Powyższy kod, w połączeniu z funkcją przykład, daje następujące wyjście: 

>>> do_something('foo')
In wrapper_func, handling action 'did something' after wrapped function returned 'bar'
'bar'
 28
Author: Martijn Pieters,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2013-02-05 08:55:35

Twój dekorator sam sobie z tym poradzi, jak 

def audit_action(function_to_decorate):
    def wrapper(*args, **kw):
        # Calling your function
        output = function_to_decorate(*args, **kw)
        # Below this line you can do post processing
        print "In Post Processing...."
    return wrapper
 3
Author: avasal,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2013-02-05 08:50:33