Python tworzenie argumentów funkcji zachowania dekoratora

Question

Próbuję napisać dekorator, który zachowuje argumenty funkcji, które dekoruje. Motywacją do tego jest napisanie dekoratora, który ładnie współdziała z pytest.fixtures.

Załóżmy, że mamy funkcję foo. Wystarczy jeden argument: a.

def foo(a): 
    pass 

Jeśli mamy spec argument foo

>>> inspect.getargspec(foo) 
ArgSpec(args=['a'], varargs=None, keywords=None, defaults=None) 

Często chcemy utworzyć dekorator gdzie funkcja wrapper przechodzi wszystkie jej argumenty verbatim do funkcji wrapped. Najbardziej oczywistym sposobem jest użycie *args i **kwargs.

def identity_decorator(wrapped): 
    def wrapper(*args, **kwargs): 
     return wrapped(*args, **kwargs) 
    return wrapper 

    def identity_decorator(wrapped): 
    def wrapper(*args, **kwargs): 
     return wrapped(*args, **kwargs) 
    return wrapper 

@identity_decorator 
def foo(a): 
    pass 

ten nie jest zaskoczeniem, wywołuje funkcję ze specyfikacją argumentu odzwierciedlającej *args i **kwargs.

>>> inspect.getargspec(foo) 
ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None) 

Czy istnieje sposób na zmianę argumentu spec, aby dopasować funkcję zawiniętą lub na początku utworzyć funkcję z właściwym argumentem spec?

Answer 1

Jak sugerowano w komentarzach, można użyć modułu decorator lub użyć eval złe moce, aby utworzyć funkcję lambda z prawidłowym podpisem:

import inspect 

def identity_decorator(wrapped): 
    argspec = inspect.getargspec(wrapped) 
    args = inspect.formatargspec(*argspec) 

    def wrapper(*args, **kwargs): 

     return wrapped(*args, **kwargs) 

    func = eval('lambda %s: wrapper%s' % (args.strip('()'), args), locals()) 

    return func 

@identity_decorator 
def foo(a): 
    pass 

To trochę hackish, ale zachowuje argumentów funkcji:

>>> inspect.getargspec(foo) 
ArgSpec(args=['a'], varargs=None, keywords=None, defaults=None) 

Answer 2

AFAIK, jest możliwe tylko od Pythona 3.3 z obiektem Signature:

def identity_decorator(wrapped): 
    def wrapper(*args, **kwargs): 
     return wrapped(*args, **kwargs) 
    wrapper.__signature__ = inspect.signature(wrapped) # the magic is here! 
    return wrapper 

Następnie można zrobić:

@identity_decorator 
def foo(a): 
    pass 

i wreszcie:

>>> inspect.getargspec(foo) 
ArgSpec(args=['a'], varargs=None, keywords=None, defaults=None)