具有自变量的类方法修饰器？

小编典典

具有自变量的类方法修饰器？

python

如何将类字段作为参数传递给类方法的装饰器？我想做的是这样的：

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

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

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

阅读 140

2020-12-20

共1个答案

小编典典

是。与其在类定义时传递实例属性，不如在运行时检查它：

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

2020-12-20