python装饰器原理与用法深入详解


Posted in Python onDecember 19, 2019

本文实例讲述了python装饰器原理与用法。分享给大家供大家参考,具体如下:

你会Python嘛?
我会!
那你给我讲下Python装饰器吧!
Python装饰器啊?我没用过哎

以上是我一个哥们面试时候发生的真实对白。

----------------------------------------------分割线------------------------------------------------------------------------------

简言之,python装饰器就是用于拓展原来函数功能的一种函数,这个函数的特殊之处在于它的返回值也是一个函数,使用python装饰器的好处就是在不用更改原函数的代码前提下给函数增加新的功能。
一般而言,我们要想拓展原来函数代码,最直接的办法就是侵入代码里面修改,例如:

import time
def func():
  print("hello")
  time.sleep(1)
  print("world")

这是我们最原始的的一个函数,然后我们试图记录下这个函数执行的总时间,那最简单的做法就是:

#原始侵入,篡改原函数
import time
def func():
  startTime = time.time()
  print("hello")
  time.sleep(1)
  print("world")
  endTime = time.time()
  msecs = (endTime - startTime)*1000
  print("time is %d ms" %msecs)

但是如果你的Boss在公司里面和你说:“小祁,这段代码是我们公司的核心代码,你不能直接去改我们的核心代码。”那该怎么办呢,我们仿照装饰器先自己试着写一下:

#避免直接侵入原函数修改,但是生效需要再次执行函数
import time
def deco(func):
  startTime = time.time()
  func()
  endTime = time.time()
  msecs = (endTime - startTime)*1000
  print("time is %d ms" %msecs)
def func():
  print("hello")
  time.sleep(1)
  print("world")
if __name__ == '__main__':
  f = func
  deco(f)#只有把func()或者f()作为参数执行,新加入功能才会生效
  print("f.__name__ is",f.__name__)#f的name就是func

这里我们定义了一个函数deco,它的参数是一个函数,然后给这个函数嵌入了计时功能。然后你可以拍着胸脯对老板说,看吧,不用动你原来的代码,我照样拓展了它的函数功能。

然后你的老板有对你说:“小祁,我们公司核心代码区域有一千万个func()函数,从func01()到func1kw(),按你的方案,想要拓展这一千万个函数功能,就是要执行一千万次deco()函数,这可不行呀,我心疼我的机器。”

好了,你终于受够你老板了,准备辞职了,然后你无意间听到了装饰器这个神器,突然发现能满足你闫博士的要求了。

我们先实现一个最简陋的装饰器,不使用任何语法糖和高级语法,看看装饰器最原始的面貌:

#既不需要侵入,也不需要函数重复执行
import time
def deco(func):
  def wrapper():
    startTime = time.time()
    func()
    endTime = time.time()
    msecs = (endTime - startTime)*1000
    print("time is %d ms" %msecs)
  return wrapper
@deco
def func():
  print("hello")
  time.sleep(1)
  print("world")
if __name__ == '__main__':
  f = func #这里f被赋值为func,执行f()就是执行func()
  f()

这里的deco函数就是最原始的装饰器,它的参数是一个函数,然后返回值也是一个函数。其中作为参数的这个函数func()就在返回函数wrapper()的内部执行。然后在函数func()前面加上@deco,func()函数就相当于被注入了计时功能,现在只要调用func(),它就已经变身为“新的功能更多”的函数了。

所以这里装饰器就像一个注入符号:有了它,拓展了原来函数的功能既不需要侵入函数内更改代码,也不需要重复执行原函数。

#带有参数的装饰器
import time
def deco(func):
  def wrapper(a,b):
    startTime = time.time()
    func(a,b)
    endTime = time.time()
    msecs = (endTime - startTime)*1000
    print("time is %d ms" %msecs)
  return wrapper
@deco
def func(a,b):
  print("hello,here is a func for add :")
  time.sleep(1)
  print("result is %d" %(a+b))
if __name__ == '__main__':
  f = func
  f(3,4)
  #func()

然后你满足了Boss的要求后,Boss又说:“小祁,我让你拓展的函数好多可是有参数的呀,有的参数还是个数不定的那种,你的装饰器搞的定不?”然后你嘿嘿一笑,深藏功与名!

#带有不定参数的装饰器
import time
def deco(func):
  def wrapper(*args, **kwargs):
    startTime = time.time()
    func(*args, **kwargs)
    endTime = time.time()
    msecs = (endTime - startTime)*1000
    print("time is %d ms" %msecs)
  return wrapper
@deco
def func(a,b):
  print("hello,here is a func for add :")
  time.sleep(1)
  print("result is %d" %(a+b))
@deco
def func2(a,b,c):
  print("hello,here is a func for add :")
  time.sleep(1)
  print("result is %d" %(a+b+c))
if __name__ == '__main__':
  f = func
  func2(3,4,5)
  f(3,4)
  #func()

最后,你的老板说:“可以的,小祁,我这里一个函数需要加入很多功能,一个装饰器怕是搞不定,装饰器能支持多个嘛"

最后你就把这段代码丢给了他:

#多个装饰器
import time
def deco01(func):
  def wrapper(*args, **kwargs):
    print("this is deco01")
    startTime = time.time()
    func(*args, **kwargs)
    endTime = time.time()
    msecs = (endTime - startTime)*1000
    print("time is %d ms" %msecs)
    print("deco01 end here")
  return wrapper
def deco02(func):
  def wrapper(*args, **kwargs):
    print("this is deco02")
    func(*args, **kwargs)
    print("deco02 end here")
  return wrapper
@deco01
@deco02
def func(a,b):
  print("hello,here is a func for add :")
  time.sleep(1)
  print("result is %d" %(a+b))
if __name__ == '__main__':
  f = func
  f(3,4)
  #func()
'''
this is deco01
this is deco02
hello,here is a func for add :
result is 7
deco02 end here
time is 1003 ms
deco01 end here
'''

多个装饰器执行的顺序就是从最后一个装饰器开始,执行到第一个装饰器,再执行函数本身。

盗用评论里面一位童鞋的例子:

def dec1(func):
  print("1111")
  def one():
    print("2222")
    func()
    print("3333")
  return one
def dec2(func):
  print("aaaa")
  def two():
    print("bbbb")
    func()
    print("cccc")
  return two
@dec1
@dec2
def test():
  print("test test")
test()

输出:

aaaa
1111
2222
bbbb
test test
cccc
3333

装饰器的外函数和内函数之间的语句是没有装饰到目标函数上的,而是在装载装饰器时的附加操作。

17~20行是装载装饰器的过程,相当于执行了test=dect1(dect2(test)),此时先执行dect2(test),结果是输出aaaa、将func指向函数test、并返回函数two,然后执行dect1(two),结果是输出1111、将func指向函数two、并返回函数one,然后进行赋值。

用函数替代了函数名test。 22行则是实际调用被装载的函数,这时实际上执行的是函数one,运行到func()时执行函数two,再运行到func()时执行未修饰的函数test。

希望本文所述对大家Python程序设计有所帮助。

Python 相关文章推荐
Python中Random和Math模块学习笔记
May 18 Python
python数据类型判断type与isinstance的区别实例解析
Oct 31 Python
Python 多进程并发操作中进程池Pool的实例
Nov 01 Python
Sanic框架配置操作分析
Jul 17 Python
Python创建一个空的dataframe,并循环赋值的方法
Nov 08 Python
详解python执行shell脚本创建用户及相关操作
Apr 11 Python
python爬虫豆瓣网的模拟登录实现
Aug 21 Python
Python使用正则实现计算字符串算式
Dec 29 Python
python+selenium 脚本实现每天自动登记的思路详解
Mar 11 Python
python 在sql语句中使用%s,%d,%f说明
Jun 06 Python
Python中的特殊方法以及应用详解
Sep 20 Python
详解pandas赋值失败问题解决
Nov 29 Python
python列表生成器迭代器实例解析
Dec 19 #Python
Python tensorflow实现mnist手写数字识别示例【非卷积与卷积实现】
Dec 19 #Python
Python: 传递列表副本方式
Dec 19 #Python
python内置模块collections知识点总结
Dec 19 #Python
Python操作redis和mongoDB的方法
Dec 19 #Python
Python 实现Serial 与STM32J进行串口通讯
Dec 18 #Python
实现Python与STM32通信方式
Dec 18 #Python
You might like
php数据库配置文件一般做法分享
2012/07/07 PHP
apache中为php 设置虚拟目录
2014/12/17 PHP
利用Homestead快速运行一个Laravel项目的方法详解
2017/11/14 PHP
浅谈laravel aliases别名的原理
2019/10/24 PHP
PHP SESSION跨页面传递失败解决方案
2020/12/11 PHP
iframe 异步加载技术及性能分析
2011/07/19 Javascript
Javascript表格翻页效果的具体实现
2013/10/05 Javascript
jQuery实现页面内锚点平滑跳转特效的方法总结
2015/05/11 Javascript
Angular Js文件上传之form-data
2015/08/28 Javascript
jquery动态增加删减表格行特效
2015/11/20 Javascript
基于javascript实现动态显示当前系统时间
2016/01/28 Javascript
JavaScript获取客户端IP的方法(新方法)
2016/03/11 Javascript
JS onkeypress兼容性写法详解
2016/04/27 Javascript
第一次动手实现bootstrap table分页效果
2016/09/22 Javascript
js实现弹窗暗层效果
2017/01/16 Javascript
如何写好你的JavaScript【推荐】
2017/03/02 Javascript
jQuery实现base64前台加密解密功能详解
2017/08/29 jQuery
简单谈谈关于Angular Cli打包的事
2017/09/05 Javascript
详解小程序退出页面时清除定时器
2019/04/28 Javascript
layer.open 子页面弹出层向父页面传输数据的例子
2019/09/26 Javascript
layer弹出层倒计时关闭的实现方法
2019/09/27 Javascript
用JS实现一个简单的打砖块游戏
2019/12/11 Javascript
详解Python的Django框架中Manager方法的使用
2015/07/21 Python
Python Xml文件添加字节属性的方法
2018/03/31 Python
Python学习笔记之open()函数打开文件路径报错问题
2018/04/28 Python
python解决js文件utf-8编码乱码问题(推荐)
2018/05/02 Python
详解Python正则表达式re模块
2019/03/19 Python
Python通过Tesseract库实现文字识别
2020/03/05 Python
详解Python中的编码问题(encoding与decode、str与bytes)
2020/09/30 Python
Selenium环境变量配置(火狐浏览器)及验证实现
2020/12/07 Python
浅析HTML5:'data-'属性的作用
2018/01/23 HTML / CSS
销售总经理岗位职责
2014/03/15 职场文书
《大江保卫战》教学反思
2014/04/11 职场文书
中学生的1000字检讨书
2014/10/11 职场文书
质量负责人岗位职责
2015/02/15 职场文书
2015年学校工作总结范文
2015/04/20 职场文书