メソッドにデコレーターを付ける方法

Question

クラスのメソッドにデコレーターを付けたい時はどのようにすればようのでしょうか？

a.visible=Trueの時，a.add(2,3)で5が出力されるようにしたいと考えています
次のようにした時，エラーが出てしまいます．

def is_true(arg):
  def wrapper1(func):
    def wrapper2(*args,**kargs):
      if arg:
        func(*args,**kargs)
    return wrapper2
  return wrapper1

class A():
  def __init__(self):
    self.visible = True
    self.selected = True

  @is_true(self.visible)
  def add(a,b):
    print(a+b)

a = A()
a.add(2,3)

#>>TypeError: add() takes 2 positional arguments but 3 were given

Answer 1

実行してみたところ、 @is_true(self.visible)で以下のエラーが起こりました。

NameError: name 'self' is not defined

この時点ではselfは定義されていないからです。
なぜkt19906さんのコードでこのエラーが出なかったのかが不思議です。

以下のようなコードならこのエラーを回避できます。

def is_true(attr):
  def wrapper1(func):
    def wrapper2(self,*args,**kargs):
      if self.__getattribute__(attr):
        func(self,*args,**kargs)
    return wrapper2
  return wrapper1

class A():
  def __init__(self):
    self.visible = True
    self.selected = True
  @is_true('visible')
  def add(self,a,b):
    print(a+b)

実行結果

>>> a = A()
>>> a.add(2,3)
5

Comments

[kt19906]

if self.attrにすると，'A' object has no attribute 'attr'となってしまうので
self.__getattribute__(attr)とするんですね
特殊メソッドの知識が全然なかったので勉強になりました
ありがとうございます！