Pythonは関数の引数を保持するデコレータを作成します。

Question

私が装飾する関数の引数を保持するデコレータを作成しようとしています。これを行う動機はpytest.fixturesとうまくやりとりするデコレータを書くことです。

fooがあるとします。単一の引数aが必要です。

def foo(a): 
    pass 

我々がfoo

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

の引数の仕様を取得した場合私たちは頻繁にwrapper機能がwrapped関数にそのままそのすべての引数を渡すデコレータを作成したいです。これを行う最も明白な方法は*argsと**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 

これは、当然のことながら、*argsと**kwargs反射引数仕様と機能を生成します。

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

包まれた機能と一致するか、最初に右引数の仕様と機能を作成するには、引数の仕様を変更するためのいずれかの方法がありますか？

Answer 1

コメントで示唆したように、あなたがdecoratorモジュールを使用することができますかあなたは正しい署名でラムダ関数を作成するためにeval悪の力を使用することができます。

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 

これはちょっとハックですが、それは、関数の引数を保存し：

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

Answer 2

私の知る限り、それはSignatureオブジェクトのみのPython 3.3ので可能である。

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

次に、あなたが行うことができます：

@identity_decorator 
def foo(a): 
    pass 

し、最終的に：

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