首頁  >  問答  >  主體

python - 用__call__ 實作裝飾器功能

用一個類別的__call__ 來實作一個裝飾器,如下程式碼



class Check(object):
    def __init__(self, name):
        self.name = name

    def __call__(self, func):

        print "1111111111"

        def _decorator(*args, **kwargs):
            print "2222222222"
            return func(*args, **kwargs)
          

如果把這個類別當作裝飾器修飾一個函數。

@Check("param")
def param_check(request):
    "python code....."
    return Response("ok")

問題來了。啟動一個django或其他任意一個python進程。執行的時候這個param_check路由的時候,print "1111111111" 只是第一次列印,後面再呼叫的時候就不列印1111111111111。
而 print "2222222222" 每次列印的。我的理解是第一次 是__call__ 在定義這個裝飾器,所以只執行了一次,後面再次請求時已經定義好了,只執行函數體的部分,也就是 print 222222222的部分。誰知道這個python裝飾器的特性

某草草某草草2641 天前796

全部回覆(1)我來回復

  • 迷茫

    迷茫2017-06-28 09:26:02

    其實類式裝飾器時,有個需要注意的地方,有參數和無參數的類式裝飾器其實是不一樣的

    不含參數的類式裝飾器

    如果創建了一個不帶參數的裝飾器,被裝飾的方法會傳遞給裝飾器的構造器(__init__),然後在被裝飾的函數被調用的時候,裝飾器的__call__()方法就會執行。

    class Check(object):
    
        def __init__(self, func):
            self.func = func
    
        def __call__(self, *args):
    
            print("111111")
            self.func(*args)
            print("222222")
    
    @Check
    def param_check(request):
        print(request)
    
    
    
    param_check('hello')
    param_check('world')
    
    111111
    hello
    222222
    111111
    world
    222222

    需要注意,在裝飾階段,__init__ 函數執行,在被裝飾的方法被呼叫的時候,__call__ 執行。

    帶參數的類別裝飾器

    class Check(object):
        def __init__(self, name):
            self.name = name
    
        def __call__(self, func):
    
            print ("1111111111")
    
            def decorator(*args, **kwargs):
                print ("2222222222")
                return func(*args, **kwargs)
            return decorator
    
    @Check('parm')
    def param_check():
        print('Hello')
    
    param_check()
    param_check()

    輸出:

    1111111111
    2222222222
    Hello
    2222222222
    Hello

    可以很明顯的看出來, __call__只會在裝飾階段被調用一次

    想了解的更詳細看這裡: http://www.artima.com/weblogs...

    回覆
    0
  • 取消回覆