当前位置 : 主页 > 编程语言 > python >

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

来源:互联网 收集:自由互联 发布时间:2021-06-25
假设我有以下内容: def with_connection(f): def decorated(*args, **kwargs): f(get_connection(...), *args, **kwargs) return decorated@with_connectiondef spam(connection): # Do something 我想测试垃圾邮件功能,而不必经历设
假设我有以下内容:

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

@with_connection
def spam(connection):
    # Do something

我想测试垃圾邮件功能,而不必经历设置连接的麻烦(或者装饰者正在做的任何事情).

鉴于垃圾邮件,如何从中删除装饰器并获得底层的“未修饰”功能?

在一般情况下,你不能,因为

@with_connection
def spam(connection):
    # Do something

相当于

def spam(connection):
    # Do something

spam = with_connection(spam)

这意味着“原始”垃圾邮件可能甚至不再存在.一个(不太漂亮)黑客将是这样的:

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
网友评论