我應該如何在不更改函數名稱的情況下使用python裝飾器？

Question

def decorate(func): 
    def wrapper(*args, **kwargs): 
     return func(*args, **kwargs) 
    return wrapper 

@decorate 
def test(a=1,b=2): 
    return a+b 

print test.__name__ 

結果是包裝。有沒有辦法讓結果「測試」？

Answer 1

使用functools.wraps：

from functools import wraps 

def decorate(func): 
    @wraps(func) 
    def wrapper(*args, **kwargs): 
     return func(*args, **kwargs) 
    return wrapper 

@decorate 
def test(a=1,b=2): 
    return a+b 

print test.__name__