1. ホーム
  2. パイソン

[解決済み】クラスメソッドデコレーターでself引数?

2022-04-04 03:38:22

質問

クラスメソッド上のデコレーターにクラスフィールドを引数として渡すにはどうしたらよいですか? やりたいことは、以下のようなことです。

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

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

を渡すと、self が存在しないと文句を言われる。 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