我可以运行没有装饰器功能的装饰功能吗?

时间:2020-08-20 19:45:23

标签: python python-3.x decorator python-decorators

如果我曾经在函数中使用装饰器,那么如何在不嵌入装饰器功能的情况下单独运行该功能呢?

例如,我有一个函数printArg,它输出一个参数。为了提高可用性,我需要将其与datetime.now()“混合”。为此,我编写了一个装饰器timeCalled。现在,每次我打电话给printArg时,装饰者也叫。

是否有一种方法可以单独调用printArg,所以我不会重复自己并写另一个没有装饰器功能(printArg)的“ datetime.now()”吗?

from datetime import datetime

def timeCalled(func):
    def wrapper(*args, **kwargs):
        print(f'{datetime.now()}: Function called {func.__name__}')
        result = func(*args, **kwargs)
        return result
    return wrapper

@timeCalled
def printArg(arg):
    print(f'Your arg is {arg}')

printArg('Mama')

3 个答案:

答案 0 :(得分:2)

没有不依赖于实现细节的通用方法。在这种特殊情况下,由于func引用了wrapper中的原始函数并且在闭包中,因此可以使用:

printArg.__closure__[0].cell_contents

要检索它。

更一般而言,创建包装器时最好使用functools.wraps,这会使包装器函数看起来像原始函数。而且,它将原始函数添加到__wrapped__属性中,因此:

from datetime import datetime
from functools import wraps

def timeCalled(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print(f'{datetime.now()}: Function called {func.__name__}')
        result = func(*args, **kwargs)
        return result
    return wrapper

@timeCalled
def printArg(arg):
    print(f'Your arg is {name}')

在这种情况下,您可以使用:

printArg.__wrapped__

答案 1 :(得分:1)

装饰器只是装饰函数的包装函数,其语法为:

@timeCalled
def printArg(arg):
    pass

等效于:

def printArg(arg):
    pass
printArg = timeCalled(printArg)

因此要达到避免两次定义相同函数的目的,而只修饰一个函数,则可以定义一次函数,然后调用传递了该函数的修饰器,并将返回的函数对象分配给另一个名称代替:

def printArg(arg):
    print(f'Your arg is {name}')

timed_printArg = timeCalled(printArg)

这样您就可以调用timed_printArg('Mama')来定时调用printArg('Mama'),并直接调用printArg('Mama')来进行此操作而无需涉及装饰器。

答案 2 :(得分:1)

为修饰后的函数命名。为此,请不要使用@表示法,而是:

decorated_printArg = timeCalled(printArg)