带有自参数的类方法装饰器？

小编典典

带有自参数的类方法装饰器？

all

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

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

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

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

阅读 66

2022-05-23

共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

2022-05-23