如何将类字段作为参数传递给类方法上的装饰器?我想要做的是这样的:具有自我参数的Python类方法装饰器?
class Client(object):
def __init__(self, url):
self.url = url
@check_authorization("some_attr", self.url)
def get(self):
do_work()
它抱怨自我不存在self.url传递给装饰。有没有解决的办法?
如何将类字段作为参数传递给类方法上的装饰器?我想要做的是这样的:具有自我参数的Python类方法装饰器?
class Client(object):
def __init__(self, url):
self.url = url
@check_authorization("some_attr", self.url)
def get(self):
do_work()
它抱怨自我不存在self.url传递给装饰。有没有解决的办法?
是的。而不是在类定义时间传入实例属性,请在运行时检查它:
def check_authorization(f):
def wrapper(*args):
print args[0].url
return f(*args)
return wrapper
class Client(object):
def __init__(self, url):
self.url = url
@check_authorization
def get(self):
print 'get'
>>> Client('http://www.google.com').get()
http://www.google.com
get
装饰器拦截方法参数;第一个参数是实例,所以它读取该属性。您可以在属性名作为一个字符串传递给装饰和使用getattr
如果你不想硬编码的属性名称:
def check_authorization(attribute):
def _check_authorization(f):
def wrapper(self, *args):
print getattr(self, attribute)
return f(self, *args)
return wrapper
return _check_authorization
你不行。在课堂上没有self
,因为没有实例存在。您需要传递它,例如str
,其中包含要在实例上查找的属性名称,然后返回的函数可以执行此操作,或者完全使用其他方法。
from re import search
from functools import wraps
def is_match(_lambda, pattern):
def wrapper(f):
@wraps(f)
def wrapped(self, *f_args, **f_kwargs):
if callable(_lambda) and search(pattern, (_lambda(self) or '')):
f(self, *f_args, **f_kwargs)
return wrapped
return wrapper
class MyTest(object):
def __init__(self):
self.name = 'foo'
self.surname = 'bar'
@is_match(lambda x: x.name, 'foo')
@is_match(lambda x: x.surname, 'foo')
def my_rule(self):
print 'my_rule : ok'
@is_match(lambda x: x.name, 'foo')
@is_match(lambda x: x.surname, 'bar')
def my_rule2(self):
print 'my_rule2 : ok'
test = MyTest()
test.my_rule()
test.my_rule2()
输出继电器: my_rule2:OK
非常感谢你!这应该是被接受的答案。我正在试图弄清装饰者的类别方法。 – benshepherd 2017-03-09 12:04:42
@raphael在这个设置中,我似乎无法访问_lambda或模式。我该如何补救。 – Jonathan 2018-01-18 17:03:32
更简洁的例子可能如下:
#/usr/bin/env python3
from functools import wraps
def wrapper(method):
@wraps(method)
def _impl(self, *method_args, **method_kwargs):
return method(self, *method_args, **method_kwargs) + "!"
return _impl
class Foo:
@wrapper
def bar(self, word):
return word
f = Foo()
result = f.bar("kitty")
print(result)
,它将打印:
kitty!
这是一个你有控制权的自定义装饰器,还是一个你不能改变的装饰器? – 2012-07-30 23:35:30
这是我的装饰器,所以我完全控制它 – Mark 2012-07-30 23:36:19
它在init之前被调用我认为是问题... – 2012-07-30 23:37:57