2012-07-30 70 views
66

如何将类字段作为参数传递给类方法上的装饰器?我想要做的是这样的:具有自我参数的Python类方法装饰器?

class Client(object): 
    def __init__(self, url): 
     self.url = url 

    @check_authorization("some_attr", self.url) 
    def get(self): 
     do_work() 

它抱怨自我不存在self.url传递给装饰。有没有解决的办法?

+0

这是一个你有控制权的自定义装饰器,还是一个你不能改变的装饰器? – 2012-07-30 23:35:30

+0

这是我的装饰器,所以我完全控制它 – Mark 2012-07-30 23:36:19

+0

它在init之前被调用我认为是问题... – 2012-07-30 23:37:57

回答

98

是的。而不是在类定义时间传入实例属性,请在运行时检查它:

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 
2

你不行。在课堂上没有self,因为没有实例存在。您需要传递它,例如str,其中包含要在实例上查找的属性名称,然后返回的函数可以执行此操作,或者完全使用其他方法。

15
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

+1

非常感谢你!这应该是被接受的答案。我正在试图弄清装饰者的类别方法。 – benshepherd 2017-03-09 12:04:42

+0

@raphael在这个设置中,我似乎无法访问_lambda或模式。我该如何补救。 – Jonathan 2018-01-18 17:03:32

13

更简洁的例子可能如下:

#/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!