详解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

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