众所周知，以下两段代码几乎是等价的：
@dec
def foo():
pass foo = dec(foo)
############################################
foo = dec(foo)
一个常见的错误是认为 @ 只是隐藏了最左边的参数。
@dec(1, 2, 3)
def foo():
pass
###########################################
foo = dec(foo, 1, 2, 3)
如果以上是 @ 的工作方式，那么编写装饰器会容易得多。不幸的是，这不是做事的方式。
考虑一个装饰器Waitwhich haults
程序执行几秒钟。
如果你没有通过等待时间
那么默认值为 1 秒。
用例如下所示。
##################################################
@Wait
def print_something(something):
print(something)
##################################################
@Wait(3)
def print_something_else(something_else):
print(something_else)
##################################################
@Wait(delay=3)
def print_something_else(something_else):
print(something_else)
当Wait有参数时，比如@Wait(3)，那么调用Wait(3)
被执行前发生任何其他事情。
即下面两段代码是等价的
@Wait(3)
def print_something_else(something_else):
print(something_else)
###############################################
return_value = Wait(3)
@return_value
def print_something_else(something_else):
print(something_else)
这是个问题。
if `Wait` has no arguments:
`Wait` is the decorator.
else: # `Wait` receives arguments
`Wait` is not the decorator itself.
Instead, `Wait` ***returns*** the decorator
一种解决方案如下所示：
让我们从创建以下类开始，DelayedDecorator：
class DelayedDecorator:
def __init__(i, cls, *args, **kwargs):
print("Delayed Decorator __init__", cls, args, kwargs)
i._cls = cls
i._args = args
i._kwargs = kwargs
def __call__(i, func):
print("Delayed Decorator __call__", func)
if not (callable(func)):
import io
with io.StringIO() as ss:
print(
"If only one input, input must be callable",
"Instead, received:",
repr(func),
sep="
",
file=ss
)
msg = ss.getvalue()
raise TypeError(msg)
return i._cls(func, *i._args, **i._kwargs)
现在我们可以这样写：
dec = DelayedDecorator(Wait, delay=4)
@dec
def delayed_print(something):
print(something)
注意：
-
dec 不接受多个参数。
-
dec 只接受要包装的函数。
进口检验
PolyArgDecoratorMeta 类（类型）：
定义称呼（等等，*args，**kwargs）：
尝试：
arg_count = len(args)
如果（arg_count == 1）：
如果可调用（args [0]）：
超类 = inspect.getmro(PolyArgDecoratorMeta)[1]
r = 超类。称呼（等等，args[0]）
别的：
r = DelayedDecorator（等待，*args，**kwargs）
别的：
r = DelayedDecorator（等待，*args，**kwargs）
最后：
经过
返回
导入时间
等待类（metaclass=PolyArgDecoratorMeta）：
定义在里面(i, func, delay = 2):
i._func = 函数
i._delay = 延迟
def __call__(i, *args, **kwargs):
time.sleep(i._delay)
r = i._func(*args, **kwargs)
return r
下面两段代码是等价的：
@Wait
def print_something(something):
print (something)
##################################################
def print_something(something):
print(something)
print_something = Wait(print_something)
我们可以很慢的打印"something"到控制台，如下：
print_something("something")
#################################################
@Wait(delay=1)
def print_something_else(something_else):
print(something_else)
##################################################
def print_something_else(something_else):
print(something_else)
dd = DelayedDecorator(Wait, delay=1)
print_something_else = dd(print_something_else)
##################################################
print_something_else("something")
最后的笔记
它可能看起来像很多代码，但您不必每次都编写类DelayedDecorator 和PolyArgDecoratorMeta。您必须亲自编写的唯一代码如下，相当短：
from PolyArgDecoratorMeta import PolyArgDecoratorMeta
import time
class Wait(metaclass=PolyArgDecoratorMeta):
def __init__(i, func, delay = 2):
i._func = func
i._delay = delay
def __call__(i, *args, **kwargs):
time.sleep(i._delay)
r = i._func(*args, **kwargs)
return r