python笔记35-装饰器

前言

python装饰器本质上就是一个函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外的功能,装饰器的返回值也是一个函数对象。 很多python初学者学到面向对象类和方法是一道大坎,那么python中的装饰器是你进入Python高级语法大门的一道坎。

计算函数运行时间

假设你写了几个函数,有一天领导心血来潮说,你把每个函数的运行时长(结束时间-开始时间)统计下,作为一个python实习生的你可能会这样写

原始函数

import timedef func_a():
print("hello")
time.sleep(0.5)def func_b():
print("world")
time.sleep(0.8)if __name__ == '__main__':
func_a()
func_b()

添加运行时长

作为一个实习生的你,可能想到的解决办法如下

import timedef func_a():
start = time.time()
print("hello")
time.sleep(0.5)
end = time.time()
print("运行时长:%.4f 秒" % (end-start))def func_b():
start = time.time()
print("world")
time.sleep(0.8)
end = time.time()
print("运行时长:%.4f 秒" % (end-start))if __name__ == '__main__':
func_a()
func_b()

运行结果:

hello
运行时长:0.5009 秒
world
运行时长:0.8008 秒

上面的代码虽然满足了领导的要求,但是如果你写的函数很多的话,每个函数都这样去添加,会显得代码很臃肿,有很多重复代码。 有一天你边上的一个python老司机看了下你的代码,给你指了条路:装饰器

函数装饰器

装饰器可以写成函数式装饰器,也可以写成一个类装饰器,先从简单的函数装饰器开始学习。 python装饰器本质上就是一个函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外的功能,装饰器的返回值也是一个函数对象。

runtime函数就是一个装饰器了,它对原函数做了包装并返回了另外一个函数,额外添加了一些功能。在函数上方使用@语法糖就可以调用这个装饰器了

import timedef runtime(func):
def wrapper():
start = time.time()
f = func()     # 原函数
end = time.time()
print("运行时长:%.4f 秒" % (end-start))
return f
return wrapper@runtime
def func_a():
print("hello")
time.sleep(0.5)@runtime
def func_b():
print("world")
time.sleep(0.8)if __name__ == '__main__':
func_a()
func_b()

运行结果

hello
运行时长:0.5001 秒
world
运行时长:0.8001 秒

函数带参数装饰器

上面的runtime就是一个简单的装饰器模型了,但并不强壮,如果函数里面带有参数,那就不管用了,并且函数的参数是不固定的,这时候就需要用到*args,**kwargs两兄弟了

import timedef runtime(func):
def wrapper(*args, **kwargs):
start = time.time()
f = func(*args, **kwargs)     # 原函数
end = time.time()
print("运行时长:%.4f 秒" % (end-start))
return f
return wrapper@runtime
def func_a(a):
print("hello"+a)
time.sleep(0.5)@runtime
def func_b(b, c="xx"):
print("world"+b+c)
time.sleep(0.8)if __name__ == '__main__':
func_a("a")
func_b("b", c="xxx")

类装饰器

关于__call__方法,不得不先提到一个概念,就是可调用对象(callable),我们平时自定义的函数、内置函数和类都属于可调用对象, 但凡是可以把一对括号()应用到某个对象身上都可称之为可调用对象,判断对象是否为可调用对象可以用函数 callable。 如果在类中实现了__call__方法,那么实例对象也将成为一个可调用对象

import timeclass runtime(object):
def __init__(self, func):
self.func = funcdef __call__(self, *args, **kwargs):
start = time.time()
f = self.func(*args, **kwargs)     # 原函数
end = time.time()
print("运行时长:%.4f 秒" % (end-start))
return f@runtime
def func_a(a):
print("hello"+a)
time.sleep(0.5)@runtime
def func_b(b, c="xx"):
print("world"+b+c)
time.sleep(0.8)if __name__ == '__main__':
func_a("a")
func_b("b", c="xxx")

装饰器带参数

快到年底了,领导说运行的速度先不要太快了,让客户先加钱,然后再以正常的速度显示,那么现在的需求是让每个函数的运行时间加50%,该如何实现呢? 这就到了装饰器的高级语法,装饰器也需要带上参数了

函数装饰器

import timedef runtime(slowly=1):
def wrapper(func):
def inner_wrapper(*args, **kwargs):
start = time.time()
f = func(*args, **kwargs)     # 原函数
end = time.time()
t = end-start
time.sleep((slowly-1)*t)  # 延迟效果
new_end = time.time()
print("运行时长:%.4f 秒" % (new_end-start))
return f
return inner_wrapper
return wrapper@runtime(1.5)
def func_a(a):
print("hello"+a)
time.sleep(0.5)@runtime(1.5)
def func_b(b, c="xx"):
print("world"+b+c)
time.sleep(0.8)if __name__ == '__main__':
func_a("a")
func_b("b", c="xxx")

类装饰器

import timeclass runtime(object):
def __init__(self, slowly=1):
self.slowly = slowlydef __call__(self, func):
def wrapper(*args, **kwargs):
start = time.time()
f = func(*args, **kwargs)     # 原函数
end = time.time()
t = end-start
time.sleep((self.slowly-1)*t)  # 延迟效果
new_end = time.time()
print("运行时长:%.4f 秒" % (new_end-start))
return f
return wrapper@runtime(1.5)
def func_a(a):
print("hello"+a)
time.sleep(0.5)@runtime(1.5)
def func_b(b, c="xx"):
print("world"+b+c)
time.sleep(0.8)if __name__ == '__main__':
func_a("a")
func_b("b", c="xxx")

使用场景

用哪些地方需要使用装饰器呢?

  • 如果你用过locust,设置权重会用到@task(1),
  • 如果你用过pytest框架,使用fixture功能的时候经常会用到@pytest.fixture(scope="function")
  • allure里面可以添加测试步骤 @allure.step('修改购物车')
  • 被大量使用于Flask和Django web框架中,检查是否被授权去使用一个web应用的端点(endpoint)。如 @login_required
  • 也可以自己写个装饰器添加日志

本文分享自微信公众号 - 从零开始学自动化测试(yoyoketang),作者:上海悠悠

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2019-05-01

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 面试题-python 什么是装饰器(decorator )?

    python装饰器本质上就是一个函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外的功能,装饰器的返回值也是一个函数对象。 很多python初学者学到...

    上海-悠悠
  • python笔记36-装饰器之wraps

    前面一篇对python装饰器有了初步的了解了,但是还不够完美,领导看了后又提出了新的需求,希望运行的日志能显示出具体运行的哪个函数。

    上海-悠悠
  • Cypress web自动化28-运行器界面调试元素定位和操作

    Cypress提供了一个很好的测试运行器, 它为你提供了一套可视化结构的测试和断言套件, 很快你也会看到命令, 页面事件, 网络请求等. 当你还没熟练掌握元素定...

    上海-悠悠
  • 单例模式《单例模式概念、什么时候能用到单例模式、单例模式的好处》

    __new__方法是用来创建实例对象的,通过类去创建方法的时候,实际上类自动调用的__new__方法。

    清菡
  • Python之路【第七篇】:Python

    py3study
  • day11-装饰器

      通常为了是代码更加优美,于是引入语法糖的概念。这里将对上述第二段代码进行改进,他们的作用和结果完全相同。

    py3study
  • Python 学习入门(18)—— 字符串

    一种方法是 a.isdigit(),但这种方法对于包含正负号的数字字符串无效,更为准确的为:

    阳光岛主
  • ON-LSTM:能表示语言层次的LSTM

    LSTM作为序列模型一直是自然语言处理的最佳选择之一,即使transformer出现了也依然无法撼动LSTM在NLP界的江湖地位。

    beyondGuo
  • 面向对象语言的三大特征: 封装 继承 多态(二)——继承

    子类去重写父类中的方法, 当子类重写了父类中的方法,子类再调用该方法时 调用的是子类重写之后的

    诡途
  • 面试题系列第6篇:JVM字符串常量池及String的intern方法详解?

    关于字符串的比较在前面文章中已经详解过,本篇文章基于字符串常量池的存储及在使用intern方法时所引起的内存变化进行一步深层次的讲解。

    程序新视界

扫码关注云+社区

领取腾讯云代金券