Python编程中装饰器的使用示例解析


Posted in Python onJune 20, 2016

装饰函数和方法

我们先定义两个简单的数学函数,一个用来计算平方和,一个用来计算平方差:

# get square sum
def square_sum(a, b):
  return a**2 + b**2

# get square diff
def square_diff(a, b):
  return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

在拥有了基本的数学功能之后,我们可能想为函数增加其它的功能,比如打印输入。我们可以改写函数来实现这一点:

# modify: print input

# get square sum
def square_sum(a, b):
  print("intput:", a, b)
  return a**2 + b**2

# get square diff
def square_diff(a, b):
  print("input", a, b)
  return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

我们修改了函数的定义,为函数增加了功能。

现在,我们使用装饰器来实现上述修改:

def decorator(F):
  def new_F(a, b):
    print("input", a, b)
    return F(a, b)
  return new_F

# get square sum
@decorator
def square_sum(a, b):
  return a**2 + b**2

# get square diff
@decorator
def square_diff(a, b):
  return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

装饰器可以用def的形式定义,如上面代码中的decorator。装饰器接收一个可调用对象作为输入参数,并返回一个新的可调用对象。装饰器新建了一个可调用对象,也就是上面的new_F。new_F中,我们增加了打印的功能,并通过调用F(a, b)来实现原有函数的功能。

定义好装饰器后,我们就可以通过@语法使用了。在函数square_sum和square_diff定义之前调用@decorator,我们实际上将square_sum或square_diff传递给decorator,并将decorator返回的新的可调用对象赋给原来的函数名(square_sum或square_diff)。 所以,当我们调用square_sum(3, 4)的时候,就相当于:

square_sum = decorator(square_sum)
square_sum(3, 4)

我们知道,Python中的变量名和对象是分离的。变量名可以指向任意一个对象。从本质上,装饰器起到的就是这样一个重新指向变量名的作用(name binding),让同一个变量名指向一个新返回的可调用对象,从而达到修改可调用对象的目的。

与加工函数类似,我们可以使用装饰器加工类的方法。

如果我们有其他的类似函数,我们可以继续调用decorator来修饰函数,而不用重复修改函数或者增加新的封装。这样,我们就提高了程序的可重复利用性,并增加了程序的可读性。

含参的装饰器

在上面的装饰器调用中,比如@decorator,该装饰器默认它后面的函数是唯一的参数。装饰器的语法允许我们调用decorator时,提供其它参数,比如@decorator(a)。这样,就为装饰器的编写和使用提供了更大的灵活性。

# a new wrapper layer
def pre_str(pre=''):
  # old decorator
  def decorator(F):
    def new_F(a, b):
      print(pre + "input", a, b)
      return F(a, b)
    return new_F
  return decorator

# get square sum
@pre_str('^_^')
def square_sum(a, b):
  return a**2 + b**2

# get square diff
@pre_str('T_T')
def square_diff(a, b):
  return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

上面的pre_str是允许参数的装饰器。它实际上是对原有装饰器的一个函数封装,并返回一个装饰器。我们可以将它理解为一个含有环境参量的闭包。当我们使用@pre_str('^_^')调用的时候,Python能够发现这一层的封装,并把参数传递到装饰器的环境中。该调用相当于:

square_sum = pre_str('^_^') (square_sum)

装饰类

在上面的例子中,装饰器接收一个函数,并返回一个函数,从而起到加工函数的效果。在Python 2.6以后,装饰器被拓展到类。一个装饰器可以接收一个类,并返回一个类,从而起到加工类的效果。

def decorator(aClass):
  class newClass:
    def __init__(self, age):
      self.total_display  = 0
      self.wrapped     = aClass(age)
    def display(self):
      self.total_display += 1
      print("total display", self.total_display)
      self.wrapped.display()
  return newClass

@decorator
class Bird:
  def __init__(self, age):
    self.age = age
  def display(self):
    print("My age is",self.age)

eagleLord = Bird(5)
for i in range(3):
  eagleLord.display()

在decorator中,我们返回了一个新类newClass。在新类中,我们记录了原来类生成的对象(self.wrapped),并附加了新的属性total_display,用于记录调用display的次数。我们也同时更改了display方法。

通过修改,我们的Bird类可以显示调用display的次数了。

Python 相关文章推荐
python 正则式 概述及常用字符
May 07 Python
python实现无证书加密解密实例
Oct 27 Python
深入讲解Python函数中参数的使用及默认参数的陷阱
Mar 13 Python
Window10+Python3.5安装opencv的教程推荐
Apr 02 Python
python 应用之Pycharm 新建模板默认添加编码格式-作者-时间等信息【推荐】
Jun 17 Python
pandas删除行删除列增加行增加列的实现
Jul 06 Python
使用python模拟命令行终端的示例
Aug 13 Python
Python操作qml对象过程详解
Sep 26 Python
通过实例简单了解Python中yield的作用
Dec 11 Python
python获取引用对象的个数方式
Dec 20 Python
python 多进程和协程配合使用写入数据
Oct 30 Python
python+pytest接口自动化之token关联登录的实现
Apr 06 Python
12步入门Python中的decorator装饰器使用方法
Jun 20 #Python
深入学习Python中的装饰器使用
Jun 20 #Python
Python中Iterator迭代器的使用杂谈
Jun 20 #Python
实例讲解Python编程中@property装饰器的用法
Jun 20 #Python
Python的包管理器pip更换软件源的方法详解
Jun 20 #Python
python3.5使用tkinter制作记事本
Jun 20 #Python
浅谈python抛出异常、自定义异常, 传递异常
Jun 20 #Python
You might like
DC动漫人物排行
2020/03/03 欧美动漫
大师制作的中短波矿石收音机
2020/04/02 无线电
URL Rewrite的设置方法
2007/01/02 PHP
基于php下载文件的详解
2013/06/02 PHP
析构函数与php的垃圾回收机制详解
2013/10/28 PHP
php构造方法中析构方法在继承中的表现
2016/04/12 PHP
PHP编写的图片验证码类文件分享
2016/06/06 PHP
网上应用的一个不错common.js脚本
2007/08/08 Javascript
如何让DIV可编辑、可拖动示例代码
2013/09/18 Javascript
JavaScript cookie的设置获取删除详解
2014/02/11 Javascript
nodejs下打包模块archiver详解
2014/12/03 NodeJs
JavaScript函数作用域链分析
2015/02/13 Javascript
javascript中undefined与null的区别
2015/08/16 Javascript
Node.js与Sails ~项目结构与Mvc实现及日志机制
2015/10/14 Javascript
用 js 的 selection range 操作选择区域内容和图片
2017/04/18 Javascript
jquery Ajax实现Select动态添加数据
2017/06/08 jQuery
react-router browserHistory刷新页面404问题解决方法
2017/12/29 Javascript
产制造追溯系统之通过微信小程序实现移动端报表平台
2019/06/03 Javascript
深度解读vue-resize的具体用法
2020/07/08 Javascript
javascript实现京东快递单号的查询效果
2020/11/30 Javascript
[05:34]2014DOTA2国际邀请赛中国区预选赛精彩TOPPLAY第二弹
2014/06/25 DOTA
python去掉字符串中重复字符的方法
2014/02/27 Python
python实现保存网页到本地示例
2014/03/16 Python
Python注释详解
2016/06/01 Python
python开发环境PyScripter中文乱码问题解决方案
2016/09/11 Python
python爬虫使用cookie登录详解
2017/12/27 Python
python 多个参数不为空校验方法
2019/02/14 Python
PYQT5开启多个线程和窗口,多线程与多窗口的交互实例
2019/12/13 Python
Pycharm小白级简单使用教程
2020/01/08 Python
Pycharm创建文件时自动生成文件头注释(自定义设置作者日期)
2020/11/24 Python
美国现代家具和家居商店:Apt2B
2016/08/29 全球购物
奥地利手表、香水、化妆品和珠宝购物网站:Brasty.at
2021/01/17 全球购物
酒店人事专员岗位职责
2013/12/19 职场文书
阳光体育活动实施方案
2014/05/25 职场文书
上课随便讲话检讨书
2014/09/12 职场文书
医院2014国庆节活动策划方案
2014/09/21 职场文书