前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >让python装饰器不再晦涩难懂

让python装饰器不再晦涩难懂

作者头像
qsjs
发布2020-09-28 15:49:44
3030
发布2020-09-28 15:49:44
举报

在学习修饰器函数之前,考虑如下的两种情形: a. 有一个函数fn,我们不能修改函数的源代码,但是要扩展函数的功能,怎么实现呢? 我们可以定义一个新的函数 deco, 然后执行fn=deco(fn)这样的赋值操作,那么在原来fn 被调用的时候,其实是执行了 deco(fn)的操作. 如果这个方式可行,我们面临如下的问题: deco函数必须接受一个函数作为参数,同时deco函数也必须要返回一个函数. 很显然,接受的函数必须是函数fn, 而返回的函数既需要具备fn的功能,还需有扩展的功能。 因此 deco 函数的结构可以如下:

代码语言:javascript
复制
def deco(fn):
    def wrapper(*args,**kwargs):  #此处的参数目的是传递给 fn 函数. 
        CODE ...   #New function code if any
        fn(*args,**kwargs)
        CODE ...   #New function code if any
    return wrapper
#调用方式如下:
fn=deco(fn)
fn(*args,**kwargs)    #不改变fn函数的调用方式

b. 有函数wrapper, 我们在不改变该函数调用方式情况下,如何传递额外的参数给函数wrapper ? 我们可以利用赋值的方式改变wrapper,比如 wrapper=func(args), 其中args就是我们要传递的额外的参数,func 是一个新的函数. 该func函数需要: 接受我们要传递的额外的参数args, 同时需要返回一个函数,返回的函数应该就是wrapper 函数. 至于传入的参数args 如何给到wrapper 函数呢 ,需要在wrapper的代码中来接受这个参数,一种可能的结构如下:

代码语言:javascript
复制
def func(args):
    CODE....  #Code if any required.   
    tmp_para=args     #使用tmp 来接受额外的参数args
    def wrapper():
        extra_para=tmp_para    # 实现额外参数args传递给wrapper函数.
        CODE...   # wrapper 函数的code 。
    return wrapper 
    CODE....   #如果存在,这部分code 可以是处理参数s 相关的 
#调用方式如下:
wrapper=func(args)   #args 就是额外要传递的参数 
wrapper(*args,**kwargs)   #分别是wrapper自己本身的参数

在上面的两种情形中,我们分别实现了: A. 通过赋值 fn=deco(fn) , 在不改变函数源代码的情况下,扩展了函数fn的功能. B. 通过赋值wrapper=func(args),利用新的函数func 来传递额外的参数给wrapper函数,但是wrapper函数的源代码必须要进行修改, 如果wrapper函数的源代码不能修改,那么传递额外参数给wrapper又有何意义呢? 下面看一个例子

代码语言:javascript
复制
#!/usr/bin/env python
# -*- coding:utf-8 -*-
#Author: PandaEye

def welcome(name):      #A very simple function.
        print("Welcome to my blog: %s !" % name)

def wrapper(gift):
        ext_para=gift  #Extra parameter is : gift , pass it to function deco.

        def deco(fn):    #The deco extend the function and no change the code of fn. 
                def fun(*args,**kwargs):
                        print("I am the new function, send you 10 points more for congratulations!")
                        fn(*args,**kwargs)
                print("Congratulations, you got the 1st, send the gift %s to you !" % ext_para)
                return fun
        return deco

deco=wrapper("BMW")
welcome=deco(welcome)

welcome("XiaoMa")

在上述的例子中,主程序中的函数welcome的调用方式始终没有改变,但是函数功能经过了两次的扩展: 其中第一次使用deco函数进行扩展. 在第一次deco扩展的基础上,我们需要传递额外的参数gift给deco, 很显然无法直接传递给deco,因为deco所接受的参数必须是fn, 所以我们用wrapper函数进行传递. 传递该参数给deco之后,我们修改了deco, 因为deco 要处理这个传入的参数,那么就必须要修改deco 的代码。至此完成了两次扩展.

理解了上述过程,其实就理解了装饰器函数,装饰器函数: 装饰器函数其实就是把上述 两行的赋值代码变成了 @wrapper("BMW"),然后调整位置而已。 对于装饰器函数的编写过程.

个人更乐于: 从deco 函数写起,然后再做二次扩展(也就是为deco函数传递参数),这就是外围的wrapper函数,写完deco和wrapper函数,那么装饰器函数就完成了.

下面是上面例子的装饰器常用写法,两者非常的相似,理解了上述的调用,就理解了这个 装饰器函数了.

代码语言:javascript
复制
#!/usr/bin/env python
# -*- coding:utf-8 -*-
#Author: PandaEye

def wrapper(gift):
        ext_para=gift  #Extra parameter is : gift , pass it to function deco.

        def deco(fn):    #The deco extend the function and no change the code of fn.
                def fun(*args,**kwargs):
                        print("I am the new function, send you 10 points more for congratulations!")
                        fn(*args,**kwargs)
                print("Congratulations, you got the 1st, send the gift %s to you !" % ext_para)
                return fun
        return deco

#deco=wrapper("BMW") 
#welcome=deco(welcome)
@wrapper("BMW")           #这一条语句和紧邻的上面的两条语句作用相同,这是装饰器函数常用表达方式, 调用装饰器函数之前,必须先定义,所以wrapper 函数定义一定要在这一句之前,否则报错
def welcome(name):      #A very simple function.
        print("Welcome to my blog: %s !" % name)

welcome("XiaoMa")
本文原创,转载请注明出处 !
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 个人更乐于: 从deco 函数写起,然后再做二次扩展(也就是为deco函数传递参数),这就是外围的wrapper函数,写完deco和wrapper函数,那么装饰器函数就完成了.
  • 本文原创,转载请注明出处 !
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档