詳解Python實現同時支持帶調用和不調用帶裝飾器

更新時間：2023年06月13日 11:28:13 作者：臨淵

一般來說，不帶參數裝飾器，再使用時不加括號，帶參數的裝飾器使用時必須加括號，這篇文章主要介紹了Python實現同時支持帶調用和不調用帶裝飾器的相關知識,需要的朋友可以參考下

一般來說，不帶參數裝飾器，再使用時不加括號，帶參數的裝飾器使用時必須加括號。例如

不帶參裝飾器

def info(func):  # 裝飾器，參數接收一個函數，返回一個函數
    @functools.wraps(func)
    def wrapper(*args, **kwargs):   # 與原函數功能一致，支持任意參數
        print(f'調用函數: {func.__name__} 參數：{args} {kwargs}')
        return func(*args, **kwargs)   # 內部包裹調用原函數
    return wrapper  # 返回和原函數功能一致的函數
@info   # 使用時不加括號
def add(a, b):
    return a + b

帶參數的裝飾器（返回裝飾器的函數）

def info(show_result=True):
    def _info(func):  # 裝飾器，參數接收一個函數，返回一個函數
        @functools.wraps(func)
        def wrapper(*args, **kwargs):  # 與原函數功能一致，支持任意參數
            print(f'調用函數: {func.__name__} 參數：{args} {kwargs}')
            start_time = time.time()
            result = func(*args, **kwargs)  # 內部包裹調用原函數
            if show_result is True:
                print(f'調用結果: {result} 耗時: {time.time()-start_time}')
            return result
        return wrapper  # 返回替換后的新函數
    return _info
@info()  # 即使使用默認參數也必須加括號調用（因為調用后才能得到裝飾器）
def add(a, b):
    return a + b

注意: 帶參數裝飾器雖然通過參數支持定制，但是使用時必須加括號，用戶很容易遺漏或分不清是否需要加括號

能不能像@pytest.fixture那樣，既可以不帶參數，也可以帶參數使用呢，例如：

import pytest
@pytest.fixture
def a(): ...
@pytest.fixture()
def b(): ...
@pytest.fixture(scope='module')
def c(): ...

這就要求，我們裝飾器外部函數，即可以本身作為一個裝飾器（接收一個原函數，返回一個同功能函數），也可以根據參數調用后返回一個裝飾器，實現如下：

def info(func=None, show_result=False):  # 第一個參數為函數
    def _info(func):  # 裝飾器，參數接收一個函數，返回一個函數
        @functools.wraps(func)
        def wrapper(*args, **kwargs):  # 與原函數功能一致，支持任意參數
            print(f'調用函數: {func.__name__} 參數：{args} {kwargs}')
            start_time = time.time()
            result = func(*args, **kwargs)  # 內部包裹調用原函數
            if show_result is True:
                print(f'調用結果: {result} 耗時: {time.time()-start_time}')
            return result
        return wrapper  # 返回替換后的新函數
    if func is not None:  # 不帶括號（作為裝飾器）使用時
        return _info(func)  # 返回 同功能函數wrapper（即_info的調用結果）
    return _info  # 否則，帶括號調用（作為返回裝飾器的函數）使用時，返回裝飾器 _info
@info  # 即使使用默認參數也必須加括號調用（因為調用后才能得到裝飾器）
def add(a, b):
    return a + b
@info()
def sub(a, b):
    return a - b
@info(show_result=True)
def mul(a, b):
    return a * b