帶有參數的Python裝飾器，多次運行函數？

Question

我想編寫一個Python裝飾器來裝飾unittest.TestCase的測試函數，以決定這個函數應該運行的目標主機。看到這個例子：

class MyTestCase(unittest.TestCase): 
    @target_host(["host1.com", "host2.com"]) 
    def test_my_command(self): 
     #do something here against the target host 

在裝飾功能，我希望能夠執行對所有主機這個測試中，我該怎麼做呢？ target_host的聲明應該返回一個新的函數，但可以返回測試運行器可以執行的多個函數嗎？

謝謝！

Answer 1

您只能返回一個對象（所以在技術上可以返回一組函數）。如果你想避免驚人的每個人，並且如果你想調用結果，你最好返回一個函數。但是這個函數很可能會在循環中調用其他幾個函數......你知道這會導致什麼嗎？

您需要一個裝飾器的工廠，它返回一個閉包，調用它們應用到的每個參數集的函數一次。在代碼（包括functools.wraps保持名稱和文檔字符串，可能是有用與否，我傾向於在默認情況下將其列入）：

def call_with_each(*arg_tuples): 
    def decorate(f): 
     @functools.wraps(f) 
     def decorator(): 
      for arg_tuple in arg_tuples: 
       f(*arg_tuple) 
     return decorator 
    return decorate 

# useage example: 
@call_with_each((3,), (2,)) # note that we pass several singleton tuples 
def f(x): 
    print x 
# calling f() prints "3\n2\n" 

支持關鍵字參數需要更多的代碼也許有些醜陋，但是是可能的。如果它始終是一個參數，則代碼可以簡化（def call_with_each(*args)，for arg in args: f(arg)等）。

Answer 2

如果您希望裝飾器接受參數，那麼使用基於類的裝飾器是一種方法。根據我的經驗，他們最終能夠更輕鬆地進行推理和維護。它們很簡單，__init__接受裝飾器的任何參數，__call__返回裝飾函數。其中一個寫裝飾器的問題是如果它們傳遞參數或者不參與，它們的行爲完全不同。這是很容易在基於類的裝飾，以解決這個問題，讓你的裝飾，以接受參數或不：

class target: 
    def __init__(self, targets): 
     """Arguments for decorator""" 
     self.targets = None 
     if not hasattr(targets, '__call__'): 
      # check we are actually passed arguments! 
      # if targets has __call__ attr, we were called w/o arguments 
      self.targets = targets 

    def __call__(self, f): 
     """Returns decorated function""" 
     if self.targets: 
      def newf(*args, **kwargs): 
       for target in self.targets: 
        f(target) 
      return newf 
     else: 
      return f 

現在，如果我們使用參數的裝飾，它會按預期工作，調用我們的函數3倍：

>>> @target([1,2,3]) 
..: def foo(x): print x 
... 

>>> foo() 
1 
2 
3 

但是，如果我們不帶參數調用，我們將返回，而不是原來的功能：

>>> @target  
def foo(x): print x 
..: 

>>> foo(3) 
<<< 3