python装饰器同时支持有参数和无参数的练习题

'''

预备知识:

……

@decorator

def f(*args,**kwargs):

pass

# 此处@decorator  等价于 f = decorator(f)

@decorator2(*args,**kwargs)

def f2(*args,**kwargs):

pass

# 此处@decorator2(*args,**kwargs)  等价于: 第一步，执行函数decorator2(*args,**kwargs),获得中间结果temp;  第二步，执行装饰器函数（即@temp）

'''

import functools,time

def log(*args,**kwargs):

def metric(fn):

@functools.wraps(fn)

def t(*args,**kwargs):  #计算函数执行消耗的时间

t0 = time.time()

result = fn(*args, **kwargs)

t1 = time.time()

print('%s executed in %s ms' % (fn.__name__, t1 - t0))

return result

return t

for i in args:

if type(i).__name__ == 'function':    #判断传入的参数是函数名称（即log后面没有带参数）

return metric(i)   #返回并执行装饰器函数metric(),相当于log是metric的别名。

else:  # log后面带普通参数的情形

print(i)   # 理论上，log后面可传入无限个参数，如何使用应结合具体情形，本例仅遍历打印，且参数中不包含‘function’类型。

return metric  # log后面带普通参数，说明log()是个普通函数，执行log()后，返回装饰器函数的名字metric，再执行metric().

# 测试

@log("execute")

def fast(x, y):

time.sleep(0.0012)

return x + y

@log

def slow(x, y, z):

time.sleep(0.1234)

return x * y * z

f = fast(11, 22)

s = slow(11, 22, 33)

if f != 33:

print('测试失败!')

elif s != 7986:

print('测试失败!')