如何从Python函数中剥离装饰器

Question

假设我有以下几点：

def with_connection(f):
    def decorated(*args, **kwargs):
        f(get_connection(...), *args, **kwargs)
    return decorated

@with_connection
def spam(connection):
    # Do something

我想测试该spam功能，而无需经历建立连接的麻烦（或装饰者所做的任何事情）。

给定spam，我如何从中剥离装饰器并获得基础的“未装饰”功能？

Answer 1

在一般情况下，您不能这样做，因为

@with_connection
def spam(connection):
    # Do something

相当于

def spam(connection):
    # Do something

spam = with_connection(spam)

这意味着“原始”垃圾邮件甚至可能不再存在。一个（不太漂亮的）hack是这样的：

def with_connection(f):
    def decorated(*args, **kwargs):
        f(get_connection(...), *args, **kwargs)
    decorated._original = f
    return decorated

@with_connection
def spam(connection):
    # Do something

spam._original(testcon) # calls the undecorated function