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类的动态修改的实例方法
Mar 24 Python
Python 实现数据库(SQL)更新脚本的生成方法
Jul 09 Python
Python自然语言处理之词干,词形与最大匹配算法代码详解
Nov 16 Python
Python判断两个list是否是父子集关系的实例
May 04 Python
使用Python进行体育竞技分析(预测球队成绩)
May 16 Python
Python3 实现文件批量重命名示例代码
Jun 03 Python
python3读取csv文件任意行列代码实例
Jan 13 Python
Python 找出出现次数超过数组长度一半的元素实例
May 11 Python
Python 字符串池化的前提
Jul 03 Python
Python3交互式shell ipython3安装及使用详解
Jul 11 Python
Python logging自定义字段输出及打印颜色
Nov 30 Python
解决Python import .pyd 可能遇到路径的问题
Mar 04 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
php开发工具之vs2005图解
2008/01/12 PHP
php压缩多个CSS为一个css的代码并缓存
2011/04/21 PHP
yii操作cookie实例简介
2014/07/09 PHP
支付宝接口开发集成支付环境小结
2015/03/17 PHP
php获取服务器操作系统相关信息的方法
2016/10/08 PHP
PHP回调函数概念与用法实例分析
2017/11/03 PHP
js 延迟加载 改变JS的位置加快网页加载速度
2012/12/11 Javascript
如何使用jQUery获取选中radio对应的值(一句代码)
2013/06/03 Javascript
javascript异步处理工作机制详解
2015/04/13 Javascript
jquery实现通用的内容渐显Tab选项卡效果
2015/09/07 Javascript
浅谈json取值(对象和数组)
2016/06/24 Javascript
javascript实现无法关闭的弹框
2016/11/27 Javascript
AngularJS指令中的绑定策略实例分析
2016/12/14 Javascript
javascript实现Java中的Map对象功能的实例详解
2017/08/21 Javascript
JS实现区分中英文并统计字符个数的方法示例
2018/06/09 Javascript
vue+echarts实现动态绘制图表及异步加载数据的方法
2018/10/17 Javascript
基于Vue实现的多条件筛选功能的详解(类似京东和淘宝功能)
2019/05/07 Javascript
详解Vue前端生产环境发布配置实战篇
2019/05/07 Javascript
用Python写的图片蜘蛛人代码
2012/08/27 Python
python解析模块(ConfigParser)使用方法
2013/12/10 Python
python中实现k-means聚类算法详解
2017/11/11 Python
Django中多种重定向方法使用详解
2019/07/17 Python
PyQt5通信机制 信号与槽详解
2019/08/07 Python
python selenium 执行完毕关闭chromedriver进程示例
2019/11/15 Python
Python turtle画图库&&画姓名实例
2020/01/19 Python
CSS3模拟动画下拉菜单效果
2017/04/12 HTML / CSS
html5仿支付宝密码框的实现代码
2017/09/06 HTML / CSS
HTML5的结构和语义(2):结构
2008/10/17 HTML / CSS
西班牙美妆电商:Perfume’s Club(有中文站)
2018/08/08 全球购物
QA工程师岗位职责
2013/11/20 职场文书
外贸采购员求职的自我评价
2013/11/26 职场文书
酒店副总岗位职责
2013/12/24 职场文书
员工评语大全
2014/01/19 职场文书
解除租房协议书
2014/12/03 职场文书
会计工作总结范文2014
2014/12/23 职场文书
Vue通过懒加载提升页面响应速度
2021/05/10 Vue.js