python利器-装饰器
说装饰器之前，先举一个生活的例子，比如秋天的时候，我们只穿一件夹克就可以保暖，但是到了冬天的时候，我想让这件夹克更加保暖一点，给夹克加点棉，到春天的时候，天气那么冷了，还想穿这件夹克，但是因为加了棉，会比较热，需要把棉去掉。但是这样显得比较麻烦，假如我们在不改变夹克的基础上，我们穿一件保暖衣服，天气热的时候就脱掉，让我们随时能够保暖也不至于太热，这个保暖衣服就有点相当于python的装饰器，python装饰器可以扩展原来函数的功能，并且不改变原来函数，用好装饰器，有时候能让我写代码事半功倍。
一、体验装饰器
请看以下函数,现在有一个需求是统计sleep_time函数的运行时间，这时候我可能第一时间，应该会增加一个函数用来统计sleep_time的运行，原谅我这么笨。但是如果我们需要统计其他函数呢？又编写其他函数来统计吗？那岂不是要命？装饰器很好解决了这个问题，让函数自己增加统计自己运行的时间。
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def sleep_time():
time.sleep(2)
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def sleep_time():
time.sleep(2)
def count_time():
start_time = time.time()
sleep_time()
end_time = time.time()
use_time = end_time - start_time
print(f'函数sleep_time运行的时间是{use_time}')
if __name__ == '__main__':
count_time()
增加装饰器，函数count_sec就是一个装饰器，并且通过@语法糖使用装饰器，我们直接运行sleep_time函数，效果是和count_time函数是一样的。我们可以看到被装饰器修饰的函数，会被当作参数传入装饰器函数。
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def count_sec(func):
def wrapper():
start_time = time.time()
func()
end_time = time.time()
use_time = end_time - start_time
print(f'{func.__name__}运行消耗的时间是{use_time}')
return wrapper
@count_sec
def sleep_time():
time.sleep(2)
def count_time():
start_time = time.time()
sleep_time()
end_time = time.time()
use_time = end_time - start_time
print(f'函数sleep_time运行的时间是{use_time}')
if __name__ == '__main__':
sleep_time()
二、被装饰函数带参数
修改sleep_time函数和装饰器函数，如下，运行后，效果是一样的。
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def count_sec(func):
def wrapper(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)
end_time = time.time()
use_time = end_time - start_time
print(f'{func.__name__}运行消耗的时间是{use_time}')
return wrapper
@count_sec
def sleep_time(sec):
time.sleep(sec)
if __name__ == '__main__':
sleep_time(2)
三、被装饰函数有返回值
再修改sleep_time函数和装饰器函数，如下，运行后，效果是一样的。
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def count_sec(func):
def wrapper(*args, **kwargs):
start_time = time.time()
return_value = func(*args, **kwargs)
end_time = time.time()
use_time = end_time - start_time
print(f'{func.__name__},{return_value}运行消耗的时间是{use_time}')
return return_value
return wrapper
@count_sec
def sleep_time(sec):
time.sleep(sec)
return "sleep end"
if __name__ == '__main__':
sleep_time(2)四、装饰器带参数
要装饰器带参数，再包一层函数即可，如下
#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
def count_who_sleep(who):
def count_sec(func):
def wrapper(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)
end_time = time.time()
use_time = end_time - start_time
print(f'{who}睡觉，睡了多久{use_time}')
return wrapper
return count_sec
@count_who_sleep(who="dog")
def sleep_time(sec):
time.sleep(sec)
return "sleep end"
if __name__ == '__main__':
sleep_time(2)除了函数装饰器，还有类装饰器，但是在这里就不分享，有兴趣的同学可以去了解下。今天就分享到这里！
本文分享自 暴走的软件测试Tester 微信公众号，前往查看
如有侵权，请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ，欢迎热爱写作的你一起参与！
腾讯云开发者
