装饰器
python装饰器就是用于在不改变原函数代码的情况下拓展新功能的一种函数,这个函数的特殊之处在于它的返回值也是一个函数,使用python装饰器的好处就是在不用更改原函数的代码前提下给函数增加新的功能。
装饰器分为有参和无参两种:
无参:
import time
def happy(func):
def f1(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)
end_time = time.time()
print('run time is %s' % (end_time-start_time))
return f1
@happy #auto=happy(auto)
def auto(name, password):
print(name, password)
@happy
def shadow():
time.sleep(3)
print("hello_world")
shadow()
auto('wzc', '123456')
auto函数为原函数,我们需要对其进行功能扩展
因为原函数可能有形参或者没有,因此我们将函数定义为 def f1(*args, **kwargs),这样不管是有参还是无参,都可以接收,不会报错
这里的happy函数就是最原始的装饰器,把auto当做了一个参数传了进去 auto=happy(auto)
然后返回值也是一个函数,返回一个f1 相当于auto=f1
其中作为参数的这个函数auto就在返回函数f1的内部执行。在函数auto前面加上@happy,auto函数就相当于被注入了计时功能,现在只要调用auto,它就已经变身为“新的功能更多”的函数了。
有参
有参装饰器与无参装饰器的区别就是在外面多加一个嵌套函数,实现闭包功能
import time
def mom(x='blee'):
def happy(func):
def f1(*args, **kwargs):
print(x)
start_time = time.time()
func(*args, **kwargs)
end_time = time.time()
print('run time is %s' % (end_time-start_time))
return f1
return happy
@mom('red')
def auto(name, password):
print(name, password)
auto('wzc', '123456')
在原函数前面写@装饰器名,并加入参数 @mom('red')
这时候我们先看mom('red'),传回一个函数happy,等同于 @happy
接下来就和上面一样,这样就实现了有参装饰器