检查函数是否被修饰

Question

我正在尝试在类方法中构建一个控制结构，该方法将函数作为输入，并且如果函数被修饰或未修饰，则具有不同的行为。关于如何构建一个is_decorated行为如下的函数的任何想法：

def dec(fun):
    # do decoration


def func(data):
    # do stuff


@dec
def func2(data):
    # do other stuff


def is_decorated(func):
    # return True if func has decorator, otherwise False


is_decorated(func)  # False
is_decorated(func2) # True

Answer 1

是的，这相对容易，因为函数可以添加任意属性，因此装饰器函数可以在执行其操作时添加一个属性：

def dec(fun):
    def wrapped(*args, **kwargs):
        pass

    wrapped.i_am_wrapped = True
    return wrapped

def func(data):
    ... # do stuff

@dec
def func2(data):
    ... # do other stuff

def is_decorated(func):
    return getattr(func, 'i_am_wrapped', False)


print(is_decorated(func))   # -> False
print(is_decorated(func2))  # -> True