功能：

為已存在的函數(shù)或?qū)ο筇砑宇~外的功能

?

原則：

• 不改變源代碼為其添加功能
• 不改變函數(shù)的調(diào)用方式

?

方法：

裝飾器 = 高階函數(shù) + 嵌套函數(shù)

（高階函數(shù)：一個(gè)函數(shù)可以作為參數(shù)傳遞給另外一個(gè)函數(shù)，或者，一個(gè)函數(shù)的返回值是一個(gè)函數(shù)，即函數(shù)的入口地址）

1. 函數(shù)名作為參數(shù)傳遞給裝飾器（@decorator_name）
2. 裝飾函數(shù)返回函數(shù)名（函數(shù)地址）

注意：

1. 默認(rèn)情況下，裝飾器會修改名字和文檔說明，但是可以使用 functools 中 的 @wraps() 解決。@wraps接受一個(gè)函數(shù)來進(jìn)行裝飾，并加入了復(fù)制函數(shù)名稱、注釋文檔、參數(shù)列表等等的功能。這可以讓我們在裝飾器里面訪問在裝飾之前的函數(shù)的屬性。

            
              def decorator(func):
    """
    decorator __doc__
    """
    # @wraps(func)
    def wrapper(*args, **kwargs):
        """wrapper __doc__"""
        func()
    return wrapper


@decorator
def test():
    """test __doc__"""
    time.sleep(0.5)


test(1, 2)
print("function name:", test.__name__)
print("function doc :", test.__doc__)

# output：
# function name: wrapper
# function doc : wrapper __doc__

# 加了@wraps(func)后的output：
# function name: test
# function doc : test __doc__
            
          

?

?

例子：

1. 函數(shù)作為裝飾器

此例子實(shí)現(xiàn)了一個(gè)計(jì)算調(diào)用函數(shù)所占用的時(shí)間

            
              import time
from functools import wraps


def decorator(func):
    """
    function：decorator
    """
    @wraps(func)
    def wrapper(*args, **kwargs):
        """function：wrapper"""
        start = time.time()
        ret = func(*args, **kwargs)
        end = time.time()
        print("function {0} run time: {1}".format(func.__name__, end - start))
        # print("function {fun} run time: {time}".format(fun=func.__name__, time=end - start))
        return ret
    return wrapper


@decorator
def test(a, b, name="Lizo"):
    """function：test"""
    time.sleep(0.5)
    print(a, b, name)

            
          

2.類作為裝飾器

為什么可以使用類作為裝飾器？因?yàn)樵赑ython中，一切皆對象，其實(shí)函數(shù)也是一個(gè)對象，如果一個(gè)類實(shí)現(xiàn)了? __call__(self) 方法后，就可以像調(diào)用函數(shù)一樣，直接加一個(gè)括號就可以調(diào)用。

            
              class Runtime:
    def __init__(self):
        pass

    def __call__(self, func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            start = time.time()
            ret = func(*args, **kwargs)
            end = time.time()
            print("function: {func} run time: {time}".format(func=func.__name__, time=end - start))
            return ret
        return wrapper


#使用裝飾器方法1
runtime = Runtime()
@runtime
def test_class_decorator1():
    print("in the test_class_decorator")
    time.sleep(0.2)


#使用裝飾器方法2
@Runtime()
def test_class_decorator2():
    print("in the test_class_decorator")
    time.sleep(0.2)
            
          

?