デコレータで名前付き引数を使用するにはどうすればよいですか?
-
06-07-2019 - |
質問
次の機能がある場合:
def intercept(func):
# do something here
@intercept(arg1=20)
def whatever(arg1,arg2):
# do something here
arg1 が20の場合にのみインターセプトが起動するようにします。名前付きパラメーターを関数に渡すことができるようにしたいと思います。どうすればこれを達成できますか?
ここに小さなコードのサンプルがあります:
def intercept(func):
def intercepting_func(*args,**kargs):
print "whatever"
return func(*args,**kargs)
return intercepting_func
@intercept(a="g")
def test(a,b):
print "test with %s %s" %(a,b)
test("g","d")
これにより、次の例外TypeErrorがスローされます。intercept()が予期しないキーワード引数 'a'を取得しました
解決
from functools import wraps
def intercept(target,**trigger):
def decorator(func):
names = getattr(func,'_names',None)
if names is None:
code = func.func_code
names = code.co_varnames[:code.co_argcount]
@wraps(func)
def decorated(*args,**kwargs):
all_args = kwargs.copy()
for n,v in zip(names,args):
all_args[n] = v
for k,v in trigger.iteritems():
if k in all_args and all_args[k] != v:
break
else:
return target(all_args)
return func(*args,**kwargs)
decorated._names = names
return decorated
return decorator
例:
def interceptor1(kwargs):
print 'Intercepted by #1!'
def interceptor2(kwargs):
print 'Intercepted by #2!'
def interceptor3(kwargs):
print 'Intercepted by #3!'
@intercept(interceptor1,arg1=20,arg2=5) # if arg1 == 20 and arg2 == 5
@intercept(interceptor2,arg1=20) # elif arg1 == 20
@intercept(interceptor3,arg2=5) # elif arg2 == 5
def foo(arg1,arg2):
return arg1+arg2
>>> foo(3,4)
7
>>> foo(20,4)
Intercepted by #2!
>>> foo(3,5)
Intercepted by #3!
>>> foo(20,5)
Intercepted by #1!
>>>
functools.wraps
は、「シンプルデコレータ」の機能を実行します。ウィキ上では; __ doc __
、 __ name __
およびその他のデコレータの属性を更新します。
他のヒント
覚えておいてください
@foo
def bar():
pass
は次と同等です:
def bar():
pass
bar = foo(bar)
その場合:
@foo(x=3)
def bar():
pass
これは次と同等です:
def bar():
pass
bar = foo(x=3)(bar)
したがって、デコレータは次のようにする必要があります。
def foo(x=1):
def wrap(f):
def f_foo(*args, **kw):
# do something to f
return f(*args, **kw)
return f_foo
return wrap
つまり、 def wrap(f)
は実際にはデコレーターであり、 foo(x = 3)
はデコレーターを返す関数呼び出しです。
デコレータで* argsおよび** kwargsを使用してこれを行うことができます:
def intercept(func, *dargs, **dkwargs):
def intercepting_func(*args, **kwargs):
if (<some condition on dargs, dkwargs, args and kwargs>):
print 'I intercepted you.'
return func(*args, **kwargs)
return intercepting_func
デコレータの動作を制御する引数をどのように渡すかは、あなた次第です。
これをエンドユーザーに対してできるだけ透過的にするには、「シンプルデコレータ」を使用します。 Python wiki またはMichele Simionatoの&quot; decorator decorator&quot;
所属していません StackOverflow