python中的__call__

丁壯發表於2018-09-15

    如果python中的一個類定義了 __call__ 方法,那麼這個類它的例項就可以作為函式呼叫,也就是實現了 () 運算子,即可呼叫物件協議

    下面是一個簡單的例子:

class TmpTest:
def __init__(self, x, y):
self.x = x
self.y = y

def __call__(self, x, y):
self.x, self.y = x, y


a = TmpTest(1, 2)
a(4, 5)
print(a.x, a.y)
4 5

 

    在本文中不討論裝飾部分的內容,借用裝飾器來講解一個__call__方法的使用,如果需要將一個類作為裝飾器,那需要為這個類實現__call__方法,一個使用__call__來實現類裝飾器的例子:

class TmpTest:
def __init__(self, func):
self.func=func

def __call__(self, *args,**kwargs):
result=self.func(*args,**kwargs)
return result


@TmpTest
def add_num(x,y):
return x+y

print(add_num(1,0))
1

 

相關文章