當使用具有可調用對象的裝飾器時返回用戶定義的函數名稱

Question

請考慮以下代碼片段。

def print_timing(func): 
    import time 
    def wrapper(*args, **kwargs): 
     t1 = time.time() 
     res = func(*args, **kwargs) 
     t2 = time.time() 
     print '%s took %0.3f s ~ %0.0f min and %0.1f sec' % (func.func_name, t2-t1, int(t2 - t1)/60, (t2-t1) % 60) 
     return res 
    return wrapper 

@print_timing                  | 
def foo():                   | 
    return 'foo' 

class name(object): 
     def __init__(self, name): 
       self.name = name 
     @print_timing 
     def __call__(self): 
       return self.name 

bar = name("bar") 
print bar() 

這將返回：

__call__ took 0.000 s ~ 0 min and 0.0 sec 
bar 

目的bar表現得像稱爲bar函數，但與裝飾print_timing使用時暴露的__call__內部實現細節。有沒有一種方法（也許傳遞合適的參數傳遞給__init__功能）來改變name對象，因此它返回，而不是

bar took 0.000 s ~ 0 min and 0.0 sec 

？我想要一個解決方案，讓print_timing修飾器繼續使用普通功能。運行 print foo()給

foo took 0.000 s ~ 0 min and 0.0 sec 
foo 

Answer 1

只要您只使用方法上的裝飾，他們將通過self作爲第一個參數：

def print_timing(func): 
    import time 
    def wrapper(*args, **kwargs): 
     t1 = time.time() 
     res = func(*args, **kwargs) 
     t2 = time.time() 
     funcname = func.__name__ 
     # Special case; a "name" instance has a "name" attribute we want to use instead. 
     if len(args) >= 1 and isinstance(args[0], name): 
      funcname = args[0].name 
     print '%s took %0.3f s ~ %0.0f min and %0.1f sec' % (funcname, t2-t1, int(t2 - t1)/60, (t2-t1) % 60) 
     return res 
    return wrapper 

更新：包裝程序現在使用func.__name__默認，但如果你使用這個在name類（如你原來的問題），它將使用實例的name屬性。

我已使用isinstance測試來確定name屬性將存在，​​但您可以使用鴨式輸入（if hasattr(args[0], 'name')）; name變量非常通用，但是在任意類方法上使用時，最有可能會得到意想不到的結果。

Answer 2

號類建立時發生的裝飾和實例是建立在__init__()調用發生。您需要讓裝飾器將該函數轉換爲描述符並讓該描述符從實例中獲取名稱。

Answer 3

使用@print_timing爲一類裝飾：

@print_timing 
class name(object): 
    ... 

沒有變化是必要的;你的包裝對象現在是一個函數，當它被認爲是一個類時，但是我從你的問題（和它是一個可調用的類）推斷出它並不重要（如果是這樣，你可以修改裝飾器到使返回的包裝對象「漂亮」）。