Python裝飾器如何實現(xiàn)功能擴展

在Python中，裝飾器是一種特殊的函數(shù)，它可以用來修改其他函數(shù)的行為。裝飾器通過接收一個函數(shù)作為參數(shù)，然后返回一個新的函數(shù)，這個新函數(shù)通常會包含原始函數(shù)的功能，并在此基礎(chǔ)上添加或修改一些行為。

要實現(xiàn)功能擴展，你可以使用裝飾器來包裝原始函數(shù)，然后在新的函數(shù)中調(diào)用原始函數(shù)，并在適當?shù)臅r機添加額外的功能。下面是一個簡單的示例：

def my_decorator(func):
    def wrapper():
        print("Something is happening before the function is called.")
        func()
        print("Something is happening after the function is called.")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

say_hello()

在這個例子中，我們定義了一個名為my_decorator的裝飾器。這個裝飾器接收一個函數(shù)func作為參數(shù)，然后定義了一個名為wrapper的新函數(shù)。在wrapper函數(shù)中，我們首先打印一條消息，然后調(diào)用原始函數(shù)func，最后再打印一條消息。最后，裝飾器返回wrapper函數(shù)。

我們使用@my_decorator語法將裝飾器應(yīng)用于名為say_hello的函數(shù)。這實際上是將say_hello函數(shù)作為參數(shù)傳遞給my_decorator，并將返回的wrapper函數(shù)賦值給say_hello。因此，當我們調(diào)用say_hello時，實際上是在調(diào)用wrapper函數(shù)，從而實現(xiàn)了功能擴展。

輸出結(jié)果如下：

Something is happening before the function is called.
Hello!
Something is happening after the function is called.

這個簡單的示例展示了如何使用裝飾器來實現(xiàn)功能擴展。你可以根據(jù)需要修改wrapper函數(shù)，以添加更多的功能或修改原始函數(shù)的行為。