检查函数是否被装饰

Question

我正在尝试在 class 方法中构建一个控制结构，该方法将一个函数作为输入，并在函数是否被修饰时具有不同的行为。关于如何构建一个行为如下的函数 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

一般来说，你不能。留下痕迹的装饰器没有什么特别的。每优秀 Primer on Python Decorators tutorial, "Decorators provide a simple syntax for calling higher-order functions."

教程中的一个快速说明性示例：

def do_twice(func):
    def wrapper_do_twice():
        func()
        func()
    return wrapper_do_twice

from decorators import do_twice

@do_twice
def say_whee():
    print("Whee!")

你看到 wrapper_do_twice 在 do_twice 中是如何真正地调用传递的函数两次了吗？它绝不会修改函数，所以不可能告诉（没有 inspection，不要那样做）函数被修饰了。

但是，如果您正在编写自己的装饰器，或者知道您可以利用您正在使用的装饰器正在修改装饰函数（与示例相反上面，不修改函数），那么你可以直接检查是否有任何标记被留下。有关示例，请参阅问题。

Answer 2

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

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

检查函数是否被装饰

Checking whether a function is decorated

python

python-3.x

closures

decorator

python-decorators